diff --git a/netutils/bandwidth.py b/netutils/bandwidth.py index c7eded1..247a264 100644 --- a/netutils/bandwidth.py +++ b/netutils/bandwidth.py @@ -37,6 +37,7 @@ def _get_bytes_mapping() -> t.Dict[str, t.Dict[str, int]]: def _normalize_bw(speed: str) -> str: + speed = speed.replace("bit", "b") per_second_mapping = { "b": "bps", "Kb": "Kbps", diff --git a/tests/unit/test_bandwidth.py b/tests/unit/test_bandwidth.py index d1afd8f..1833441 100644 --- a/tests/unit/test_bandwidth.py +++ b/tests/unit/test_bandwidth.py @@ -6,10 +6,13 @@ name_to_bits = [ {"sent": "10Mbps", "received": 10000000}, + {"sent": "10 Mbit", "received": 10000000}, {"sent": "10 Mbps", "received": 10000000}, {"sent": "1Gbps", "received": 1000000000}, + {"sent": "1 Gbit", "received": 1000000000}, {"sent": "1 Gbps", "received": 1000000000}, {"sent": "100Gbps", "received": 100000000000}, + {"sent": "100 Gbit", "received": 100000000000}, {"sent": "100 Gbps", "received": 100000000000}, {"sent": "100 Gb", "received": 100000000000}, {"sent": "10GBps", "received": 80000000000},