diff --git a/tests/test_utils_tqdm_mod.py b/tests/test_utils_tqdm_mod.py index ed5025797..0b4e9368b 100644 --- a/tests/test_utils_tqdm_mod.py +++ b/tests/test_utils_tqdm_mod.py @@ -32,7 +32,7 @@ def test_complete_progress_bar(): bar = get_bar_visual(last) assert last.startswith("TestBar: 100%") - assert "3.00/3.00" in last + assert "3/3" in last assert "Hz" in last # Check that the bar us full @@ -57,7 +57,7 @@ def test_set_progress_multiple_points(): ("70%", "3.50/5.00"), ("100%", "5.00/5.00"), ]): - assert lines[i].startswith(f"SetBar:{expected_progress}") + assert re.search(r"^SetBar:\s+" + re.escape(expected_progress), lines[i]) assert expected_value in lines[i] bar = get_bar_visual(lines[i]) assert len(bar.strip()) > 0 @@ -72,7 +72,7 @@ def test_custom_unit(): bar = get_bar_visual(last) assert last.startswith("StepBar: 100%") - assert "4.00/4.00" in last + assert "4/4" in last assert "step/s" in last assert len(bar.strip()) > 0 @@ -87,7 +87,7 @@ def test_clamp_above_total(): bar = get_bar_visual(last) assert last.startswith("ClampBar: 100%") - assert "10.0/10.0" in last + assert "10/10" in last assert "Hz" in last # Check that the bar us full