chore: relax duckdb pin to 1.5.x

examples/duckdb_example.py

@@ -2,7 +2,7 @@
 # /// script
 # dependencies = [
 #   "pyarrow==20.0.0",
-#   "duckdb==1.2.1",
+#   "duckdb==1.5.6",
 #   "substrait[extensions] @ file:///${PROJECT_ROOT}/"
 # ]
 # ///
@@ -18,7 +18,7 @@
 
 try:
     duckdb.install_extension("substrait")
-except duckdb.duckdb.HTTPException:
+except duckdb.HTTPException:
     duckdb.install_extension("substrait", repository="community")
 duckdb.load_extension("substrait")
 
@@ -31,7 +31,7 @@
 
 
 def read_duckdb_named_table(name: str, conn):
-    pa_schema = conn.sql(f"SELECT * FROM {name} LIMIT 0").arrow().schema
+    pa_schema = conn.sql(f"SELECT * FROM {name} LIMIT 0").to_arrow_table().schema
     substrait_schema = (
         pa_substrait.serialize_schema(pa_schema).to_pysubstrait().base_schema
     )

pyproject.toml

@@ -20,7 +20,7 @@ extensions = ["substrait-antlr==0.106.0", "pyyaml"]
 sql = ["sqloxide", "deepdiff"]
 
 [dependency-groups]
-dev = ["pytest >= 7.0.0", "substrait-antlr==0.106.0", "pyyaml", "sqloxide", "deepdiff", "duckdb<=1.2.2; python_version < '3.14'", "datafusion"]
+dev = ["pytest >= 7.0.0", "substrait-antlr==0.106.0", "pyyaml", "sqloxide", "deepdiff", "duckdb>=1.5,<1.6", "datafusion"]
 
 [tool.pytest.ini_options]
 pythonpath = "src"

tests/integration/test_sql_engine_roundtrip.py

@@ -99,11 +99,13 @@ def assert_query_duckdb(query: str, ignore_order=True):
         conn = duckdb.connect(db)
 
         def duckdb_schema_resolver(name: str):
-            pa_schema = conn.sql(f"SELECT * FROM {name} LIMIT 0").arrow().schema
+            pa_schema = (
+                conn.sql(f"SELECT * FROM {name} LIMIT 0").to_arrow_table().schema
+            )
             return pa_substrait.serialize_schema(pa_schema).to_pysubstrait().base_schema
 
-        conn.register("stores", data)
-        conn.register("sales", sales_data)
+        conn.from_arrow(data).create("stores")
+        conn.from_arrow(sales_data).create("sales")
 
         plan = convert(query, "duckdb", duckdb_schema_resolver, registry)
 
@@ -115,8 +117,8 @@ def duckdb_schema_resolver(name: str):
 
         sql_out = conn.sql(query)
 
-        substrait_arrow = substrait_out.arrow()
-        sql_arrow = sql_out.arrow()
+        substrait_arrow = substrait_out.to_arrow_table()
+        sql_arrow = sql_out.to_arrow_table()
 
         if ignore_order:
             substrait_arrow = sort_arrow(substrait_arrow)
@@ -283,12 +285,12 @@ def test_order_by(engine: str):
     )
 
 
-@pytest.mark.parametrize("engine", engines_duckdb_xfail)
+@pytest.mark.parametrize("engine", engines)
 def test_select_limit(engine: str):
     assert_query("""SELECT store_id FROM stores ORDER BY store_id LIMIT 2""", engine)
 
 
-@pytest.mark.parametrize("engine", engines_duckdb_xfail)
+@pytest.mark.parametrize("engine", engines)
 def test_select_limit_offset(engine: str):
     assert_query(
         """SELECT store_id FROM stores ORDER BY store_id LIMIT 2 OFFSET 2""", engine