Skip to content

Commit e012094

Browse files
committed
Merge branch 'two_side_probes_grouping' of github.com:samuelgarcia/spikeinterface into two_side_probes_grouping
2 parents afc34c0 + 743e27e commit e012094

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

src/spikeinterface/core/tests/test_baserecording.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,6 @@ def test_BaseRecording(create_cache_folder):
219219

220220
assert np.array_equal(traces2, rec_p.get_traces(segment_index=0))
221221

222-
223222
# from probeinterface.plotting import plot_probe_group, plot_probe
224223
# import matplotlib.pyplot as plt
225224
# plot_probe(probe)
@@ -229,13 +228,19 @@ def test_BaseRecording(create_cache_folder):
229228
# test different group mode
230229
probe = Probe(ndim=2)
231230
positions_two_side = positions + positions
232-
shank_ids= ["a", "a", "a", "b", "b", "b"] * 2
231+
shank_ids = ["a", "a", "a", "b", "b", "b"] * 2
233232
contact_sides = ["front"] * 6 + ["back"] * 6
234-
probe.set_contacts(positions=positions_two_side, shapes="circle", shape_params={"radius": 5}, shank_ids=shank_ids, contact_sides=contact_sides)
233+
probe.set_contacts(
234+
positions=positions_two_side,
235+
shapes="circle",
236+
shape_params={"radius": 5},
237+
shank_ids=shank_ids,
238+
contact_sides=contact_sides,
239+
)
235240
probe.set_device_channel_indices(np.arange(12))
236241
probe.create_auto_shape()
237242
traces = np.zeros((1000, 12), dtype="int16")
238-
rec = NumpyRecording([traces], 30000.)
243+
rec = NumpyRecording([traces], 30000.0)
239244
rec1 = rec.set_probe(probe, group_mode="auto")
240245
assert np.unique(rec1.get_property("group")).size == 4
241246
rec2 = rec.set_probe(probe, group_mode="by_probe")
@@ -245,7 +250,6 @@ def test_BaseRecording(create_cache_folder):
245250
rec4 = rec.set_probe(probe, group_mode="by_side")
246251
assert np.unique(rec4.get_property("group")).size == 4
247252

248-
249253
# set unconnected probe
250254
probe = Probe(ndim=2)
251255
positions = [[0.0, 0.0], [0.0, 15.0], [0, 30.0]]

0 commit comments

Comments
 (0)