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):