Skip to content

Commit f7018c6

Browse files
committed
zach resquest fix test
1 parent c1cb3d2 commit f7018c6

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

src/spikeinterface/core/tests/test_unitsaggregationsorting.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,10 @@ def test_unitsaggregationsorting(create_cache_folder):
3434
spiketrain1_1 = sorting1.get_unit_spike_train(unit_ids[1], segment_index=seg)
3535
spiketrains2_0 = sorting2.get_unit_spike_train(unit_ids[0], segment_index=seg)
3636
spiketrains3_2 = sorting3.get_unit_spike_train(unit_ids[2], segment_index=seg)
37-
assert np.allclose(spiketrain1_1, sorting_agg.get_unit_spike_train(str(unit_ids[1]), segment_index=seg))
37+
assert np.allclose(spiketrain1_1, sorting_agg.get_unit_spike_train(unit_ids[1], segment_index=seg))
38+
assert np.allclose(spiketrains2_0, sorting_agg.get_unit_spike_train(num_units + unit_ids[0], segment_index=seg))
3839
assert np.allclose(
39-
spiketrains2_0, sorting_agg.get_unit_spike_train(str(num_units + unit_ids[0]), segment_index=seg)
40-
)
41-
assert np.allclose(
42-
spiketrains3_2, sorting_agg.get_unit_spike_train(str(2 * num_units + unit_ids[2]), segment_index=seg)
40+
spiketrains3_2, sorting_agg.get_unit_spike_train(2 * num_units + unit_ids[2], segment_index=seg)
4341
)
4442

4543
# test rename units

src/spikeinterface/core/unitsaggregationsorting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def __init__(self, sorting_list, renamed_unit_ids=None):
4646
else:
4747
default_unit_ids = [str(i) for i in range(num_all_units)]
4848
if all_ids_are_same_type and np.issubdtype(unit_ids_dtypes[0], np.integer):
49-
unit_ids = np.arange(num_all_units, dtype=dtype)
49+
unit_ids = np.arange(num_all_units, dtype=np.uint64)
5050
else:
5151
unit_ids = default_unit_ids
5252

0 commit comments

Comments
 (0)