test(monitor): Do not perform equality checks with floating point val…

…ues.
jacksonpradolima · Dec 22, 2024 · 647c27d · 647c27d
1 parent 1d539c5
commit 647c27d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/coleman4hcs/tests/utils/test_monitor.py b/coleman4hcs/tests/utils/test_monitor.py
@@ -66,8 +66,9 @@ def test_collect_single_record(monitor_collector, mock_scenario_provider, mock_m
     assert record['scenario'] == "TestScenario"
     assert record['experiment'] == 1
     assert record['policy'] == "TestPolicy"
-    assert record['fitness'] == 0.8
-    assert record['cost'] == 0.6
+    # Use pytest.approx for floating-point comparisons
+    assert record['fitness'] == pytest.approx(0.8, rel=1e-6)
+    assert record['cost'] == pytest.approx(0.6, rel=1e-6)
 
 
 def test_collect_from_temp(monitor_collector, mock_scenario_provider, mock_metric):