#858 fix test failures

pybamm-team · Mar 15, 2020 · 6da78cb · 6da78cb
1 parent f3747fa
commit 6da78cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git 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
@@ -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,
         )