pybamm-team#3443 Fix expression tree Jax evaluator test

agriyakhetarpal · Nov 22, 2023 · 5fd45c6 · 5fd45c6
1 parent 30db13b
commit 5fd45c6
Showing 1 changed file with 1 addition and 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
@@ -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)