@@ -114,12 +114,10 @@ def check_sortings_equal(
114114
115115 def _sorted_spike_vector (SX ):
116116 spikes = SX .to_spike_vector ()
117- order = np .lexsort (
118- (spikes ["sample_index" ], spikes ["unit_index" ], spikes ["segment_index" ])
119- )
117+ order = np .lexsort ((spikes ["sample_index" ], spikes ["unit_index" ], spikes ["segment_index" ]))
120118 return spikes [order ]
121-
122- def _slice_spikes (spikes , start_frame = None , end_frame = None ):
119+
120+ def _slice_spikes (spikes , start_frame = None , end_frame = None ):
123121 mask = np .ones (spikes .size , dtype = bool )
124122 if start_frame is not None :
125123 mask &= spikes ["sample_index" ] >= start_frame
@@ -132,17 +130,16 @@ def _slice_spikes(spikes, start_frame = None, end_frame = None):
132130 assert_array_equal (s1 , s2 )
133131
134132 for start_frame , end_frame in [
135- (None , None ),
136- (30 , None ),
137- (None , max_spike_index - 30 ),
138- (30 , max_spike_index - 30 ),
139- ]:
133+ (None , None ),
134+ (30 , None ),
135+ (None , max_spike_index - 30 ),
136+ (30 , max_spike_index - 30 ),
137+ ]:
140138
141139 slice1 = _slice_spikes (s1 , start_frame , end_frame )
142140 slice2 = _slice_spikes (s2 , start_frame , end_frame )
143141 assert np .array_equal (slice1 , slice2 )
144142
145-
146143 if check_annotations :
147144 check_extractor_annotations_equal (SX1 , SX2 )
148145 if check_properties :
0 commit comments