def test_sqltable_add_column(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db['test'].add_column('a')
db['test'].add_column('b')
db['test'].add_column('c', data=[1, 2, 3])
assert_equal(db.get_column('test', 'a').values, [None, None, None])
assert_equal(db.get_column('test', 'b').values, [None, None, None])
assert_equal(db.get_column('test', 'c').values, [1, 2, 3])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
assert_equal(db.column_names('test'), ['a', 'b', 'c'])
assert_equal(len(db['test']), 3)
def test_sql_add_table_columns(self):
db = SQLDatabase(':memory:')
db.add_table('test', columns=['a', 'b'])
assert_equal(db.get_column('test', 'a').values, [])
assert_equal(db.get_column('test', 'b').values, [])
assert_equal(db.column_names('test'), ['a', 'b'])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
def test_sql_add_table_from_data_dict(self):
d = {'a': np.arange(10, 20), 'b': np.arange(20, 30)}
db = SQLDatabase(':memory:')
db.add_table('test', data=d)
assert_equal(db.get_column('test', 'a').values, np.arange(10, 20))
assert_equal(db.get_column('test', 'b').values, np.arange(20, 30))
assert_equal(db.column_names('test'), ['a', 'b'])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
def test_sql_add_table_from_data_ndarray(self):
dtype = [('a', 'i4'), ('b', 'f8')]
data = np.array([(1, 2.0), (3, 4.0), (5, 6.0), (7, 8.0)], dtype=dtype)
db = SQLDatabase(':memory:')
db.add_table('test', data=data)
assert_equal(db.get_column('test', 'a').values, [1, 3, 5, 7])
assert_equal(db.get_column('test', 'b').values, [2.0, 4.0, 6.0, 8.0])
assert_equal(db.column_names('test'), ['a', 'b'])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
def test_sql_add_column_name_and_data(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a', data=np.arange(10, 20))
db.add_column('test', 'b', data=np.arange(20, 30))
assert_equal(db.get_column('test', 'a').values, np.arange(10, 20))
assert_equal(db.get_column('test', 'b').values, np.arange(20, 30))
assert_equal(db.column_names('test'), ['a', 'b'])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
def test_sql_set_item(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a')
db.add_column('test', 'b')
db.add_row('test', dict(a=1, b=2))
db.add_row('test', dict(a=3, b=4))
db.add_row('test', dict(a=5, b=6))
db.set_item('test', 'a', 0, 10)
db.set_item('test', 'b', 1, 'a')
assert_equal(db.get_column('test', 'a').values, [10, 3, 5])
assert_equal(db.get_column('test', 'b').values, [2, 'a', 6])
def test_sql_add_row(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a')
db.add_column('test', 'b')
db.add_row('test', dict(a=1, b=2))
db.add_row('test', dict(a=3, b=4))
db.add_row('test', dict(a=5, b=6))
assert_equal(db.get_column('test', 'a').values, [1, 3, 5])
assert_equal(db.get_column('test', 'b').values, [2, 4, 6])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
def test_sql_get_column(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a', data=np.arange(10, 20))
db.add_column('test', 'b', data=np.arange(20, 30))
assert_equal(db.get_column('test', 'a').values, np.arange(10, 20))
assert_equal(db.get_column('test', 'b').values, np.arange(20, 30))
assert_is_instance(db.get_column('test', 'a'), SQLColumn)
assert_is_instance(db.get_column('test', 'b'), SQLColumn)
# same access from table
assert_equal(
db.get_table('test').get_column('a').values, np.arange(10, 20))
assert_equal(
db.get_table('test').get_column('b').values, np.arange(20, 30))
assert_is_instance(db.get_table('test').get_column('a'), SQLColumn)
assert_is_instance(db.get_table('test').get_column('b'), SQLColumn)
with pytest.raises(KeyError):
db.get_column('test', 'c')
with pytest.raises(KeyError):
db.get_table('test').get_column('c')
def test_sqltable_add_row_add_columns(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a')
db.add_column('test', 'b')
db['test'].add_row(dict(a=1, b=2))
db['test'].add_row(dict(a=3, c=4), add_columns=False)
db['test'].add_row(dict(a=5, d=6), add_columns=True)
assert_equal(db.get_column('test', 'a').values, [1, 3, 5])
assert_equal(db.get_column('test', 'b').values, [2, None, None])
assert_equal(len(db), 1)
assert_equal(db.table_names, ['test'])
assert_equal(db.column_names('test'), ['a', 'b', 'd'])
def test_sql_set_column(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a')
db.add_column('test', 'b')
db.add_row('test', dict(a=1, b=2))
db.add_row('test', dict(a=3, b=4))
db.add_row('test', dict(a=5, b=6))
db.set_column('test', 'a', [10, 20, 30])
db.set_column('test', 'b', [20, 40, 60])
assert_equal(db.get_column('test', 'a').values, [10, 20, 30])
assert_equal(db.get_column('test', 'b').values, [20, 40, 60])
with pytest.raises(KeyError):
db.set_column('test', 'c', [10, 20, 30])
with pytest.raises(ValueError):
db.set_column('test', 'a', [10, 20, 30, 40])
def test_sql_set_row(self):
db = SQLDatabase(':memory:')
db.add_table('test')
db.add_column('test', 'a')
db.add_column('test', 'b')
db.add_row('test', dict(a=1, b=2))
db.add_row('test', dict(a=3, b=4))
db.add_row('test', dict(a=5, b=6))
db.set_row('test', 0, dict(a=10, b=20))
db.set_row('test', 1, dict(a=20, b=40))
db.set_row('test', 2, dict(a=30, b=60))
assert_equal(db.get_column('test', 'a').values, [10, 20, 30])
assert_equal(db.get_column('test', 'b').values, [20, 40, 60])
with pytest.raises(IndexError):
db.set_row('test', 3, dict(a=10, b=20))
with pytest.raises(IndexError):
db.set_row('test', -4, dict(a=10, b=20))