mirror of https://github.com/coqui-ai/TTS.git
19 lines
804 B
Python
19 lines
804 B
Python
import unittest
|
|
import os
|
|
from tests import get_tests_input_path
|
|
|
|
from TTS.tts.datasets.preprocess import common_voice
|
|
|
|
|
|
class TestPreprocessors(unittest.TestCase):
|
|
|
|
def test_common_voice_preprocessor(self): #pylint: disable=no-self-use
|
|
root_path = get_tests_input_path()
|
|
meta_file = "common_voice.tsv"
|
|
items = common_voice(root_path, meta_file)
|
|
assert items[0][0] == 'The applicants are invited for coffee and visa is given immediately.'
|
|
assert items[0][1] == os.path.join(get_tests_input_path(), "clips", "common_voice_en_20005954.wav")
|
|
|
|
assert items[-1][0] == "Competition for limited resources has also resulted in some local conflicts."
|
|
assert items[-1][1] == os.path.join(get_tests_input_path(), "clips", "common_voice_en_19737074.wav")
|