2020import shutil
2121import tempfile
2222import unittest
23+ import random
2324import pandas as pd
2425import 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+
4045class 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