def test_n_channels(self): test_sig_1a = Signal(np.random.rand(10)) self.assertEqual(test_sig_1a.n_channels(), 1) test_sig_1b = Signal(np.random.rand(1,10)) self.assertEqual(test_sig_1b.n_channels(), 1) test_sig_2 = Signal(np.random.rand(2,10)) self.assertEqual(test_sig_2.n_channels(), 2)