diff --git a/tests/integration/test_writes/test_writes.py b/tests/integration/test_writes/test_writes.py index 837dba4dfe..6135a345bd 100644 --- 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}')