From 42bab27ad3d125a389dd0408db36d3f76d8f19f9 Mon Sep 17 00:00:00 2001 From: Brady Planden Date: Fri, 4 Oct 2024 13:30:58 +0100 Subject: [PATCH] test: update test_optimisation for flaky CI --- tests/unit/test_optimisation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_optimisation.py b/tests/unit/test_optimisation.py index dd013618..a7200f81 100644 --- a/tests/unit/test_optimisation.py +++ b/tests/unit/test_optimisation.py @@ -18,7 +18,7 @@ def dataset(self): return pybop.Dataset( { "Time [s]": np.linspace(0, 360, 10), - "Current function [A]": 1e-3 * np.ones(10), + "Current function [A]": 1e-2 * np.ones(10), "Voltage [V]": np.ones(10), } ) @@ -134,12 +134,12 @@ def test_no_optimisation_parameters(self, model, dataset): ) @pytest.mark.unit def test_optimiser_kwargs(self, cost, optimiser): - optim = optimiser(cost=cost, maxiter=1, tol=1e-6) + optim = optimiser(cost=cost, maxiter=3, tol=1e-6) cost_bounds = cost.parameters.get_bounds() # Check maximum iterations results = optim.run() - assert results.n_iterations == 1 + assert results.n_iterations == 3 if optimiser in [pybop.GradientDescent, pybop.Adam, pybop.NelderMead]: # Ignored bounds