test: update plot2d for unbounded cost

pybop-team · Oct 3, 2024 · 929f72d · 929f72d
1 parent bbafc54
commit 929f72d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/unit/test_plots.py b/tests/unit/test_plots.py
@@ -105,10 +105,11 @@ def test_cost_plots(self, cost):
         pybop.plot2d(cost, gradient=True, steps=5)
 
         # Test without bounds
-        for param in cost.problem.parameters:
+        for param in cost.parameters:
             param.bounds = None
-        with pytest.raises(ValueError):
-            pybop.plot2d(cost, steps=5)
+        pybop.plot2d(cost, steps=5)
+
+        # Test with bounds
         pybop.plot2d(cost, bounds=np.array([[0.5, 0.8], [0.4, 0.7]]), steps=5)
 
     @pytest.fixture