TTS/tests/inference_tests/test_synthesize.py

28 lines
977 B
Python
Raw Normal View History

2021-04-27 12:10:45 +00:00
import os
2021-05-10 21:13:52 +00:00
from tests import get_tests_output_path, run_cli
2021-04-27 12:10:45 +00:00
def test_synthesize():
"""Test synthesize.py with diffent arguments."""
output_path = os.path.join(get_tests_output_path(), "output.wav")
run_cli("tts --list_models")
# single speaker model
run_cli(f'tts --text "This is an example." --out_path "{output_path}"')
run_cli(
2021-09-10 17:27:33 +00:00
"tts --model_name tts_models/en/ljspeech/glow-tts " f'--text "This is an example." --out_path "{output_path}"'
)
run_cli(
2021-09-10 17:27:33 +00:00
"tts --model_name tts_models/en/ljspeech/glow-tts "
"--vocoder_name vocoder_models/en/ljspeech/multiband-melgan "
f'--text "This is an example." --out_path "{output_path}"'
)
2021-04-27 12:10:45 +00:00
# multi-speaker model
run_cli("tts --model_name tts_models/en/vctk/sc-glow-tts --list_speaker_idxs")
run_cli(
f'tts --model_name tts_models/en/vctk/sc-glow-tts --speaker_idx "p304" '
f'--text "This is an example." --out_path "{output_path}"'
)