Skip to content

Commit 15ee1c1

Browse files
committed
[python] Fix flake8 E501 in table_query_auth_test predicate literals
1 parent 711d224 commit 15ee1c1

2 files changed

Lines changed: 18 additions & 3 deletions

File tree

paimon-python/pypaimon/read/table_read.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,9 @@ def _record_generator():
111111
reader = self._create_reader_for_split(split)
112112
try:
113113
if isinstance(reader, RecordBatchReader):
114+
blob_indices = getattr(reader, 'blob_field_indices', None)
115+
vector_indices = getattr(reader, 'vector_field_indices', None)
116+
file_io = self.table.file_io
114117
for arrow_batch in iter(reader.read_arrow_batch, None):
115118
has_rk = ROW_KIND_COLUMN in arrow_batch.schema.names
116119
if has_rk:
@@ -123,7 +126,11 @@ def _record_generator():
123126
arrow_batch.column(j)[row_idx].as_py()
124127
for j in data_cols
125128
)
126-
row = OffsetRow(row_tuple, 0, len(data_cols))
129+
row = OffsetRow(
130+
row_tuple, 0, len(data_cols),
131+
file_io=file_io,
132+
blob_field_indices=blob_indices,
133+
vector_field_indices=vector_indices)
127134
if has_rk:
128135
kind_str = arrow_batch.column(rk_idx)[row_idx].as_py()
129136
row.set_row_kind_byte(RowKind.from_string(kind_str).value)

paimon-python/pypaimon/tests/table_query_auth_test.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,12 @@ def test_convert_plan_preserves_snapshot_id(self):
310310

311311
split = MagicMock()
312312
original_plan = Plan([split], snapshot_id=42)
313+
predicate_json = (
314+
'{"kind":"LEAF","transform":{"name":"FIELD_REF",'
315+
'"fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'
316+
)
313317
auth_result = TableQueryAuthResult(
314-
filter=['{"kind":"LEAF","transform":{"name":"FIELD_REF","fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'],
318+
filter=[predicate_json],
315319
column_masking=None)
316320
converted = auth_result.convert_plan(original_plan)
317321
assert converted.snapshot_id == 42
@@ -323,8 +327,12 @@ def test_convert_plan_preserves_none_snapshot_id(self):
323327

324328
split = MagicMock()
325329
original_plan = Plan([split], snapshot_id=None)
330+
predicate_json = (
331+
'{"kind":"LEAF","transform":{"name":"FIELD_REF",'
332+
'"fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'
333+
)
326334
auth_result = TableQueryAuthResult(
327-
filter=['{"kind":"LEAF","transform":{"name":"FIELD_REF","fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'],
335+
filter=[predicate_json],
328336
column_masking=None)
329337
converted = auth_result.convert_plan(original_plan)
330338
assert converted.snapshot_id is None

0 commit comments

Comments
 (0)