Skip to content

Commit 7289dc6

Browse files
committed
Extend peak_pipeline tests
1 parent a0ac3f5 commit 7289dc6

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

spikeinterface/sortingcomponents/tests/test_peak_pipeline.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,18 @@ def test_run_peak_pipeline():
130130
assert np.array_equal(amplitudes, amplitudes2)
131131
assert np.array_equal(amplitudes2, amplitudes3)
132132

133+
waveforms_rms_file = folder / 'waveforms_rms.npy'
134+
assert waveforms_rms_file.is_file()
135+
waveforms_rms3 = np.load(waveforms_rms_file)
136+
assert np.array_equal(waveforms_rms, waveforms_rms2)
137+
assert np.array_equal(waveforms_rms2, waveforms_rms3)
138+
139+
denoised_waveforms_rms_file = folder / 'denoised_waveforms_rms.npy'
140+
assert denoised_waveforms_rms_file.is_file()
141+
denoised_waveforms_rms3 = np.load(denoised_waveforms_rms_file)
142+
assert np.array_equal(denoised_waveforms_rms, denoised_waveforms_rms2)
143+
assert np.array_equal(denoised_waveforms_rms2, denoised_waveforms_rms3)
144+
133145

134146
# Test pickle mechanism
135147
for node in nodes:

0 commit comments

Comments
 (0)