def test_cut_audio_mask(self): cut = MonoCut('cut', start=0, duration=2, channel=0, recording=Mock(sampling_rate=16000)) mask = cut.supervisions_audio_mask() assert mask.sum() == 0
def test_cut_audio_mask(self, supervisions, alignment): cut = MonoCut('cut', start=0, duration=2, channel=0, recording=Mock(sampling_rate=16000), supervisions=supervisions) mask = cut.supervisions_audio_mask(use_alignment_if_exists=alignment) if alignment == "word": ones = np.index_exp[list( chain(range(0, 1600), range(3200, 6400), range(9600, 12800)))] zeros = np.index_exp[list( chain(range(1600, 3200), range(6400, 9600), range(12800, 32000)))] else: ones = np.index_exp[list(chain(range(0, 8000), range(9600, 12800)))] zeros = np.index_exp[list( chain(range(8000, 9600), range(12800, 32000)))] assert (mask[ones] == 1).all() assert (mask[zeros] == 0).all()