From d406231d9ef9a74fa72fe4c5ef88adf5c10ec485 Mon Sep 17 00:00:00 2001 From: "xiaohongbo.xhb" Date: Fri, 7 Aug 2026 07:56:35 -0700 Subject: [PATCH] [python] Stabilize deferred BLOB limit tests --- paimon-python/pypaimon/tests/deferred_blob_resolve_test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/paimon-python/pypaimon/tests/deferred_blob_resolve_test.py b/paimon-python/pypaimon/tests/deferred_blob_resolve_test.py index 2fbe0fb62751..61a184e1393b 100644 --- a/paimon-python/pypaimon/tests/deferred_blob_resolve_test.py +++ b/paimon-python/pypaimon/tests/deferred_blob_resolve_test.py @@ -279,10 +279,10 @@ def test_iterator_passes_remaining_limit_across_splits(self): ) counting_file_io = _BlobCountingFileIO(table.file_io) table.file_io = counting_file_io + splits = table.new_read_builder().new_scan().plan().splits() read_builder = table.new_read_builder().with_projection( ["sample_id", "payload", "score"] ).with_limit(2) - splits = read_builder.new_scan().plan().splits() rows = list(read_builder.new_read().to_iterator(splits)) @@ -305,7 +305,7 @@ def test_iterator_applies_limit_after_auth_filter(self): auth_result = _RejectScoreOneAuthResult() splits = [ QueryAuthSplit(split, auth_result) - for split in read_builder.new_scan().plan().splits() + for split in table.new_read_builder().new_scan().plan().splits() ] scores = [ @@ -313,7 +313,8 @@ def test_iterator_applies_limit_after_auth_filter(self): for row in read_builder.new_read().to_iterator(splits) ] - self.assertEqual([0, 2], scores) + self.assertEqual(2, len(scores)) + self.assertNotIn(1, scores) self.assertEqual(2, counting_file_io.blobs_fetched) def test_auth_blob_filter_keeps_eager_resolution(self):