From e99f28f4858bbb56631f1b9143722c54c3a84985 Mon Sep 17 00:00:00 2001 From: Andrea Dainese Date: Thu, 23 Apr 2026 12:57:42 +0200 Subject: [PATCH] Add bit/Kbit/Mbit/... to bandwidth._normalize_bw() --- netutils/bandwidth.py | 1 + tests/unit/test_bandwidth.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/netutils/bandwidth.py b/netutils/bandwidth.py index c7eded16..247a2649 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 d1afd8fd..18334413 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},