@@ -112,19 +112,6 @@ def check_sortings_equal(
112112
113113 max_spike_index = SX1 .to_spike_vector ()["sample_index" ].max ()
114114
115- def _sorted_spike_vector (SX ):
116- spikes = SX .to_spike_vector ()
117- order = np .lexsort ((spikes ["sample_index" ], spikes ["unit_index" ], spikes ["segment_index" ]))
118- return spikes [order ]
119-
120- def _slice_spikes (spikes , start_frame = None , end_frame = None ):
121- mask = np .ones (spikes .size , dtype = bool )
122- if start_frame is not None :
123- mask &= spikes ["sample_index" ] >= start_frame
124- if end_frame is not None :
125- mask &= spikes ["sample_index" ] <= end_frame
126- return spikes [mask ]
127-
128115 s1 = SX1 .to_spike_vector ()
129116 s2 = SX2 .to_spike_vector ()
130117 assert_array_equal (s1 , s2 )
@@ -158,3 +145,12 @@ def check_extractor_properties_equal(EX1, EX2) -> None:
158145
159146 for property_name in EX1 .get_property_keys ():
160147 assert_array_equal (EX1 .get_property (property_name ), EX2 .get_property (property_name ))
148+
149+
150+ def _slice_spikes (spikes , start_frame = None , end_frame = None ):
151+ mask = np .ones (spikes .size , dtype = bool )
152+ if start_frame is not None :
153+ mask &= spikes ["sample_index" ] >= start_frame
154+ if end_frame is not None :
155+ mask &= spikes ["sample_index" ] <= end_frame
156+ return spikes [mask ]
0 commit comments