def initialize(self, db, tblname, schema=None, cols=None): self.table = database.initialize_table(db, tblname, schema=schema) self._columns = cols or [col.name for col in self.table.c]
def test_initialize_table(db): assert isinstance(database.initialize_table(db, TBL), sqlalchemy.sql.schema.Table) with pytest.raises(KeyError): database.initialize_table(db, 'does_not_exist')