TTS/tests/test_synthesize.py

29 lines
1001 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-04-27 13:04:20 +00:00
"tts --model_name tts_models/en/ljspeech/speedy-speech-wn "
2021-04-27 12:10:45 +00:00
f'--text "This is an example." --out_path "{output_path}"'
)
run_cli(
"tts --model_name tts_models/en/ljspeech/speedy-speech-wn "
"--vocoder_name vocoder_models/en/ljspeech/multiband-melgan "
f'--text "This is an example." --out_path "{output_path}"'
)
# multi-speaker model
run_cli("tts --model_name tts_models/en/vctk/sc-glow-tts --list_speaker_idxs")
run_cli(
2021-04-27 13:04:20 +00:00
f'tts --model_name tts_models/en/vctk/sc-glow-tts --speaker_idx "p304" '
2021-04-27 12:10:45 +00:00
f'--text "This is an example." --out_path "{output_path}"'
)