use con

apache · Sep 6, 2024 · d93bbaa · d93bbaa
1 parent aa1d4ba
commit d93bbaa
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/integration/test_writes/test_writes.py b/tests/integration/test_writes/test_writes.py
@@ -649,11 +649,12 @@ def test_duckdb_url_import(warehouse: Path, arrow_table_with_null: pa.Table) ->
 
     import duckdb
 
+    con = duckdb.connect(database=":memory:")
     query = "SELECT * FROM duckdb_extensions() WHERE extension_name = 'iceberg'"
-    print(f"Before: {duckdb.sql(query)}")
-    duckdb.sql("INSTALL iceberg; LOAD iceberg;")
-    print(f"After: {duckdb.sql(query)}")
-    result = duckdb.sql(
+    print(f"Before: \n{con.sql(query)}")
+    con.sql("INSTALL iceberg; LOAD iceberg;")
+    print(f"After: \n{con.sql(query)}")
+    result = con.sql(
         f"""
     SELECT *
     FROM iceberg_scan('{location}')