From 9b84c1ec2dd6c7c9bbb3073aeca798e1a936f580 Mon Sep 17 00:00:00 2001 From: Brady Planden Date: Mon, 19 Aug 2024 14:53:38 +0100 Subject: [PATCH] tests: adds direct try-except coverage, w/ verbose --- tests/unit/test_problem.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/unit/test_problem.py b/tests/unit/test_problem.py index 8ab7efda..3724cf84 100644 --- a/tests/unit/test_problem.py +++ b/tests/unit/test_problem.py @@ -150,6 +150,11 @@ def test_fitting_problem(self, parameters, dataset, model, signal): # Test model.simulate with an initial state problem.evaluate(inputs=[1e-5, 1e-5]) + # Test try-except + problem.verbose = True + out = problem.evaluate(inputs=[0.0, 0.0]) + assert not np.isfinite(out["Voltage [V]"]) + # Test problem construction errors for bad_dataset in [ pybop.Dataset({"Time [s]": np.array([0])}), @@ -206,6 +211,7 @@ def test_fitting_problem_eis(self, parameters): assert problem.domain == "Frequency [Hz]" # Test try-except + problem.verbose = True out = problem.evaluate(inputs=[0.0, 0.0]) assert not np.isfinite(out["Impedance"])