use con

Author: kevinjqliu

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}')