Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add OOV case to tokenizer tests

This commit is contained in:
Eren Gölge 2021-11-25 17:31:25 +01:00
parent 4e8f9d6f10
commit 961e98a461
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/text_tests/test_tokenizer.py
View File

@ -52,6 +52,16 @@ class TestTTSTokenizer(unittest.TestCase):
self.tokenizer.print_logs()
self.tokenizer_ph.print_logs()
def test_not_found_characters(self):
self.ph = ESpeak("en-us")
self.tokenizer_local = TTSTokenizer(use_phonemes=True, characters=IPAPhonemes(), phonemizer=self.ph)
self.assertEqual(len(self.tokenizer.not_found_characters), 0)
text = "Yolk of one egg beaten light"
ids = self.tokenizer_local.text_to_ids(text)
text_hat = self.tokenizer_local.ids_to_text(ids)
self.assertEqual(self.tokenizer_local.not_found_characters, ['̩'])
self.assertEqual(text_hat, "jˈoʊk ʌv wˈʌn ˈɛɡ bˈiːʔn lˈaɪt")
def test_init_from_config(self):
@dataclass
class Characters(Coqpit):