Merge pull request #2405 from maxbachmann/unittests

improve test runtime
pull/2410/head
Åke 2019-12-02 10:52:08 +01:00 committed by GitHub
commit f7e8a4ddf4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions

View File

@ -46,30 +46,28 @@ class TestTTS(unittest.TestCase):
"</foo_invalid end=whatever>" "</foo_invalid end=whatever>"
sentence_extra_ssml = "<whispered>whisper tts<\\whispered>" sentence_extra_ssml = "<whispered>whisper tts<\\whispered>"
tts = TestTTS("en-US", {}, TestTTSValidator(None))
# test valid ssml # test valid ssml
tts = TestTTS("en-US", {}, TestTTSValidator(None), tts.ssml_tags = ['speak', 'prosody']
ssml_tags=['speak', 'prosody'])
self.assertEqual(tts.validate_ssml(sentence), sentence) self.assertEqual(tts.validate_ssml(sentence), sentence)
# test extra ssml # test extra ssml
tts = TestTTS("en-US", {}, TestTTSValidator(None), tts.ssml_tags = ['whispered']
ssml_tags=['whispered'])
self.assertEqual(tts.validate_ssml(sentence_extra_ssml), self.assertEqual(tts.validate_ssml(sentence_extra_ssml),
sentence_extra_ssml) sentence_extra_ssml)
# test unsupported extra ssml # test unsupported extra ssml
tts = TestTTS("en-US", {}, TestTTSValidator(None), tts.ssml_tags = ['speak', 'prosody']
ssml_tags=['speak', 'prosody'])
self.assertEqual(tts.validate_ssml(sentence_extra_ssml), self.assertEqual(tts.validate_ssml(sentence_extra_ssml),
"whisper tts") "whisper tts")
# test mixed valid / invalid ssml # test mixed valid / invalid ssml
tts = TestTTS("en-US", {}, TestTTSValidator(None), tts.ssml_tags = ['speak', 'prosody']
ssml_tags=['speak', 'prosody'])
self.assertEqual(tts.validate_ssml(sentence_bad_ssml), sentence) self.assertEqual(tts.validate_ssml(sentence_bad_ssml), sentence)
# test unsupported ssml # test unsupported ssml
tts = TestTTS("en-US", {}, TestTTSValidator(None)) tts.ssml_tags = []
self.assertEqual(tts.validate_ssml(sentence), sentence_no_ssml) self.assertEqual(tts.validate_ssml(sentence), sentence_no_ssml)
self.assertEqual(tts.validate_ssml(sentence_bad_ssml), self.assertEqual(tts.validate_ssml(sentence_bad_ssml),