From 3dd6b359daa86252878e480dddbe9dc719e98c2b Mon Sep 17 00:00:00 2001 From: Heberto Mayorquin Date: Tue, 17 Dec 2024 12:27:04 -0600 Subject: [PATCH] fully restore sorting anlayzer --- .../core/tests/test_sortinganalyzer.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/spikeinterface/core/tests/test_sortinganalyzer.py b/src/spikeinterface/core/tests/test_sortinganalyzer.py index 8d8beaa491..15f089f784 100644 --- a/src/spikeinterface/core/tests/test_sortinganalyzer.py +++ b/src/spikeinterface/core/tests/test_sortinganalyzer.py @@ -84,8 +84,8 @@ def test_SortingAnalyzer_binary_folder(tmp_path, dataset): # test select_units see https://github.com/SpikeInterface/spikeinterface/issues/3041 # this bug requires that we have an info.json file so we calculate templates above - select_units_sorting_analyer = sorting_analyzer.select_units(unit_ids=["1"]) - assert len(select_units_sorting_analyer.unit_ids) == "1" + select_units_sorting_analyer = sorting_analyzer.select_units(unit_ids=[1]) + assert len(select_units_sorting_analyer.unit_ids) == 1 folder = tmp_path / "test_SortingAnalyzer_binary_folder" if folder.exists(): @@ -129,11 +129,11 @@ def test_SortingAnalyzer_zarr(tmp_path, dataset): # test select_units see https://github.com/SpikeInterface/spikeinterface/issues/3041 # this bug requires that we have an info.json file so we calculate templates above - select_units_sorting_analyer = sorting_analyzer.select_units(unit_ids=["1"]) - assert len(select_units_sorting_analyer.unit_ids) == "1" - remove_units_sorting_analyer = sorting_analyzer.remove_units(remove_unit_ids=["1"]) + select_units_sorting_analyer = sorting_analyzer.select_units(unit_ids=[1]) + assert len(select_units_sorting_analyer.unit_ids) == 1 + remove_units_sorting_analyer = sorting_analyzer.remove_units(remove_unit_ids=[1]) assert len(remove_units_sorting_analyer.unit_ids) == len(sorting_analyzer.unit_ids) - 1 - assert "1" not in remove_units_sorting_analyer.unit_ids + assert 1 not in remove_units_sorting_analyer.unit_ids # test no compression sorting_analyzer_no_compression = create_sorting_analyzer(