def test_queries(self):
migrate_csv.build_phone_call_graph(migrate_csv.Inputs, data_path, keyspace_name)
with self._session.transaction().read() as transaction:
first_actual_answer = queries.query_examples[0].get("query_function")("", transaction)
first_expected_answer = [u"+370 351 224 5176", u"+54 398 559 0423", u"+62 107 530 7500", u"+63 815 962 6097",
u"+7 690 597 4443", u"+263 498 495 0617", u"+81 308 988 7153", u"+81 746 154 2598"]
self.assertItemsEqual(first_actual_answer, first_expected_answer)
second_actual_answer = queries.query_examples[1].get("query_function")("", transaction)
second_expected_answer = [u"+351 272 414 6570", u"+30 419 575 7546", u"+1 254 875 4647", u"+86 892 682 0628",
u"+33 614 339 0298", u"+351 515 605 7915", u"+86 922 760 0418", u"+63 808 497 1769",
u"+86 825 153 5518", u"+48 894 777 5173", u"+27 117 258 4149", u"+86 202 257 8619"]
self.assertItemsEqual(second_actual_answer, second_expected_answer)
third_actual_answer = queries.query_examples[2].get("query_function")("", transaction)
third_expected_answer = [u"+86 892 682 0628", u"+54 398 559 0423"]
self.assertItemsEqual(third_actual_answer, third_expected_answer)
forth_actual_answer = queries.query_examples[3].get("query_function")("", transaction)
forth_expected_answer = [u"+261 860 539 4754", u"+81 308 988 7153", u"+62 107 530 7500", u"+261 860 539 4754",
u"+81 308 988 7153", u"+62 107 530 7500"]
self.assertItemsEqual(forth_actual_answer, forth_expected_answer)
fifth_actual_answer = queries.query_examples[4].get("query_function")("", transaction)
fifth_expected_answer = [1242.7714285714285, 1699.4308943089432]
self.assertItemsEqual(fifth_actual_answer, fifth_expected_answer)
def test_csv_migration(self):
migrate_csv.build_phone_call_graph(migrate_csv.Inputs, data_path,
database_name)
self.assert_migration_results()
