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