|
1 | 1 | import pytest |
2 | 2 |
|
3 | 3 | import numpy as np |
4 | | -from spikeinterface import generate_ground_truth_recording, create_sorting_analyzer, load, SortingAnalyzer, Templates, aggregate_channels |
| 4 | +from spikeinterface import ( |
| 5 | + generate_ground_truth_recording, |
| 6 | + create_sorting_analyzer, |
| 7 | + load, |
| 8 | + SortingAnalyzer, |
| 9 | + Templates, |
| 10 | + aggregate_channels, |
| 11 | +) |
5 | 12 | from spikeinterface.core.motion import Motion |
6 | 13 | from spikeinterface.core.generate import generate_unit_locations, generate_templates |
7 | 14 | from spikeinterface.core.testing import check_recordings_equal, check_sortings_equal |
@@ -183,15 +190,15 @@ def test_load_aggregate_recording_from_json(generate_recording_sorting, tmp_path |
183 | 190 |
|
184 | 191 | recording, _ = generate_recording_sorting |
185 | 192 |
|
186 | | - recording.set_property("group", [0,0,1,1]) |
187 | | - list_of_recs = list(recording.split_by('group').values()) |
| 193 | + recording.set_property("group", [0, 0, 1, 1]) |
| 194 | + list_of_recs = list(recording.split_by("group").values()) |
188 | 195 | aggregated_rec = aggregate_channels(list_of_recs) |
189 | 196 |
|
190 | 197 | recording_path = tmp_path / "aggregated_recording" |
191 | 198 | aggregated_rec.save_to_folder(folder=recording_path) |
192 | 199 | loaded_rec = load(recording_path / "provenance.json", base_folder=recording_path) |
193 | 200 |
|
194 | | - assert np.all(loaded_rec.get_property('group') == recording.get_property('group')) |
| 201 | + assert np.all(loaded_rec.get_property("group") == recording.get_property("group")) |
195 | 202 |
|
196 | 203 |
|
197 | 204 | @pytest.mark.streaming_extractors |
|
0 commit comments