Unverified Commit 45761f0a authored by jimchen90's avatar jimchen90 Committed by GitHub
Browse files

Update sample rate (#820)



* Update sample rate of libritts test
Co-authored-by: default avatarJi Chen <jimchen90@devfair0160.h2.fair>
parent 6897aff1
...@@ -33,8 +33,8 @@ class TestLibriTTS(TempDirMixin, TorchaudioTestCase): ...@@ -33,8 +33,8 @@ class TestLibriTTS(TempDirMixin, TorchaudioTestCase):
os.makedirs(file_dir, exist_ok=True) os.makedirs(file_dir, exist_ok=True)
path = os.path.join(file_dir, filename) path = os.path.join(file_dir, filename)
data = get_whitenoise(sample_rate=8000, duration=6, n_channels=1, dtype='int16', seed=i) data = get_whitenoise(sample_rate=24000, duration=2, n_channels=1, dtype='int16', seed=i)
save_wav(path, data, 8000) save_wav(path, data, 24000)
cls.data.append(normalize_wav(data)) cls.data.append(normalize_wav(data))
original_text_filename = f'{"_".join(str(u) for u in utterance_id)}.original.txt' original_text_filename = f'{"_".join(str(u) for u in utterance_id)}.original.txt'
...@@ -61,7 +61,7 @@ class TestLibriTTS(TempDirMixin, TorchaudioTestCase): ...@@ -61,7 +61,7 @@ class TestLibriTTS(TempDirMixin, TorchaudioTestCase):
expected_ids = self.utterance_ids[i] expected_ids = self.utterance_ids[i]
expected_data = self.data[i] expected_data = self.data[i]
self.assertEqual(expected_data, waveform, atol=5e-5, rtol=1e-8) self.assertEqual(expected_data, waveform, atol=5e-5, rtol=1e-8)
assert sample_rate == 8000 assert sample_rate == 24000
assert speaker_id == expected_ids[0] assert speaker_id == expected_ids[0]
assert chapter_id == expected_ids[1] assert chapter_id == expected_ids[1]
assert original_text == self.original_text assert original_text == self.original_text
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment