From fdd5aa873424056a6a924b3ac9316d2a8c08a05f Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Mon, 4 Jul 2022 01:22:29 -0400 Subject: [PATCH] tests --- tts/text/pipeline/test_english_pipeline_v1.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 tts/text/pipeline/test_english_pipeline_v1.py diff --git a/tts/text/pipeline/test_english_pipeline_v1.py b/tts/text/pipeline/test_english_pipeline_v1.py new file mode 100644 index 0000000..938cfc2 --- /dev/null +++ b/tts/text/pipeline/test_english_pipeline_v1.py @@ -0,0 +1,26 @@ +from tts.text.pipeline.english_pipeline_v1 import EnglishPipelineV1 + + +class TestEnglishPipelineV1: + + def test_no_input(self): + pipeline = EnglishPipelineV1() + output = pipeline.user_input_to_sequence("") + assert output == [] + + def test_whitespace(self): + pipeline = EnglishPipelineV1() + output = pipeline.user_input_to_sequence(" \n \t ") + assert output == [] + + def test_single_word(self): + pipeline = EnglishPipelineV1() + output = pipeline.user_input_to_sequence("hello") + assert output == [106, 73, 117, 123] # NB: Arpabet encoded + + def test_two_words(self): + pipeline = EnglishPipelineV1() + output = pipeline.user_input_to_sequence("hello hello") + assert output == [106, 73, 117, 123, 11, 106, 73, 117, 123] + + \ No newline at end of file