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")))