diff --git a/tests/integration/test_solvers/test_external_variables.py b/tests/integration/test_solvers/test_external_variables.py index cd2a43abb9..f8f796284f 100644 --- a/tests/integration/test_solvers/test_external_variables.py +++ b/tests/integration/test_solvers/test_external_variables.py @@ -50,7 +50,7 @@ def test_external_variables_SPMe(self): t = sim.solution.t[-1] y = sim.solution.y[:, -1] u = external_variables - sim.built_model.variables[var].evaluate(t, y, u) + sim.built_model.variables[var].evaluate(t, y, u=u) sim.solution[var](t) diff --git a/tests/unit/test_expression_tree/test_operations/test_convert_to_casadi.py b/tests/unit/test_expression_tree/test_operations/test_convert_to_casadi.py index 017cae7f7a..9dbd48492e 100644 --- a/tests/unit/test_expression_tree/test_operations/test_convert_to_casadi.py +++ b/tests/unit/test_expression_tree/test_operations/test_convert_to_casadi.py @@ -154,6 +154,7 @@ def myfunction(x, y): def test_convert_input_parameter(self): casadi_t = casadi.MX.sym("t") casadi_y = casadi.MX.sym("y", 10) + casadi_ydot = casadi.MX.sym("ydot", 10) casadi_us = { "Input 1": casadi.MX.sym("Input 1"), "Input 2": casadi.MX.sym("Input 2"), @@ -165,18 +166,19 @@ def test_convert_input_parameter(self): # Input only self.assert_casadi_equal( - pybamm_u1.to_casadi(casadi_t, casadi_y, casadi_us), casadi_us["Input 1"] + pybamm_u1.to_casadi(casadi_t, casadi_y, casadi_ydot, casadi_us), + casadi_us["Input 1"] ) # More complex expr = pybamm_u1 + pybamm_y self.assert_casadi_equal( - expr.to_casadi(casadi_t, casadi_y, casadi_us), + expr.to_casadi(casadi_t, casadi_y, casadi_ydot, casadi_us), casadi_us["Input 1"] + casadi_y, ) expr = pybamm_u2 * pybamm_y self.assert_casadi_equal( - expr.to_casadi(casadi_t, casadi_y, casadi_us), + expr.to_casadi(casadi_t, casadi_y, casadi_ydot, casadi_us), casadi_us["Input 2"] * casadi_y, ) @@ -194,13 +196,13 @@ def test_convert_external_variable(self): # External only self.assert_casadi_equal( - pybamm_u1.to_casadi(casadi_t, casadi_y, casadi_us), casadi_us["External 1"] + pybamm_u1.to_casadi(casadi_t, casadi_y, u=casadi_us), casadi_us["External 1"] ) # More complex expr = pybamm_u2 + pybamm_y self.assert_casadi_equal( - expr.to_casadi(casadi_t, casadi_y, casadi_us), + expr.to_casadi(casadi_t, casadi_y, u=casadi_us), casadi_us["External 2"] + casadi_y, )