refactor: make test robust against floating point

ing-bank · Jul 4, 2022 · 41ea553 · 41ea553
1 parent fbf7e41
commit 41ea553
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/popmon/base/test_pipeline.py b/tests/popmon/base/test_pipeline.py
@@ -98,7 +98,9 @@ def test_popmon_pipeline(test_pipeline):
         np.power(np.log(datastore["x"]), 2) * datastore["weights"]
     ) / np.sum(datastore["weights"])
 
-    assert test_pipeline.transform(datastore)["res"] == expected_result
+    np.testing.assert_array_almost_equal(
+        test_pipeline.transform(datastore)["res"], expected_result, decimal=12
+    )
 
 
 def test_pipeline_repr(test_pipeline):