@@ -44,7 +44,8 @@ def __init__(self, *args, **kwargs):
4444 self .ref2 = delay_odeint (20. , self .eq1 , args = {'xdelay' : case2_delay }, state_delays = {'x' : case2_delay }, method = 'euler' )
4545
4646 @parameterized .named_parameters (
47- {'testcase_name' : f'constant_delay_{ name } ' , 'method' : name }
47+ {'testcase_name' : f'constant_delay_{ name } ' ,
48+ 'method' : name }
4849 for name in get_supported_methods ()
4950 )
5051 def test1 (self , method ):
@@ -54,8 +55,8 @@ def test1(self, method):
5455 case1 = delay_odeint (20. , self .eq1 , args = {'xdelay' : case1_delay }, state_delays = {'x' : case1_delay }, method = method )
5556 case2 = delay_odeint (20. , self .eq1 , args = {'xdelay' : case2_delay }, state_delays = {'x' : case2_delay }, method = method )
5657
57- self .assertTrue ((case1 . x - self .ref1 . x ).mean () < 1e-3 )
58- self .assertTrue ((case2 . x - self .ref2 . x ).mean () < 1e-3 )
58+ self .assertTrue ((case1 [ 'x' ] - self .ref1 [ 'x' ] ).mean () < 1e-3 )
59+ self .assertTrue ((case2 [ 'x' ] - self .ref2 [ 'x' ] ).mean () < 1e-3 )
5960
6061 # fig, axs = plt.subplots(2, 1)
6162 # fig.tight_layout(rect=[0, 0, 1, 0.95], pad=3.0)
@@ -90,7 +91,7 @@ def test1(self, method):
9091 case1 = delay_odeint (4. , self .eq , args = {'xdelay' : delay1 }, state_delays = {'x' : delay1 }, dt = 0.01 , method = method )
9192 case2 = delay_odeint (4. , self .eq , args = {'xdelay' : delay2 }, state_delays = {'x' : delay2 }, dt = 0.01 , method = method )
9293
93- self .assertTrue ((case1 . x - self .ref1 . x ).mean () < 1e-1 )
94- self .assertTrue ((case2 . x - self .ref2 . x ).mean () < 1e-1 )
94+ self .assertTrue ((case1 [ 'x' ] - self .ref1 [ 'x' ] ).mean () < 1e-1 )
95+ self .assertTrue ((case2 [ 'x' ] - self .ref2 [ 'x' ] ).mean () < 1e-1 )
9596
9697
0 commit comments