From c20ffdadb908d601e546323b113e994445546891 Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Mon, 2 Oct 2023 10:23:47 +0200 Subject: [PATCH] Tiny rewrite in tests --- src/spikeinterface/curation/tests/test_sortingview_curation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/spikeinterface/curation/tests/test_sortingview_curation.py b/src/spikeinterface/curation/tests/test_sortingview_curation.py index 22085f2f77..ce6c7dd5a6 100644 --- a/src/spikeinterface/curation/tests/test_sortingview_curation.py +++ b/src/spikeinterface/curation/tests/test_sortingview_curation.py @@ -174,8 +174,9 @@ def test_label_inheritance_int(): duration = 20.0 num_timepoints = int(sampling_frequency * duration) num_spikes = 1000 + num_units = 7 times = np.int_(np.sort(np.random.uniform(0, num_timepoints, num_spikes))) - labels = np.random.randint(1, 8, size=num_spikes) # 7 units: 1 to 7 + labels = np.random.randint(1, 1 + num_units, size=num_spikes) # 7 units: 1 to 7 sorting = se.NumpySorting.from_times_labels(times, labels, sampling_frequency)