pybamm-team#2188 changelog and coverage

js1tr3 · Aug 12, 2024 · c20cb24 · c20cb24
1 parent 3e1389a
commit c20cb24
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tests/unit/test_expression_tree/test_operations/test_copy.py b/tests/unit/test_expression_tree/test_operations/test_copy.py
@@ -60,6 +60,7 @@ def test_symbol_new_copy(self):
             pybamm.maximum(a, b),
             pybamm.SparseStack(mat, mat),
             pybamm.Equality(a, b),
+            pybamm.EvaluateAt(a, 0),
         ]:
             self.assertEqual(symbol, symbol.new_copy())
 

diff --git a/tests/unit/test_expression_tree/test_operations/test_jac.py b/tests/unit/test_expression_tree/test_operations/test_jac.py
@@ -236,6 +236,12 @@ def test_index(self):
         jac = ind.jac(vec).evaluate(y=np.linspace(0, 2, 5)).toarray()
         np.testing.assert_array_equal(jac, np.array([[0, 0, 0, 0, 0]]))
 
+    def test_evluate_at(self):
+        y = pybamm.StateVector(slice(0, 4))
+        expr = pybamm.EvaluateAt(y, 2)
+        jac = expr.jac(y).evaluate(y=np.linspace(0, 2, 4))
+        np.testing.assert_array_equal(jac, 0)
+
     def test_jac_of_number(self):
         """Jacobian of a number should be zero"""
         a = pybamm.Scalar(1)