From d80054d52ab63d3ebcdd0d0d156976ee62126859 Mon Sep 17 00:00:00 2001 From: Kevin Liu Date: Sat, 9 Mar 2024 08:49:01 -0800 Subject: [PATCH] set int property --- tests/integration/test_writes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/test_writes.py b/tests/integration/test_writes.py index e653bc868a..a3c603141b 100644 --- a/tests/integration/test_writes.py +++ b/tests/integration/test_writes.py @@ -404,7 +404,7 @@ def get_data_files_count(identifier: str) -> int: # writes multiple data files once target file size is overridden target_file_size = arrow_table_with_null.nbytes - tbl = tbl.transaction().set_properties({TableProperties.WRITE_TARGET_FILE_SIZE_BYTES: str(target_file_size)}).commit_transaction() + tbl = tbl.transaction().set_properties({TableProperties.WRITE_TARGET_FILE_SIZE_BYTES: target_file_size}).commit_transaction() assert str(target_file_size) == tbl.properties.get(TableProperties.WRITE_TARGET_FILE_SIZE_BYTES) assert target_file_size < bigger_arrow_tbl.nbytes tbl.overwrite(bigger_arrow_tbl) @@ -412,7 +412,7 @@ def get_data_files_count(identifier: str) -> int: # writes half the number of data files when target file size doubles target_file_size = arrow_table_with_null.nbytes * 2 - tbl = tbl.transaction().set_properties({TableProperties.WRITE_TARGET_FILE_SIZE_BYTES: str(target_file_size)}).commit_transaction() + tbl = tbl.transaction().set_properties({TableProperties.WRITE_TARGET_FILE_SIZE_BYTES: target_file_size}).commit_transaction() assert str(target_file_size) == tbl.properties.get(TableProperties.WRITE_TARGET_FILE_SIZE_BYTES) assert target_file_size < bigger_arrow_tbl.nbytes tbl.overwrite(bigger_arrow_tbl)