@@ -51,17 +51,14 @@ def test_integrate(self):
51
51
self .assertEqual (solution .termination , "final time" )
52
52
53
53
def test_integrate_failure (self ):
54
- # Turn off warnings to ignore sqrt error
55
- warnings .simplefilter ("ignore" )
56
-
57
54
t = casadi .MX .sym ("t" )
58
55
y = casadi .MX .sym ("y" )
59
56
u = casadi .MX .sym ("u" )
60
57
sqrt_decay = - np .sqrt (y )
61
58
62
59
y0 = np .array ([1 ])
63
60
t_eval = np .linspace (0 , 3 , 100 )
64
- solver = pybamm .CasadiSolver ()
61
+ solver = pybamm .CasadiSolver (regularity_check = False )
65
62
rhs = casadi .Function ("rhs" , [t , y , u ], [sqrt_decay ])
66
63
# Expect solver to fail when y goes negative
67
64
with self .assertRaises (pybamm .SolverError ):
@@ -84,21 +81,16 @@ def test_integrate_failure(self):
84
81
disc = pybamm .Discretisation (mesh , spatial_methods )
85
82
disc .process_model (model )
86
83
# Solve with failure at t=2
87
- solver = pybamm .CasadiSolver (rtol = 1e-8 , atol = 1e-8 , method = "idas" )
88
84
t_eval = np .linspace (0 , 20 , 100 )
89
85
with self .assertRaises (pybamm .SolverError ):
90
86
solver .solve (model , t_eval )
91
87
# Solve with failure at t=0
92
88
model .initial_conditions = {var : 0 }
93
89
disc .process_model (model )
94
- solver = pybamm .CasadiSolver (rtol = 1e-8 , atol = 1e-8 , method = "idas" )
95
90
t_eval = np .linspace (0 , 20 , 100 )
96
91
with self .assertRaises (pybamm .SolverError ):
97
92
solver .solve (model , t_eval )
98
93
99
- # Turn warnings back on
100
- warnings .simplefilter ("default" )
101
-
102
94
def test_bad_mode (self ):
103
95
with self .assertRaisesRegex (ValueError , "invalid mode" ):
104
96
pybamm .CasadiSolver (mode = "bad mode" )
0 commit comments