Model zoo tests (#900)

* Fix VITS model multi-speaker init

* Remove gdrive support in model manager

* Add model zoo tests

tests/aux_tests/model_manager.py

@@ -1,21 +0,0 @@
-#!/usr/bin/env python3`
-import glob
-import os
-import shutil
-
-from tests import get_tests_output_path
-from TTS.utils.manage import ModelManager
-
-
-def test_if_all_models_available():
-    """Check if all the models are downloadable."""
-    print(" > Checking the availability of all the models under the ModelManager.")
-    manager = ModelManager(output_prefix=get_tests_output_path())
-    model_names = manager.list_models()
-    for model_name in model_names:
-        manager.download_model(model_name)
-        print(f" | > OK: {model_name}")
-
-    folders = glob.glob(os.path.join(manager.output_prefix, "*"))
-    assert len(folders) == len(model_names)
-    shutil.rmtree(manager.output_prefix)

tests/zoo_tests/test_models.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3`
+import glob
+import os
+import shutil
+
+from tests import get_tests_output_path, run_cli
+from TTS.tts.utils.speakers import SpeakerManager
+from TTS.utils.generic_utils import get_user_data_dir
+from TTS.utils.manage import ModelManager
+
+
+def test_run_all_models():
+    """Check if all the models are downloadable and tts models run correctly."""
+    print(" > Run synthesizer with all the models.")
+    download_dir = get_user_data_dir("tts")
+    output_path = os.path.join(get_tests_output_path(), "output.wav")
+    manager = ModelManager(output_prefix=get_tests_output_path())
+    model_names = manager.list_models()
+    for model_name in model_names:
+        model_path, _, _ = manager.download_model(model_name)
+        if "tts_models" in model_name:
+            local_download_dir = os.path.dirname(model_path)
+            # download and run the model
+            speaker_files = glob.glob(local_download_dir + "/speaker*")
+            if len(speaker_files) > 0:
+                # multi-speaker model
+                if "speaker_ids" in speaker_files[0]:
+                    speaker_manager = SpeakerManager(speaker_id_file_path=speaker_files[0])
+                elif "speakers" in speaker_files[0]:
+                    speaker_manager = SpeakerManager(d_vectors_file_path=speaker_files[0])
+                speaker_id = list(speaker_manager.speaker_ids.keys())[0]
+                run_cli(
+                    f"tts --model_name  {model_name} "
+                    f'--text "This is an example." --out_path "{output_path}" --speaker_idx "{speaker_id}"'
+                )
+            else:
+                # single-speaker model
+                run_cli(f"tts --model_name  {model_name} " f'--text "This is an example." --out_path "{output_path}"')
+            # remove downloaded models
+            shutil.rmtree(download_dir)
+        else:
+            # only download the model
+            manager.download_model(model_name)
+        print(f" | > OK: {model_name}")
+
+    folders = glob.glob(os.path.join(manager.output_prefix, "*"))
+    assert len(folders) == len(model_names)
+    shutil.rmtree(manager.output_prefix)