tests

2026-10-17 00:23:14 +00:00 · 2022-07-04 01:22:29 -04:00
parent c5ae491895
commit fdd5aa8734
1 changed files with 26 additions and 0 deletions
@@ -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]
+
+