Skip to content

Commit 3e30777

Browse files
committed
remove _sorted_spike_vector
1 parent 5578974 commit 3e30777

File tree

1 file changed

+9
-13
lines changed

1 file changed

+9
-13
lines changed

src/spikeinterface/core/testing.py

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

Comments
 (0)