def shipit(uri, schema=None):
args = parse_args()
with tempfile.NamedTemporaryFile() as f:
write_graph(
to_graph(relate(sa.create_engine(args.uri), schema=args.schema)),
f.name)
yield f.name
def shipit(uri, schema=None):
args = parse_args()
with tempfile.NamedTemporaryFile() as f:
write_graph(to_graph(
relate(sa.create_engine(args.uri), schema=args.schema)
), f.name)
yield f.name
def test_table(engine):
t = sa.Table('customer', sa.MetaData(bind=engine), autoload=True)
assert relate(t) == {
'customer': {
'primary': ['id'],
'foreign': {
'nation_id': {
'nation': 'id'
}
}
}
}
def test_to_graph(engine):
assert to_graph(relate(engine)).edges()
def test_database(engine):
assert relate(engine) == {
'customer': {
'primary': ['id'],
'foreign': {
'nation_id': {
'nation': 'id'
}
}
},
'lineitem': {
'primary': ['order_id', 'part_id', 'supplier_id', 'line_number'],
'foreign': {
'order_id': {
'orders': 'id',
},
'part_id': {
'part': 'id'
},
'supplier_id': {
'supplier': 'id'
}
}
},
'nation': {
'primary': ['id'],
'foreign': {
'region_id': {
'region': 'id'
}
}
},
'orders': {
'primary': ['id'],
'foreign': {
'customer_id': {
'customer': 'id'
}
}
},
'part': {
'primary': ['id'],
'foreign': {}
},
'partsupp': {
'primary': ['part_id', 'supplier_id'],
'foreign': {
'part_id': {
'part': 'id'
},
'supplier_id': {
'supplier': 'id'
}
}
},
'region': {
'primary': ['id'],
'foreign': {}
},
'supplier': {
'primary': ['id'],
'foreign': {
'nation_id': {
'nation': 'id'
}
}
}
}
def test_write_agraph(engine, tmpdir):
filename = str(tmpdir.join('tmp.dot'))
write_graph(to_graph(relate(engine)), filename)
assert os.path.exists(filename)
