fix: correct channel selection in tests when mono-channel recording
@@ -25,7 +25,10 @@ def check_recordings_equal(RX1, RX2, return_scaled=True, force_dtype=None):
RX2.get_traces(segment_index=segment_idx, return_scaled=return_scaled).astype(force_dtype))
sf = 0
ef = N
+ if RX1.get_num_channels() > 1:
ch = [RX1.get_channel_ids()[0], RX1.get_channel_ids()[-1]]
+ else:
+ ch = [RX1.get_channel_ids()[0]]
if force_dtype is None:
assert np.allclose(RX1.get_traces(segment_index=segment_idx, channel_ids=ch, start_frame=sf, end_frame=ef,
return_scaled=return_scaled),
