File size: 1,196 Bytes
1e495f3 b295d06 1e495f3 b295d06 1e495f3 b295d06 1e495f3 b295d06 1e495f3 42742c6 1e495f3 e5b5f3e 42742c6 1e495f3 b295d06 1e495f3 42742c6 b295d06 1e495f3 e5b5f3e 42742c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import pytest
from test_data.texts import read_translation
from lib.utils import save_csv
from environment import REPORTS_DIR
@pytest.fixture(scope="module")
def tts(get_platform):
if get_platform == "apple":
from lib.models.mac.kokoro import KokoroTTS
return KokoroTTS
elif get_platform == "intel":
from lib.models.intel.kokoro import KokoroTTS
return KokoroTTS
def test_tts_zh(tts):
tts = tts.from_language("zh")
report = []
for text in read_translation("zh", count_limit=20):
samples, sr, time_cost = tts.generate(text)
report.append([text, time_cost])
save_csv(REPORTS_DIR / "test_tts_zh.csv", ["text", "time"], report)
speed = [i[1] for i in report]
print("tts zh average speed: ", sum(speed) / len(speed))
def test_tts_en(tts):
tts = tts.from_language("en")
report = []
for text in read_translation("en",count_limit=20):
samples, sr, time_cost = tts.generate(text, speed=1.4)
report.append([text, time_cost])
save_csv(REPORTS_DIR / "test_tts_en.csv", ["text", "time"], report)
speed = [i[1] for i in report]
print("tts en average speed: ", sum(speed) / len(speed)) |