Permit B as a bytes unit in compaction_test

patch by Branimir Lambov; reviewed by Brandon Williams for CASSANDRA-18397
apache · Jul 19, 2023 · 6bffd00 · 6bffd00
1 parent 1da18ef
commit 6bffd00
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/compaction_test.py b/compaction_test.py
@@ -286,10 +286,11 @@ def test_compaction_throughput(self):
             "MB": 1,
             "MiB": 1,
             "KiB": 1. / 1024,
-            "GiB": 1024
+            "GiB": 1024,
+            "B": 1. / (1024 * 1024),
         }
 
-        units = ['MB'] if cluster.version() < LooseVersion('3.6') else ['KiB', 'MiB', 'GiB']
+        units = ['MB'] if cluster.version() < LooseVersion('3.6') else ['B', 'KiB', 'MiB', 'GiB']
         assert found_units in units
 
         logger.debug(avgthroughput)