def test_librispeech(self): librispeech.LIBRISPEECH._ext_audio = '.wav' dataset = librispeech.LIBRISPEECH(self.root_dir) print(dataset._path) num_samples = 0 for i, (data, sample_rate, utterance, speaker_id, chapter_id, utterance_id) in enumerate(dataset): self.assertEqual(data, self.samples[i][0], atol=5e-5, rtol=1e-8) assert sample_rate == self.samples[i][1] assert utterance == self.samples[i][2] assert speaker_id == self.samples[i][3] assert chapter_id == self.samples[i][4] assert utterance_id == self.samples[i][5] num_samples += 1 assert num_samples == len(self.samples) librispeech.LIBRISPEECH._ext_audio = '.flac'
def test_librispeech_path(self): librispeech.LIBRISPEECH._ext_audio = '.wav' dataset = librispeech.LIBRISPEECH(Path(self.root_dir)) self._test_librispeech(dataset)