storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

test punctuation

Browse Source
This commit is contained in:
Brandon Thomas
2022-07-02 11:25:40 -04:00
parent 4926faf42c
commit 22a6986bd4
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/text/phonetization/test_english_phonetization.py
+10
View File
@@ -20,3 +20,13 @@ class TestEnglishPhonetization:
phonetization = EnglishPhonetization(phoneme_dictionary)
output = phonetization.convert_to_phonemes("nonword is a test")
assert output == ["nonword", "IH1 Z", "AH0", "T EH1 S T"]
def test_punctuation(self):
phoneme_dictionary = CMUDict(DEFAULT_CMUDICT_PATH)
phonetization = EnglishPhonetization(phoneme_dictionary)
output = phonetization.convert_to_phonemes("this...")
assert output == ["DH IH1 S", "..."]
output = phonetization.convert_to_phonemes("this!")
assert output == ["DH IH1 S", "!"]
output = phonetization.convert_to_phonemes("this?")
assert output == ["DH IH1 S", "?"]