Skip to content

Commit 8dc720d

Browse files
committed
smoke test for whiten epsilon param.
1 parent dac882d commit 8dc720d

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/test_preprocess_pipeline.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,15 @@ def testWhiten2(dtype):
126126
assert np.allclose(np.eye(2), corr_coef, atol=2e-1)
127127

128128

129+
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
130+
def test_whiten_epsilon(dtype):
131+
"""Smoke test for epsilon argument"""
132+
L = 25
133+
sim = get_sim_object(L, dtype)
134+
noise_estimator = AnisotropicNoiseEstimator(sim)
135+
_ = sim.whiten(noise_estimator.filter, epsilon=0.01)
136+
137+
129138
@pytest.mark.parametrize("L, dtype", params)
130139
def testInvertContrast(L, dtype):
131140
sim1 = get_sim_object(L, dtype)

0 commit comments

Comments
 (0)