def test_extension(self):
dd = SQL(self.engine, 'testtable2',
schema='{name: string, amount: int32}',
primary_key='name')
data_list = [('Alice', 100), ('Bob', 50)]
data_dict = [{'name': name, 'amount': amount} for name, amount in data_list]
dd.extend(data_dict)
with self.engine.connect() as conn:
results = conn.execute('select * from testtable2')
self.assertEquals(list(results), data_list)
assert list(iter(dd)) == data_list or list(iter(dd)) == data_dict
assert (dd.as_py() == tuple(map(tuple, data_list)) or
dd.as_py() == data_dict)
def test_extension(self):
dd = SQL(self.engine, 'testtable2',
schema='{name: string, amount: int32}',
primary_key='name')
data_list = [('Alice', 100), ('Bob', 50)]
data_dict = [{'name': name, 'amount': amount} for name, amount in data_list]
dd.extend(data_dict)
with self.engine.connect() as conn:
results = conn.execute('select * from testtable2')
self.assertEquals(list(results), data_list)
assert list(iter(dd)) == data_list or list(iter(dd)) == data_dict
assert (dd.as_py() == tuple(map(tuple, data_list)) or
dd.as_py() == data_dict)
