From 31cba7c23c8d97c20f6271510fae343cd43d7ae0 Mon Sep 17 00:00:00 2001 From: zouxxyy Date: Mon, 28 Sep 2026 00:27:54 +0800 Subject: [PATCH] [python] Keep Python upsert streaming checks on the Python writer --- paimon-python/pypaimon/tests/table_upsert_by_key_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paimon-python/pypaimon/tests/table_upsert_by_key_test.py b/paimon-python/pypaimon/tests/table_upsert_by_key_test.py index 68d26e86e8db..8833b1fc8855 100644 --- a/paimon-python/pypaimon/tests/table_upsert_by_key_test.py +++ b/paimon-python/pypaimon/tests/table_upsert_by_key_test.py @@ -22,6 +22,7 @@ import pyarrow as pa import pyarrow.parquet as pq +import pytest from pypaimon.read.table_read import TableRead from pypaimon.table.special_fields import SpecialFields @@ -140,6 +141,7 @@ def test_invalid_row_group_size_fails_before_opening_output(self): _RowIdUpdateFileWriter(table, (), ['id']) output_stream.assert_not_called() + @pytest.mark.python_write @mock.patch.object(_RowIdUpdateFileWriter, '_ROW_GROUP_MAX_ROWS', 2) def test_partial_upsert_streams_original_file_group(self): schema = pa.schema([