class ITBuildGraphFromQueriesWithRealGrakn(GraphTestCase):
KEYSPACE = "it_build_graph_from_queries"
SCHEMA = ("define "
"person sub entity, has name, plays parent, plays child;"
"name sub attribute, value string;"
"parentship sub relation, relates parent, relates child;")
DATA = ('insert '
'$p isa person, has name "Bob";'
'$r(parent: $p, child: $p) isa parentship;')
def setUp(self):
self._keyspace = type(self).__name__.lower(
) # Use the name of this test class as the keyspace name
print(self._keyspace)
self._client = GraknClient(uri="localhost:48555")
def tearDown(self):
self._client.keyspaces().delete(self._keyspace)
self._client.close()
def test_graph_is_built_from_grakn_as_expected(self):
g1 = nx.MultiDiGraph()
g1.add_node('x')
g2 = nx.MultiDiGraph()
g2.add_node('x')
g2.add_node('n')
g2.add_edge('x', 'n', type='has')
g3 = nx.MultiDiGraph()
g3.add_node('x')
g3.add_node('r')
g3.add_node('y')
g3.add_edge('r', 'x', type='child')
g3.add_edge('r', 'y', type='parent')
query_sampler_variable_graph_tuples = [
('match $x isa person; get;', mock_sampler, g1),
('match $x isa person, has name $n; get;', mock_sampler, g2),
('match $x isa person; $r(child: $x, parent: $y); get;',
mock_sampler, g3),
# TODO Add functionality for loading schema at a later date
# ('match $x sub person; $x sub $type; get;', g4),
# ('match $x sub $y; get;', g5),
]
with self._client.session(keyspace=self._keyspace) as session:
with session.transaction().write() as tx:
tx.query(ITBuildGraphFromQueriesWithRealGrakn.SCHEMA)
tx.query(ITBuildGraphFromQueriesWithRealGrakn.DATA)
tx.commit()
with session.transaction().read() as tx:
combined_graph = build_graph_from_queries(
query_sampler_variable_graph_tuples, tx)
person_exp = build_thing(
next(tx.query('match $x isa person; get;')).get('x'), tx)
name_exp = build_thing(
next(tx.query('match $x isa name; get;')).get('x'), tx)
parentship_exp = build_thing(
next(tx.query('match $x isa parentship; get;')).get('x'),
tx)
expected_combined_graph = nx.MultiDiGraph()
expected_combined_graph.add_node(person_exp, type='person')
expected_combined_graph.add_node(name_exp,
type='name',
value_type='string',
value='Bob')
expected_combined_graph.add_node(parentship_exp, type='parentship')
expected_combined_graph.add_edge(parentship_exp,
person_exp,
type='child')
expected_combined_graph.add_edge(parentship_exp,
person_exp,
type='parent')
expected_combined_graph.add_edge(person_exp, name_exp, type='has')
self.assertGraphsEqual(expected_combined_graph, combined_graph)
class Test(unittest.TestCase):
def setUp(self):
self._client = GraknClient(uri="localhost:48555")
self._session = self._client.session(keyspace=keyspace_name)
with open('schemas/phone-calls-schema.gql', 'r') as schema:
define_query = schema.read()
with self._session.transaction().write() as transaction:
transaction.query(define_query)
transaction.commit()
print("Loaded the " + keyspace_name + " schema")
def test_csv_migration(self):
migrate_csv.build_phone_call_graph(migrate_csv.Inputs, data_path,
keyspace_name)
self.assert_migration_results()
def test_json_migration(self):
migrate_json.build_phone_call_graph(migrate_json.Inputs, data_path,
keyspace_name)
self.assert_migration_results()
def test_xml_migration(self):
migrate_xml.build_phone_call_graph(migrate_xml.Inputs, data_path,
keyspace_name)
self.assert_migration_results()
def test_queries(self):
queries.process_selection(0, keyspace_name)
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 assert_migration_results(self):
with self._session.transaction().read() as transaction:
number_of_people = transaction.query(
"match $x isa person; get $x; count;").next().number()
self.assertEqual(number_of_people, 30)
number_of_companies = transaction.query(
"match $x isa company; get $x; count;").next().number()
self.assertEqual(number_of_companies, 1)
number_of_contracts = transaction.query(
"match $x isa contract; get $x; count;").next().number()
self.assertEqual(number_of_contracts, 10)
number_of_calls = transaction.query(
"match $x isa call; get $x; count;").next().number()
self.assertEqual(number_of_calls, 200)
def tearDown(self):
self._client.keyspaces().delete(keyspace_name)
self._session.close()
self._client.close()
print("Deleted the " + keyspace_name + " keyspace")