Skip to content

Commit 7517d06

Browse files
committed
fix scaling seed
1 parent 6ee7299 commit 7517d06

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

src/spikeinterface/preprocessing/tests/test_scaling.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,11 @@ def test_scaling_in_preprocessing_chain():
5555
recording.set_channel_gains(gains)
5656
recording.set_channel_offsets(offsets)
5757

58-
centered_recording = CenterRecording(scale_to_uV(recording=recording))
58+
centered_recording = CenterRecording(scale_to_uV(recording=recording), seed=2205)
5959
traces_scaled_with_argument = centered_recording.get_traces(return_scaled=True)
6060

6161
# Chain preprocessors
62-
centered_recording_scaled = CenterRecording(scale_to_uV(recording=recording))
62+
centered_recording_scaled = CenterRecording(scale_to_uV(recording=recording), seed=2205)
6363
traces_scaled_with_preprocessor = centered_recording_scaled.get_traces()
6464

6565
np.testing.assert_allclose(traces_scaled_with_argument, traces_scaled_with_preprocessor)
@@ -68,3 +68,8 @@ def test_scaling_in_preprocessing_chain():
6868
traces_scaled_with_preprocessor_and_argument = centered_recording_scaled.get_traces(return_scaled=True)
6969

7070
np.testing.assert_allclose(traces_scaled_with_preprocessor, traces_scaled_with_preprocessor_and_argument)
71+
72+
73+
if __name__ == "__main__":
74+
test_scale_to_uV()
75+
test_scaling_in_preprocessing_chain()

0 commit comments

Comments
 (0)