@@ -94,11 +94,10 @@ def test_polydiff():
9494 assert (params1 ['degree' ], params1 ['window_size' ], params1 ['kernel' ]) == (6 , 50 , 'friedrichs' )
9595 assert (params2 ['degree' ], params2 ['window_size' ], params2 ['kernel' ]) == (3 , 10 , 'gaussian' )
9696
97- # This test runs in a reasonable amount of time locally but for some reason takes forever in CI
98- # def test_rtsdiff_with_irregular_step():
99- # t = np.arange(len(x))*dt; np.random.seed(7) # seed so the test can't randomly fail
100- # t_irreg = t + np.random.uniform(-dt/10, dt/10, *t.shape) # add jostle
101- # params1, val1 = optimize(rtsdiff, x, t, dxdt_truth=dxdt_truth)
102- # params2, val2 = optimize(rtsdiff, x, t_irreg, dxdt_truth=dxdt_truth)
103- # assert val2 < 1.15*val1 # optimization works and comes out similar, since jostle is small
104- # assert params1['qr_ratio']*0.85 < params2['qr_ratio'] < params1['qr_ratio']*1.15
97+ def test_rtsdiff_with_irregular_step ():
98+ t = np .arange (len (x ))* dt ; np .random .seed (0 ) # seed so the test can't randomly fail
99+ t_irreg = t + np .random .uniform (- dt / 10 , dt / 10 , * t .shape ) # add jostle
100+ params1 , val1 = optimize (rtsdiff , x , t , dxdt_truth = dxdt_truth )
101+ params2 , val2 = optimize (rtsdiff , x , t_irreg , dxdt_truth = dxdt_truth )
102+ assert val2 < 1.2 * val1 # optimization works and comes out similar, since jostle is small
103+ assert params1 ['qr_ratio' ]* 0.8 < params2 ['qr_ratio' ] < params1 ['qr_ratio' ]* 1.2
0 commit comments