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_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_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_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_not_exists_no_db():
"""Check that the table does not exist (db does not exist)"""
if os.path.exists(DBNAME):
os.remove(DBNAME)
assert not Table.exists(DBNAME, 'foo', verbose=True)
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 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)
