Skip to content

Commit f75c3d4

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 1f523e8 commit f75c3d4

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

src/spikeinterface/core/testing.py

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)