Skip to content

Commit d8808a6

Browse files
committed
#858 improve coverage
1 parent d21f0b6 commit d8808a6

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/unit/test_expression_tree/test_operations/test_convert_to_casadi.py

+8
Original file line numberDiff line numberDiff line change
@@ -66,16 +66,24 @@ def test_convert_array_symbols(self):
6666

6767
casadi_t = casadi.MX.sym("t")
6868
casadi_y = casadi.MX.sym("y", 10)
69+
casadi_y_dot = casadi.MX.sym("y_dot", 10)
6970

7071
pybamm_t = pybamm.Time()
7172
pybamm_y = pybamm.StateVector(slice(0, 10))
73+
pybamm_y_dot = pybamm.StateVectorDot(slice(0, 10))
7274

7375
# Time
7476
self.assertEqual(pybamm_t.to_casadi(casadi_t, casadi_y), casadi_t)
7577

7678
# State Vector
7779
self.assert_casadi_equal(pybamm_y.to_casadi(casadi_t, casadi_y), casadi_y)
7880

81+
# State Vector Dot
82+
self.assert_casadi_equal(
83+
pybamm_y_dot.to_casadi(casadi_t, casadi_y, casadi_y_dot),
84+
casadi_y_dot
85+
)
86+
7987
def test_special_functions(self):
8088
a = pybamm.Array(np.array([1, 2, 3, 4, 5]))
8189
self.assert_casadi_equal(pybamm.max(a).to_casadi(), casadi.MX(5), evalf=True)

0 commit comments

Comments
 (0)