Skip to content

Commit

Permalink
enable model manager test
Browse files Browse the repository at this point in the history
  • Loading branch information
erogol committed Mar 10, 2021
1 parent 38a29ce commit 4cc162a
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions tests/test_model_manager.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,20 @@
# #!/usr/bin/env python3`
# import os
# import shutil
# import glob
# from tests import get_tests_output_path
# from TTS.utils.manage import ModelManager
#!/usr/bin/env python3`
import os
import shutil
import glob
from tests import get_tests_output_path
from TTS.utils.manage import ModelManager

# TODO: re-enable this after moving models to github releases
# 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)
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)

0 comments on commit 4cc162a

Please sign in to comment.