TTS/tests/test_preprocessors.py

import os
import unittest

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")