def to_file(filename, db_name=None):
assert isinstance(filename, str)
extension = filename.split(".")[-1]
table_dump = tables(db_helper.create_default_engine(db_name))
if extension == "pkl":
with open(filename, "wb") as f:
pickle.dump(table_dump, file=f)
elif extension == "json":
with open(filename, "w", encoding="utf8") as f:
json.dump(table_dump,
fp=f,
default=json_serial,
ensure_ascii=True,
indent=2)
else:
raise ValueError(
f"The file extension '.{extension}' is not valid. Use '.pkl' or '.json'"
)
def test_CreateDefaultEngine(self):
self.assertIsInstance(db_helper.create_default_engine(),
sqlalchemy.engine.base.Engine)
def test_Tables(self):
with open(os.path.join(dir_path, "data/base_tables.json"), "r") as f:
expected_tables = json.load(f)
actual_tables = db_info.get_table_names(db_helper.create_default_engine())
self.assertSetEqual(set(expected_tables), set(actual_tables), f"All base tables should exist")
def setUp(self):
self.db = db_helper.create_default_engine()
def test_UsersColumns(self):
with open(os.path.join(dir_path, "data/Users_columns.json"), "r") as f:
expected_columns = json.load(f)
actual_columns = db_info.get_column_names(db_helper.create_default_engine(), "Users")
self.assertSetEqual(set(expected_columns), set(actual_columns), f"All columns in the User table should exist")