Add tests

coqui-ai · May 23, 2023 · acf43ac · acf43ac
1 parent 0a23914
commit acf43ac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py
@@ -248,7 +248,7 @@ def print_model_license(model_item: Dict):
 
     def download_fairseq_model(self, model_name, output_path):
         URI_PREFIX = "coqui.gateway.scarf.sh/fairseq/"
-        model_type, lang, dataset, model = model_name.split("/")
+        _, lang, _, _ = model_name.split("/")
         model_download_uri = os.path.join(URI_PREFIX, f"{lang}.tar.gz")
         self._download_tar_file(model_download_uri, output_path, self.progress_bar)
 

diff --git a/tests/inference_tests/test_python_api.py b/tests/inference_tests/test_python_api.py
@@ -83,6 +83,10 @@ def test_studio_model(self):
             wav = tts.tts(text="This is a test.", speed=2.0, emotion="Sad")
             self.assertGreater(len(wav), 0)
 
+        def test_fairseq_model(self):
+            tts = TTS(model_name="tts_models/eng/fairseq/vits")
+            tts.tts_to_file(text="This is a test.")
+
         def test_multi_speaker_multi_lingual_model(self):
             tts = TTS()
             tts.load_tts_model_by_name(tts.models[0])  # YourTTS