def test_cutoff(self): size = 10 waveform = np.arange(size, dtype=float) for cutoff in (-1, 1): # outside of [0, 1) range self.assertRaises(ValueError, waveform_features.waveform_width, waveform, cutoff=cutoff) for cutoff in np.linspace(0., 1., num=size, endpoint=False): width = waveform_features.waveform_width(waveform, cutoff=cutoff) self.assertEqual(width, size - 1)
def test_np_array(self): waveform = np.asarray(self.waveform) width = waveform_features.waveform_width(waveform) self.assertEqual(width, self.target_width)
def test_pq_quantity(self): waveform = np.asarray(self.waveform) * pq.mV width = waveform_features.waveform_width(waveform) self.assertEqual(width, self.target_width)
def test_list(self): width = waveform_features.waveform_width(self.waveform) self.assertEqual(width, self.target_width)