def client_and_prod(request, sqlite_client_and_tmp_dir, pg_client_and_schema):
# Based on: https://github.com/pytest-dev/pytest/issues/349#issue-88534390
if request.param == 'sqlite':
client, _ = sqlite_client_and_tmp_dir
product = SQLiteRelation((None, 'numbers', 'table'), client)
schema = None
else:
client, schema = pg_client_and_schema
product = PostgresRelation((schema, 'numbers', 'table'), client)
yield client, product, schema
product.delete()
def test_sqlite_product_delete(tmp_directory):
"""
>>> import tempfile
>>> tmp_directory = tempfile.mkdtemp()
"""
tmp = Path(tmp_directory)
conn = SQLAlchemyClient('sqlite:///{}'.format(tmp / "database.db"))
df = pd.DataFrame({'a': np.arange(0, 100), 'b': np.arange(100, 200)})
df.to_sql('numbers', conn.engine)
numbers = SQLiteRelation((None, 'numbers', 'table'), conn)
numbers.render({})
numbers.delete()
assert not numbers.exists()
