"vscode:/vscode.git/clone" did not exist on "8560d58bdd0f81c9eb53b8d1fa06dcf857f7af34"
Unverified Commit 6edb3355 authored by Krishna Kalyan's avatar Krishna Kalyan Committed by GitHub
Browse files

Refactor CMUARCTIC unittest (#1147)


Co-authored-by: default avatarkrishnakalyan3 <skalyan@cloudera.com>
parent 02e4f6d2
...@@ -12,19 +12,15 @@ from torchaudio_unittest.common_utils import ( ...@@ -12,19 +12,15 @@ from torchaudio_unittest.common_utils import (
) )
class TestCMUARCTIC(TempDirMixin, TorchaudioTestCase): def get_mock_dataset(root_dir):
backend = "default" """
root_dir: directory to the mocked dataset
root_dir = None """
samples = [] mocked_data = []
@classmethod
def setUpClass(cls):
cls.root_dir = cls.get_base_temp_dir()
sample_rate = 16000 sample_rate = 16000
utterance = "This is a test utterance." utterance = "This is a test utterance."
base_dir = os.path.join(cls.root_dir, "ARCTIC", "cmu_us_aew_arctic") base_dir = os.path.join(root_dir, "ARCTIC", "cmu_us_aew_arctic")
txt_dir = os.path.join(base_dir, "etc") txt_dir = os.path.join(base_dir, "etc")
os.makedirs(txt_dir, exist_ok=True) os.makedirs(txt_dir, exist_ok=True)
txt_file = os.path.join(txt_dir, "txt.done.data") txt_file = os.path.join(txt_dir, "txt.done.data")
...@@ -51,9 +47,22 @@ class TestCMUARCTIC(TempDirMixin, TorchaudioTestCase): ...@@ -51,9 +47,22 @@ class TestCMUARCTIC(TempDirMixin, TorchaudioTestCase):
utterance, utterance,
utterance_id.split("_")[1], utterance_id.split("_")[1],
) )
cls.samples.append(sample) mocked_data.append(sample)
txt.write(f'( {utterance_id} "{utterance}" )\n') txt.write(f'( {utterance_id} "{utterance}" )\n')
seed += 1 seed += 1
return mocked_data
class TestCMUARCTIC(TempDirMixin, TorchaudioTestCase):
backend = "default"
root_dir = None
samples = []
@classmethod
def setUpClass(cls):
cls.root_dir = cls.get_base_temp_dir()
cls.samples = get_mock_dataset(cls.root_dir)
def _test_cmuarctic(self, dataset): def _test_cmuarctic(self, dataset):
n_ite = 0 n_ite = 0
......
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