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()