mirror of
https://github.com/wassname/TTS.git
synced 2026-09-25 13:20:28 +08:00
🐸 Coqui Studio API integration (#2484)
* Warn when lang is not avail * Make style * Implement Coqui Studio API * Test * Update docs * Set action * Make style * Make lint * Update README * Make style * Fix action * Run actions
This commit is contained in:
@@ -2,12 +2,41 @@ import os
|
||||
import unittest
|
||||
|
||||
from tests import get_tests_data_path, get_tests_output_path
|
||||
from TTS.api import TTS
|
||||
from TTS.api import CS_API, TTS
|
||||
|
||||
OUTPUT_PATH = os.path.join(get_tests_output_path(), "test_python_api.wav")
|
||||
cloning_test_wav_path = os.path.join(get_tests_data_path(), "ljspeech/wavs/LJ001-0028.wav")
|
||||
|
||||
|
||||
class CS_APITest(unittest.TestCase):
|
||||
def test_speakers(self):
|
||||
tts = CS_API()
|
||||
self.assertGreater(len(tts.speakers), 1)
|
||||
|
||||
def test_emotions(self):
|
||||
tts = CS_API()
|
||||
self.assertGreater(len(tts.emotions), 1)
|
||||
|
||||
def test_list_calls(self):
|
||||
tts = CS_API()
|
||||
self.assertGreater(len(tts.list_voices()), 1)
|
||||
self.assertGreater(len(tts.list_speakers()), 1)
|
||||
self.assertGreater(len(tts.list_all_speakers()), 1)
|
||||
self.assertGreater(len(tts.list_speakers_as_tts_models()), 1)
|
||||
|
||||
def test_name_to_speaker(self):
|
||||
tts = CS_API()
|
||||
speaker_name = tts.list_speakers_as_tts_models()[0].split("/")[2]
|
||||
speaker = tts.name_to_speaker(speaker_name)
|
||||
self.assertEqual(speaker.name, speaker_name)
|
||||
|
||||
def test_tts(self):
|
||||
tts = CS_API()
|
||||
wav, sr = tts.tts(text="This is a test.", speaker_name=tts.list_speakers()[0].name)
|
||||
self.assertEqual(sr, 44100)
|
||||
self.assertGreater(len(wav), 1)
|
||||
|
||||
|
||||
class TTSTest(unittest.TestCase):
|
||||
def test_single_speaker_model(self):
|
||||
tts = TTS(model_name="tts_models/de/thorsten/tacotron2-DDC", progress_bar=False, gpu=False)
|
||||
@@ -26,6 +55,30 @@ class TTSTest(unittest.TestCase):
|
||||
self.assertIsNone(tts.speakers)
|
||||
self.assertIsNone(tts.languages)
|
||||
|
||||
def test_studio_model(self):
|
||||
tts = TTS(model_name="coqui_studio/en/Torcull Diarmuid/coqui_studio")
|
||||
tts.tts_to_file(text="This is a test.")
|
||||
|
||||
# check speed > 2.0 raises error
|
||||
raised_error = False
|
||||
try:
|
||||
_ = tts.tts(text="This is a test.", speed=4.0, emotion="Sad") # should raise error with speed > 2.0
|
||||
except ValueError:
|
||||
raised_error = True
|
||||
self.assertTrue(raised_error)
|
||||
|
||||
# check emotion is invalid
|
||||
raised_error = False
|
||||
try:
|
||||
_ = tts.tts(text="This is a test.", speed=2.0, emotion="No Emo") # should raise error with speed > 2.0
|
||||
except ValueError:
|
||||
raised_error = True
|
||||
self.assertTrue(raised_error)
|
||||
|
||||
# check valid call
|
||||
wav = tts.tts(text="This is a test.", speed=2.0, emotion="Sad")
|
||||
self.assertGreater(len(wav), 0)
|
||||
|
||||
def test_multi_speaker_multi_lingual_model(self):
|
||||
tts = TTS()
|
||||
tts.load_tts_model_by_name(tts.models[0]) # YourTTS
|
||||
|
||||
Reference in New Issue
Block a user