def test_message_time_graph_day_analysis(self):
capturedOutput = StringIO.StringIO()
sys.stdout = capturedOutput
ret = network.message_time_graph(self.log_data,
self.nicks,
self.nick_same_list,
DAY_BY_DAY_ANALYSIS=True)
expected_graph_list = util.load_from_disk(
self.test_data_dir + "message_time_graph/msg_time_graph_list")
sys.stdout = sys.__stdout__
capturedOutput.close()
self.assertTrue(nx.is_isomorphic(ret[0], expected_graph_list[0]))
def test_message_time_graph(self, mock_rec_list_splice,
mock_correctLastCharCR,
mock_get_year_month_day, mock_get_nick_sen_rec,
mock_check_if_msg_line,
mock_create_connected_nick_list,
mock_to_graph):
to_graph_ret = util.load_from_disk(current_directory +
"/data/message_time_graph/to_graph")
conn_list = list(connected_components(to_graph_ret))
mock_to_graph.return_value = to_graph_ret
mock_rec_list_splice.side_effect = util.load_from_disk(
current_directory + "/data/message_time_graph/rec_list_splice")
mock_create_connected_nick_list.return_value = util.load_from_disk(
current_directory + "/data/message_time_graph/conn_comp_list")
mock_check_if_msg_line.side_effect = util.load_from_disk(
current_directory + "/data/message_time_graph/check_if_msg_line")
mock_correctLastCharCR.side_effect = util.load_from_disk(
current_directory + "/data/message_time_graph/correctLastCharCR")
mock_get_nick_sen_rec.side_effect = util.load_from_disk(
current_directory + "/data/message_time_graph/get_nick_sen_rec")
#mock_correct_last_char_list.side_effect = util.load_from_disk(current_directory + "/data/message_time_graph/correct_last_char_list")
mock_get_year_month_day.side_effect = util.load_from_disk(
current_directory + "/data/message_time_graph/get_year_month_day")
capturedOutput = StringIO.StringIO()
sys.stdout = capturedOutput
ret = network.message_time_graph(self.log_data,
self.nicks,
self.nick_same_list,
DAY_BY_DAY_ANALYSIS=False)
sys.stdout = sys.__stdout__
capturedOutput.close()
mock_to_graph.assert_called_once_with(self.nick_same_list)
mock_create_connected_nick_list.assert_called_once_with(conn_list)
self.assertTrue(
nx.is_isomorphic(
ret,
util.load_from_disk(
current_directory +
"/data/message_time_graph/msg_time_aggr_graph")))
def test_message_time_graph(self):
capturedOutput = StringIO.StringIO()
sys.stdout = capturedOutput
ret = network.message_time_graph(self.log_data,
self.nicks,
self.nick_same_list,
DAY_BY_DAY_ANALYSIS=False)
sys.stdout = sys.__stdout__
capturedOutput.close()
self.assertTrue(
nx.is_isomorphic(
ret,
util.load_from_disk(self.test_data_dir +
"message_time_graph/msg_time_aggr_graph")))