From 6bffd00ea003677a590b8d6c8dee39a548a20cb9 Mon Sep 17 00:00:00 2001
From: Branimir Lambov <branimir.lambov@datastax.com>
Date: Tue, 18 Jul 2023 14:34:59 +0300
Subject: [PATCH] Permit B as a bytes unit in compaction_test

patch by Branimir Lambov; reviewed by Brandon Williams for CASSANDRA-18397
---
 compaction_test.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/compaction_test.py b/compaction_test.py
index f7b27078fd..c5d90560d9 100644
--- 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)