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)