Skip to content

Commit 16dafc0

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 03d8a94 commit 16dafc0

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

src/spikeinterface/core/tests/test_loading.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,14 @@
11
import pytest
22

33
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+
)
512
from spikeinterface.core.motion import Motion
613
from spikeinterface.core.generate import generate_unit_locations, generate_templates
714
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
183190

184191
recording, _ = generate_recording_sorting
185192

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())
188195
aggregated_rec = aggregate_channels(list_of_recs)
189196

190197
recording_path = tmp_path / "aggregated_recording"
191198
aggregated_rec.save_to_folder(folder=recording_path)
192199
loaded_rec = load(recording_path / "provenance.json", base_folder=recording_path)
193200

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"))
195202

196203

197204
@pytest.mark.streaming_extractors

0 commit comments

Comments
 (0)