@@ -532,7 +532,7 @@ def test_model_step_ode_python(self):
532532 # Step again (return 5 points)
533533 step_sol_2 = solver .step (step_sol , model , dt , npts = 5 )
534534 np .testing .assert_array_equal (
535- step_sol_2 .t , np .concatenate ([ np . array ([0 ]), np . linspace ( dt , 2 * dt , 5 ) ])
535+ step_sol_2 .t , np .array ([0 , 1 , 1 + 1e-6 , 1.25 , 1.5 , 1.75 , 2 ])
536536 )
537537 np .testing .assert_allclose (step_sol_2 .y [0 ], np .exp (- 0.1 * step_sol_2 .t ))
538538
@@ -560,22 +560,22 @@ def test_model_step_dae_python(self):
560560 dt = 1
561561 step_sol = solver .step (None , model , dt )
562562 np .testing .assert_array_equal (step_sol .t , [0 , dt ])
563- np .testing .assert_allclose (step_sol .y [0 ], np .exp (0.1 * step_sol .t ))
564- np .testing .assert_allclose (step_sol .y [- 1 ], 2 * np .exp (0.1 * step_sol .t ))
563+ np .testing .assert_allclose (step_sol .y [0 , : ], np .exp (0.1 * step_sol .t ))
564+ np .testing .assert_allclose (step_sol .y [- 1 , : ], 2 * np .exp (0.1 * step_sol .t ))
565565
566566 # Step again (return 5 points)
567567 step_sol_2 = solver .step (step_sol , model , dt , npts = 5 )
568568 np .testing .assert_array_equal (
569- step_sol_2 .t , np .concatenate ([ np . array ([0 ]), np . linspace ( dt , 2 * dt , 5 ) ])
569+ step_sol_2 .t , np .array ([0 , 1 , 1 + 1e-6 , 1.25 , 1.5 , 1.75 , 2 ])
570570 )
571- np .testing .assert_allclose (step_sol_2 .y [0 ], np .exp (0.1 * step_sol_2 .t ))
572- np .testing .assert_allclose (step_sol_2 .y [- 1 ], 2 * np .exp (0.1 * step_sol_2 .t ))
571+ np .testing .assert_allclose (step_sol_2 .y [0 , : ], np .exp (0.1 * step_sol_2 .t ))
572+ np .testing .assert_allclose (step_sol_2 .y [- 1 , : ], 2 * np .exp (0.1 * step_sol_2 .t ))
573573
574574 # Check steps give same solution as solve
575575 t_eval = step_sol .t
576576 solution = solver .solve (model , t_eval )
577- np .testing .assert_allclose (solution .y [0 ], step_sol .y [0 , :])
578- np .testing .assert_allclose (solution .y [- 1 ], step_sol .y [- 1 , :])
577+ np .testing .assert_allclose (solution .y [0 , : ], step_sol .y [0 , :])
578+ np .testing .assert_allclose (solution .y [- 1 , : ], step_sol .y [- 1 , :])
579579
580580 def test_model_solver_ode_events_casadi (self ):
581581 # Create model
@@ -794,8 +794,8 @@ def test_model_step_nonsmooth_events(self):
794794 )
795795 var1_soln = (step_solution .t % a ) ** 2 / 2 + a ** 2 / 2 * (step_solution .t // a )
796796 var2_soln = 2 * var1_soln
797- np .testing .assert_array_almost_equal (step_solution .y [0 ], var1_soln , decimal = 5 )
798- np .testing .assert_array_almost_equal (step_solution .y [- 1 ], var2_soln , decimal = 5 )
797+ np .testing .assert_array_almost_equal (step_solution .y [0 ], var1_soln , decimal = 4 )
798+ np .testing .assert_array_almost_equal (step_solution .y [- 1 ], var2_soln , decimal = 4 )
799799
800800 def test_model_solver_dae_nonsmooth (self ):
801801 whole_cell = ["negative electrode" , "separator" , "positive electrode" ]
0 commit comments