def test_nick_change_graph(self, mock_get_year_month_day,
mock_build_graphs, mock_splice_find):
mock_splice_find_data = util.load_from_disk(
self.current_directory +
"/data/user_test/nick_change_graph/splice_find_list")
mock_build_graphs_data = util.load_from_disk(
self.current_directory + "/data/user_test/build_graphs_list")
mock_get_year_month_day_data = util.load_from_disk(
self.current_directory + "/data/user_test/get_year_month_day_list")
mock_splice_find.side_effect = mock_splice_find_data
mock_build_graphs.side_effect = mock_build_graphs_data
mock_get_year_month_day.side_effect = mock_get_year_month_day_data
expected_nick_change_graph_list = util.load_from_disk(
self.current_directory + "/data/user_test/nick_change_graph_list")
expected_aggregate_nick_change_graph = util.load_from_disk(
self.current_directory +
"/data/user_test/aggregate_nick_change_graph")
nick_change_graph_list = user.nick_change_graph(self.log_data, True)
mock_splice_find.side_effect = mock_splice_find_data
mock_build_graphs.side_effect = mock_build_graphs_data
mock_get_year_month_day.side_effect = mock_get_year_month_day_data
aggregate_nick_change_graph = user.nick_change_graph(
self.log_data, False)
for i in xrange(len(nick_change_graph_list)):
self.assertTrue(
nx.is_isomorphic(expected_nick_change_graph_list[i],
nick_change_graph_list[i]))
self.assertTrue(
nx.is_isomorphic(expected_aggregate_nick_change_graph,
aggregate_nick_change_graph))
def test_degree_analysis_on_nodes(self, log_data, nicks, nick_same_list):
update_expected_output_directory(log_data)
message_number_graph = network.message_number_graph(
log_data, nicks, nick_same_list)
message_time_graph = network.message_time_graph(
log_data, nicks, nick_same_list)
nick_change_graph = user.nick_change_graph(log_data)
degree_anal_message_number = network.degree_analysis_on_graph(
message_number_graph)
degree_anal_message_time = network.degree_analysis_on_graph(
message_time_graph)
degree_anal_nick_change = network.degree_analysis_on_graph(
nick_change_graph)
expected_analysis_msg_number = []
expected_analysis_msg_time = []
expected_analysis_nick_change = []
unjson('degree_anal_message_number.json', expected_analysis_msg_number)
unjson('degree_anal_message_time.json', expected_analysis_msg_time)
unjson('degree_anal_nick_change.json', expected_analysis_nick_change)
self.assertDictEqual(degree_anal_message_number,
expected_analysis_msg_number[0],
msg=None)
self.assertDictEqual(degree_anal_message_time,
expected_analysis_msg_time[0],
msg=None)
self.assertDictEqual(degree_anal_nick_change,
expected_analysis_nick_change[0],
msg=None)
def test_nick_change_graph(self):
expected_nick_change_graph_list = util.load_from_disk(
self.test_data_dir + "nick_change_graph_list")
expected_aggregate_nick_change_graph = util.load_from_disk(
self.test_data_dir + "aggregate_nick_change_graph")
nick_change_graph_list = user.nick_change_graph(self.log_data, True)
aggregate_nick_change_graph = user.nick_change_graph(
self.log_data, False)
for i in xrange(len(nick_change_graph_list)):
self.assertTrue(
nx.is_isomorphic(expected_nick_change_graph_list[i],
nick_change_graph_list[i]))
self.assertTrue(
nx.is_isomorphic(expected_aggregate_nick_change_graph,
aggregate_nick_change_graph))
def test_degree_analysis_on_nodes(self, log_data, nicks, nick_same_list):
update_expected_output_directory(log_data)
message_number_graph = network.message_number_graph(log_data, nicks, nick_same_list)
message_time_graph = network.message_time_graph(log_data, nicks, nick_same_list)
nick_change_graph = user.nick_change_graph(log_data)
degree_anal_message_number = network.degree_analysis_on_graph(message_number_graph)
degree_anal_message_time = network.degree_analysis_on_graph(message_time_graph)
degree_anal_nick_change = network.degree_analysis_on_graph(nick_change_graph)
expected_analysis_msg_number = []
expected_analysis_msg_time = []
expected_analysis_nick_change = []
unjson('degree_anal_message_number.json', expected_analysis_msg_number)
unjson('degree_anal_message_time.json', expected_analysis_msg_time)
unjson('degree_anal_nick_change.json', expected_analysis_nick_change)
self.assertDictEqual(degree_anal_message_number, expected_analysis_msg_number[0], msg=None)
self.assertDictEqual(degree_anal_message_time, expected_analysis_msg_time[0], msg=None)
self.assertDictEqual(degree_anal_nick_change, expected_analysis_nick_change[0], msg=None)
nicks, nick_same_list = nickTracker.nick_tracker(log_data)
# ============== ANALYSIS =============
message_number_graph = network.message_number_graph(log_data, nicks,
nick_same_list, False)
message_number_graph_day_list = network.message_number_graph(
log_data, nicks, nick_same_list, True)
degree_anal_message_numder = network.degree_analysis_on_graph(
message_number_graph)
message_time_graph_list = network.message_time_graph(log_data, nicks,
nick_same_list, True)
message_time_graph = network.message_time_graph(log_data, nicks,
nick_same_list, False)
out_degree_node_number, in_degree_node_number, total_degree_node_number = network.degree_node_number_csv(
log_data, nicks, nick_same_list)
nick_change_graph_list = user.nick_change_graph(log_data, True)
bin_matrix, total_messages = network.message_number_bins_csv(
log_data, nicks, nick_same_list)
conv_len, conv_ref_time = channel.conv_len_conv_refr_time(
log_data, nicks, nick_same_list)
resp_time = channel.response_time(log_data, nicks, nick_same_list)
user.keywords_clusters(log_data, nicks, nick_same_list)
network.degree_analysis_on_graph(message_number_graph)
# adjCC_graph, adjCC_membership = community.infomap_igraph(ig_graph=None, net_file_location="/home/rohan/Desktop/adjCC.net")
# ============== OUTPUT ================
saver.draw_nx_graph(message_number_graph, output_directory,
"message_number_graph")
saver.save_csv(degree_anal_message_numder["out_degree"]["formatted_for_csv"],
def test_nick_change_graph(self, log_data):
update_expected_output_directory(log_data)
nick_change_graph = user.nick_change_graph(log_data)
self.assertTrue(compare_graph_outputs(nick_change_graph,
"nick_change_graph.gpickle"),
msg=None)
starting_date = config.STARTING_DATE ending_date = config.ENDING_DATE output_directory = config.OUTPUT_DIRECTORY # ============== INPUT================== log_data = reader.linux_input(log_directory, channel_name, starting_date, ending_date) nicks, nick_same_list = nickTracker.nick_tracker(log_data) # ============== ANALYSIS ============= message_number_graph = network.message_number_graph(log_data, nicks, nick_same_list, False) message_number_graph_day_list = network.message_number_graph(log_data, nicks, nick_same_list, True) degree_anal_message_numder = network.degree_analysis_on_graph(message_number_graph) message_time_graph_list = network.message_time_graph(log_data, nicks, nick_same_list, True) message_time_graph = network.message_time_graph(log_data, nicks, nick_same_list, False) out_degree_node_number, in_degree_node_number, total_degree_node_number = network.degree_node_number_csv(log_data, nicks, nick_same_list) nick_change_graph_list = user.nick_change_graph(log_data, True) bin_matrix, total_messages = network.message_number_bins_csv(log_data, nicks, nick_same_list) conv_len, conv_ref_time = channel.conv_len_conv_refr_time(log_data, nicks, nick_same_list) resp_time = channel.response_time(log_data, nicks, nick_same_list) user.keywords_clusters(log_data, nicks, nick_same_list) network.degree_analysis_on_graph(message_number_graph) # adjCC_graph, adjCC_membership = community.infomap_igraph(ig_graph=None, net_file_location="/home/rohan/Desktop/adjCC.net") # ============== OUTPUT ================ saver.draw_nx_graph(message_number_graph, output_directory, "message_number_graph") saver.save_csv(degree_anal_message_numder["out_degree"]["formatted_for_csv"], output_directory, "out_degree") saver.save_csv(degree_anal_message_numder["in_degree"]["formatted_for_csv"], output_directory, "in_degree") saver.save_csv(degree_anal_message_numder["total_degree"]["formatted_for_csv"], output_directory, "total_degree") saver.save_csv(out_degree_node_number, output_directory, "node_out_degree" + starting_date +'-'+ending_date)
def test_nick_change_graph(self, log_data):
update_expected_output_directory(log_data)
nick_change_graph = user.nick_change_graph(log_data)
self.assertTrue(compare_graph_outputs(nick_change_graph, "nick_change_graph.gpickle"), msg=None)