From 1fbb6705abbfb0b32eb20f9b73489ccbe8613429 Mon Sep 17 00:00:00 2001 From: Niels Pardon Date: Wed, 3 Jun 2026 09:02:44 +0200 Subject: [PATCH] test: fix consistent partition window test for urn-based extension API The ConsistentPartitionWindowRel test was written against the older uri-based extension API and broke after the migration to urns: - register_extension_dict() no longer accepts a `uri` keyword; the URN is supplied via the `urn:` field already present in the test YAML. - Plan.extension_uris was renamed to extension_urns. Co-Authored-By: Claude Opus 4.8 (1M context) --- tests/builders/plan/test_consistent_partition_window.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/builders/plan/test_consistent_partition_window.py b/tests/builders/plan/test_consistent_partition_window.py index 212cb14..4b5658d 100644 --- a/tests/builders/plan/test_consistent_partition_window.py +++ b/tests/builders/plan/test_consistent_partition_window.py @@ -54,9 +54,7 @@ """ registry = ExtensionRegistry(load_default_extensions=False) -registry.register_extension_dict( - yaml.safe_load(content), uri="https://test.example.com/test.yaml" -) +registry.register_extension_dict(yaml.safe_load(content)) struct = stt.Type.Struct( types=[i64(nullable=False), i16(nullable=False), i32(nullable=False)], @@ -106,7 +104,6 @@ def test_consistent_partition_window_single(): expected = stp.Plan( version=default_version, extension_urns=list(lead_ee.extension_urns), - extension_uris=list(lead_ee.extension_uris), extensions=list(lead_ee.extensions), relations=[ stp.PlanRel(