Skip to content

Commit f77433a

Browse files
author
yuzelin
committed
fix
1 parent d67b81e commit f77433a

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

paimon_python_java/tests/test_preicates.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import shutil
2121
import tempfile
2222
import unittest
23+
import random
2324
import pandas as pd
2425
import pyarrow as pa
2526

@@ -37,6 +38,10 @@ def _check_filtered_result(read_builder, expected_df):
3738
actual_df.reset_index(drop=True), expected_df.reset_index(drop=True))
3839

3940

41+
def _random_format():
42+
return random.choice(['avro', 'parquet', 'orc'])
43+
44+
4045
class PredicateTest(unittest.TestCase):
4146

4247
@classmethod
@@ -53,9 +58,12 @@ def setUpClass(cls):
5358
('f0', pa.int64()),
5459
('f1', pa.string()),
5560
])
56-
catalog.create_table('default.test_append', Schema(pa_schema), False)
61+
catalog.create_table('default.test_append',
62+
Schema(pa_schema, options={'file.format': _random_format()}),
63+
False)
5764
catalog.create_table('default.test_pk',
58-
Schema(pa_schema, primary_keys=['f0'], options={'bucket': '1'}),
65+
Schema(pa_schema, primary_keys=['f0'],
66+
options={'bucket': '1', 'file.format': _random_format()}),
5967
False)
6068

6169
df = pd.DataFrame({
@@ -148,7 +156,9 @@ def testAllFieldTypesWithEqual(self):
148156
# bool
149157
('_boolean', pa.bool_())
150158
])
151-
self.catalog.create_table('default.test_all_field_types', Schema(pa_schema), False)
159+
self.catalog.create_table('default.test_all_field_types',
160+
Schema(pa_schema, options={'file.format': _random_format()}),
161+
False)
152162
table = self.catalog.get_table('default.test_all_field_types')
153163
write_builder = table.new_batch_write_builder()
154164
write = write_builder.new_write()

0 commit comments

Comments
 (0)