Replaced the bind with the connection method

This commit is contained in:
2024-01-15 20:24:38 +01:00
parent 3af9ff58d6
commit 27a95c1c23
7 changed files with 26 additions and 48 deletions

View File

@ -45,15 +45,11 @@ def destination_db() -> Generator[Session, None, None]:
def test_transfer_db(full_db: Session, destination_db: Session) -> None:
"""Tests if the data transfer between two sql tables works."""
transfer_db_function(source=full_db, destination=destination_db)
sbind = full_db.bind
dbind = destination_db.bind
assert isinstance(sbind, Engine)
assert isinstance(dbind, Engine)
assert Base.metadata.sorted_tables
for table in Base.metadata.sorted_tables + ["company"]:
pd.testing.assert_frame_equal(
pd.read_sql_table(str(table), dbind),
pd.read_sql_table(str(table), sbind),
pd.read_sql_table(str(table), destination_db.connection()),
pd.read_sql_table(str(table), full_db.connection()),
)