@@ -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