set int property

apache · Mar 9, 2024 · d80054d · d80054d
1 parent c6cb8de
commit d80054d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/test_writes.py b/tests/integration/test_writes.py
@@ -404,15 +404,15 @@ 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)
     assert get_data_files_count(identifier) == 10
 
     # 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)