def test_cut_waveform_value(self): wav = np.random.randn(3, 2000) assert np.allclose(transforms.CutWaveform(100, 1900)(wav), wav[:, 100:1900])
def test_cut_waveform_shape(self): wav = np.random.randn(3, 2000) assert transforms.CutWaveform(100, 1900)(wav).shape == (3, 1800)