From 1930e8a9e51ee6b2dfe4bfaee3e66b739ac05301 Mon Sep 17 00:00:00 2001 From: Agriya Khetarpal <74401230+agriyakhetarpal@users.noreply.github.com> Date: Thu, 23 Nov 2023 00:52:08 +0530 Subject: [PATCH] #3443 Fix expression tree Jax evaluator test --- .../test_operations/test_evaluate_python.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_expression_tree/test_operations/test_evaluate_python.py b/tests/unit/test_expression_tree/test_operations/test_evaluate_python.py index ca36804ba0..df33e0fe27 100644 --- a/tests/unit/test_expression_tree/test_operations/test_evaluate_python.py +++ b/tests/unit/test_expression_tree/test_operations/test_evaluate_python.py @@ -503,7 +503,7 @@ def test_evaluator_jax(self): expr = pybamm.exp(a * b) evaluator = pybamm.EvaluatorJax(expr) result = evaluator(t=None, y=np.array([[2], [3]])) - self.assertEqual(result, np.exp(6)) + np.testing.assert_array_almost_equal(result, np.exp(6), decimal=15) # test a constant expression expr = pybamm.Scalar(2) * pybamm.Scalar(3)