mirror of https://github.com/coqui-ai/TTS.git
25 lines
966 B
Python
25 lines
966 B
Python
import os
|
|
import unittest
|
|
|
|
import torch as T
|
|
|
|
from server.synthesizer import Synthesizer
|
|
from tests import get_tests_input_path, get_tests_output_path, get_tests_path
|
|
from utils.text.symbols import phonemes, symbols
|
|
from utils.generic_utils import load_config, save_checkpoint, setup_model
|
|
|
|
|
|
class DemoServerTest(unittest.TestCase):
|
|
def _create_random_model(self):
|
|
config = load_config(os.path.join(get_tests_output_path(), 'dummy_model_config.json'))
|
|
num_chars = len(phonemes) if config.use_phonemes else len(symbols)
|
|
model = setup_model(num_chars, 0, config)
|
|
output_path = os.path.join(get_tests_output_path())
|
|
save_checkpoint(model, None, None, None, output_path, 10, 10)
|
|
|
|
def test_in_out(self):
|
|
self._create_random_model()
|
|
config = load_config(os.path.join(get_tests_input_path(), 'server_config.json'))
|
|
synthesizer = Synthesizer(config)
|
|
synthesizer.tts("Better this test works!!")
|