Casts the size of a MemoryRecordsBuilder object

dpkp · Jun 17, 2024 · 006effb · 006effb
1 parent a6d0579
commit 006effb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/record/test_records.py b/test/record/test_records.py
@@ -198,7 +198,7 @@ def test_memory_records_builder(magic, compression_type):
     # Size should remain the same after closing. No trailing bytes
     builder.close()
     assert builder.compression_rate() > 0
-    expected_size = size_before_close * builder.compression_rate()
+    expected_size = int(size_before_close * builder.compression_rate())
     assert builder.is_full()
     assert builder.size_in_bytes() == expected_size
     buffer = builder.buffer()