ObjectiveValueComponent: Fix failing test

5 years ago · 537e7f5764
parent b42187ee0b
commit 537e7f5764
1 changed files with 2 additions and 2 deletions
--- a/src/python/miplearn/components/tests/test_objective.py
+++ b/src/python/miplearn/components/tests/test_objective.py
@ -16,8 +16,8 @@ def test_usage():
    comp.fit(instances)
    assert instances[0].lower_bound == 1183.0
    assert instances[0].upper_bound == 1183.0
-    assert comp.predict(instances).tolist() == [[1183.0, 1183.0],
-                                                [1070.0, 1070.0]]
+    assert np.round(comp.predict(instances), 2).tolist() == [[1183.0, 1183.0],
+                                                             [1070.0, 1070.0]]


 def test_obj_evaluate():