def setUpClass(cls):
with GraknClient(uri="localhost:48555") as client:
with client.session(keyspace="tube_network") as session:
with open('schemas/tube-network-schema.gql', 'r') as schema:
define_query = schema.read()
with session.transaction().write() as transaction:
transaction.query(define_query)
transaction.commit()
print("Loaded the tube_network schema")
migration.init()
print("Loaded the tube_network data")
def setUpClass(cls):
with TypeDB.core_client("localhost:1729") as client:
client.databases().create("tube_network")
with client.session("tube_network", SessionType.SCHEMA) as session:
with open('schemas/tube-network-schema.gql', 'r') as schema:
define_query = schema.read()
with session.transaction(
TransactionType.WRITE) as transaction:
transaction.query().define(define_query)
transaction.commit()
print("Loaded the tube_network schema")
migration.init()
print("Loaded the tube_network data")
def test_migration(self):
migration.init()
self.assert_migration_results()