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")