Avoid connectionless execution

coiled · Nov 13, 2024 · dbae472 · dbae472
1 parent c8f121f
commit dbae472
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dask_snowflake/tests/test_core.py b/dask_snowflake/tests/test_core.py
@@ -29,7 +29,8 @@ def table(connection_kwargs):
     yield name
 
     engine = create_engine(URL(**connection_kwargs))
-    engine.execute(f"DROP TABLE IF EXISTS {name}")
+    with engine.connect() as connection:
+        connection.execute(f"DROP TABLE IF EXISTS {name}")
 
 
 @pytest.fixture(scope="module")