def test_create_from_dataframe_invalid_pk(self): """Create a table from a dataframe with invalid primary key""" self.insert() data = self.tbl.select() Table.create( ':memory:', "Foo_2", data, primary_key='foo', verbose=True)
def test_list_tables(): """Check that the list of tables is correct""" Table.create(DBNAME, "foo", [('id', int)], verbose=True) Table.create(DBNAME, "bar", [('id', int)], verbose=True) tables = Table.list_tables(DBNAME, verbose=True) assert tables == ["foo", "bar"], tables assert Table.exists(DBNAME, 'foo', verbose=True) os.remove(DBNAME)
def test_create_from_dataframe_invalid_pk(self): """Create a table from a dataframe with invalid primary key""" self.insert() data = self.tbl.select() Table.create(':memory:', "Foo_2", data, primary_key='foo', verbose=True)
def setup(self): self.tbl = Table.create(':memory:', "Foo", self.dtypes, primary_key='id', autoincrement=True, verbose=True)
def test_create_from_dataframe(self): """Create a table from a dataframe""" self.insert() data = self.tbl.select() data.index.name = None tbl = Table.create(':memory:', "Foo_2", data, verbose=True, primary_key='id', autoincrement=True) self.check(self.idata, tbl.select())
def test_create_from_dicts(self): """Create a table from dictionaries""" cols = list(zip(*self.dtypes))[0] dicts = [dict([(cols[i], d[i]) for i in xrange(len(d))]) for d in self.idata] tbl = Table.create(':memory:', "Bar", dicts, verbose=True) self.check_index(self.idata, tbl.select()) for idx, col in enumerate(cols): self.check_data(self.idata[:, [idx]], tbl[col])
def test_create_from_dicts(self): """Create a table from dictionaries""" cols = list(zip(*self.dtypes))[0] dicts = [ dict([(cols[i], d[i]) for i in xrange(len(d))]) for d in self.idata ] tbl = Table.create(':memory:', "Bar", dicts, verbose=True) self.check_index(self.idata, tbl.select()) for idx, col in enumerate(cols): self.check_data(self.idata[:, [idx]], tbl[col])
def test_exists(): """Check that the table does exist""" Table.create(DBNAME, "foo", [('id', int)], verbose=True) assert Table.exists(DBNAME, 'foo', verbose=True) os.remove(DBNAME)
def test_not_exists(): """Check that the table does not exist (db does exist)""" tbl = Table.create(DBNAME, "foo", [('id', int)], verbose=True) tbl.drop() assert not Table.exists(DBNAME, 'foo', verbose=True)
def setup(self): self.tbl = Table.create( ':memory:', "Foo", self.dtypes, primary_key='id', autoincrement=True, verbose=True)
def test_create_from_dataframe(self): """Create a table from a dataframe""" self.insert() data = self.tbl.select() tbl = Table.create(':memory:', "Foo_2", data, verbose=True) self.check(self.idata, tbl.select())
def setup(self): self.tbl = Table.create(':memory:', "Foo", self.dtypes, verbose=True)
def setup(self): self.tbl = Table.create( ':memory:', "Foo", self.dtypes, verbose=True)