File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
spikeinterface/sortingcomponents/tests Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -130,6 +130,18 @@ def test_run_peak_pipeline():
130
130
assert np .array_equal (amplitudes , amplitudes2 )
131
131
assert np .array_equal (amplitudes2 , amplitudes3 )
132
132
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
+
133
145
134
146
# Test pickle mechanism
135
147
for node in nodes :
You can’t perform that action at this time.
0 commit comments