@@ -287,69 +287,3 @@ def _get_sorting_with_recording_attached(self, recording_for_durations, recordin
287
287
assert sorting .has_recording ()
288
288
289
289
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