def test_extract_channels(signal):
two_sig = signal.extract_channels(['chan0', 'chan1'])
assert two_sig.shape == (2, 200)
one_sig = signal.extract_channels(['chan2'])
assert one_sig.shape == (1, 200)
recombined = RasterizedSignal.concatenate_channels([two_sig, one_sig])
before = signal.as_continuous()
after = recombined.as_continuous()
assert np.array_equal(before, after)
def test_concatenate_channels(signal):
sig1 = signal
sig2 = sig1.jackknife_by_time(20, 2)
sig3 = RasterizedSignal.concatenate_channels([sig1, sig2])
assert sig1.as_continuous().shape == (3, 200)
assert sig3.as_continuous().shape == (6, 200)