Skip to content

Commit 6018d8f

Browse files
sichinagamtezzele
authored andcommitted
Shorten test time
1 parent afcd0ce commit 6018d8f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/test_bopdmd.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def test_eigs():
9696
bopdmd.fit(Z, t)
9797
np.testing.assert_allclose(sort_imag(bopdmd.eigs), expected_eigs)
9898

99-
bopdmd = BOPDMD(svd_rank=2, num_trials=100, trial_size=0.5)
99+
bopdmd = BOPDMD(svd_rank=2, num_trials=10, trial_size=0.8)
100100
bopdmd.fit(Z, t)
101101
np.testing.assert_allclose(sort_imag(bopdmd.eigs), expected_eigs)
102102

@@ -126,7 +126,7 @@ def test_A():
126126
bopdmd.fit(Z, t)
127127
np.testing.assert_allclose(bopdmd.A, expected_A)
128128

129-
bopdmd = BOPDMD(svd_rank=2, compute_A=True, num_trials=100, trial_size=0.5)
129+
bopdmd = BOPDMD(svd_rank=2, compute_A=True, num_trials=10, trial_size=0.8)
130130
bopdmd.fit(Z, t)
131131
np.testing.assert_allclose(bopdmd.A, expected_A)
132132

@@ -140,7 +140,7 @@ def test_reconstruction():
140140
bopdmd.fit(Z, t)
141141
np.testing.assert_allclose(bopdmd.reconstructed_data, Z, rtol=1e-5)
142142

143-
bopdmd = BOPDMD(svd_rank=2, num_trials=100, trial_size=0.5)
143+
bopdmd = BOPDMD(svd_rank=2, num_trials=10, trial_size=0.8)
144144
bopdmd.fit(Z, t)
145145
np.testing.assert_allclose(bopdmd.reconstructed_data, Z, rtol=1e-5)
146146

@@ -164,7 +164,7 @@ def test_forecast():
164164
bopdmd.fit(Z_uneven, t_uneven)
165165
np.testing.assert_allclose(bopdmd.forecast(t_long), Z_long, rtol=1e-2)
166166

167-
bopdmd = BOPDMD(svd_rank=2, num_trials=100, trial_size=0.5)
167+
bopdmd = BOPDMD(svd_rank=2, num_trials=10, trial_size=0.8)
168168
bopdmd.fit(Z, t)
169169
np.testing.assert_allclose(bopdmd.forecast(t_long)[0], Z_long, rtol=1e-2)
170170

@@ -347,7 +347,7 @@ def relative_error(x, x_true):
347347
optdmd.fit(Z_noisy, t)
348348
optdmd_error = relative_error(optdmd.A, expected_A)
349349

350-
test_trials = 20
350+
test_trials = 10
351351
bop_success = 0
352352
for _ in range(test_trials):
353353
bopdmd = BOPDMD(
@@ -361,4 +361,4 @@ def relative_error(x, x_true):
361361
bopdmd_error = relative_error(bopdmd.A, expected_A)
362362
bop_success += bopdmd_error < optdmd_error
363363

364-
assert bop_success >= 0.6 * test_trials
364+
assert bop_success > 0.5 * test_trials

0 commit comments

Comments
 (0)