@@ -287,69 +287,3 @@ def _get_sorting_with_recording_attached(self, recording_for_durations, recordin
287287 assert sorting .has_recording ()
288288
289289 return sorting
290-
291-
292- # TODO: deprecate original implementations ###
293- def test_time_handling (create_cache_folder ):
294- cache_folder = create_cache_folder
295- durations = [[10 ], [10 , 5 ]]
296-
297- # test multi-segment
298- for i , dur in enumerate (durations ):
299- rec = generate_recording (num_channels = 4 , durations = dur )
300- sort = generate_sorting (num_units = 10 , durations = dur )
301-
302- for segment_index in range (rec .get_num_segments ()):
303- original_times = rec .get_times (segment_index = segment_index )
304- new_times = original_times + 5
305- rec .set_times (new_times , segment_index = segment_index )
306-
307- sort .register_recording (rec )
308- assert sort .has_recording ()
309-
310- rec_cache = rec .save (folder = cache_folder / f"rec{ i } " )
311-
312- for segment_index in range (sort .get_num_segments ()):
313- assert rec .has_time_vector (segment_index = segment_index )
314- assert sort .has_time_vector (segment_index = segment_index )
315-
316- # times are correctly saved by the recording
317- assert np .allclose (
318- rec .get_times (segment_index = segment_index ), rec_cache .get_times (segment_index = segment_index )
319- )
320-
321- # spike times are correctly adjusted
322- for u in sort .get_unit_ids ():
323- spike_times = sort .get_unit_spike_train (u , segment_index = segment_index , return_times = True )
324- rec_times = rec .get_times (segment_index = segment_index )
325- assert np .all (spike_times >= rec_times [0 ])
326- assert np .all (spike_times <= rec_times [- 1 ])
327-
328-
329- def test_frame_slicing ():
330- duration = [10 ]
331-
332- rec = generate_recording (num_channels = 4 , durations = duration )
333- sort = generate_sorting (num_units = 10 , durations = duration )
334-
335- original_times = rec .get_times ()
336- new_times = original_times + 5
337- rec .set_times (new_times )
338-
339- sort .register_recording (rec )
340-
341- start_frame = 3 * rec .get_sampling_frequency ()
342- end_frame = 7 * rec .get_sampling_frequency ()
343-
344- rec_slice = rec .frame_slice (start_frame = start_frame , end_frame = end_frame )
345- sort_slice = sort .frame_slice (start_frame = start_frame , end_frame = end_frame )
346-
347- for u in sort_slice .get_unit_ids ():
348- spike_times = sort_slice .get_unit_spike_train (u , return_times = True )
349- rec_times = rec_slice .get_times ()
350- assert np .all (spike_times >= rec_times [0 ])
351- assert np .all (spike_times <= rec_times [- 1 ])
352-
353-
354- if __name__ == "__main__" :
355- test_frame_slicing ()
0 commit comments