Spaces:
Running
Running
File size: 2,706 Bytes
edc06cb |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
""" `TTSEngineManager` クラスのテスト"""
import pytest
from fastapi import HTTPException
from voicevox_engine.dev.tts_engine.mock import MockTTSEngine
from voicevox_engine.tts_pipeline.tts_engine import LATEST_VERSION, TTSEngineManager
def test_tts_engines_register_engine() -> None:
"""TTSEngineManager.register_engine() で TTS エンジンを登録できる。"""
# Inputs
tts_engines = TTSEngineManager()
# Test
tts_engines.register_engine(MockTTSEngine(), "0.0.1")
def test_tts_engines_versions() -> None:
"""TTSEngineManager.versions() でバージョン一覧を取得できる。"""
# Inputs
tts_engines = TTSEngineManager()
tts_engines.register_engine(MockTTSEngine(), "0.0.1")
tts_engines.register_engine(MockTTSEngine(), "0.0.2")
# Expects
true_versions = ["0.0.1", "0.0.2"]
# Outputs
versions = tts_engines.versions()
# Test
assert true_versions == versions
def test_tts_engines_get_engine_existing() -> None:
"""TTSEngineManager.get_engine() で登録済み TTS エンジンを取得できる。"""
# Inputs
tts_engines = TTSEngineManager()
tts_engine1 = MockTTSEngine()
tts_engine2 = MockTTSEngine()
tts_engines.register_engine(tts_engine1, "0.0.1")
tts_engines.register_engine(tts_engine2, "0.0.2")
# Expects
true_acquired_tts_engine = tts_engine2
# Outputs
acquired_tts_engine = tts_engines.get_engine("0.0.2")
# Test
assert true_acquired_tts_engine == acquired_tts_engine
def test_tts_engines_get_engine_latest() -> None:
"""TTSEngineManager.get_engine(LATEST_VERSION) で最新版の TTS エンジンを取得できる。"""
# Inputs
tts_engines = TTSEngineManager()
tts_engine1 = MockTTSEngine()
tts_engine2 = MockTTSEngine()
tts_engine3 = MockTTSEngine()
tts_engines.register_engine(tts_engine1, "0.0.1")
tts_engines.register_engine(tts_engine2, "0.0.2")
tts_engines.register_engine(tts_engine3, "0.1.0")
# Expects
true_acquired_tts_engine = tts_engine3
# Outputs
acquired_tts_engine = tts_engines.get_engine(LATEST_VERSION)
# Test
assert true_acquired_tts_engine == acquired_tts_engine
def test_tts_engines_get_engine_missing() -> None:
"""TTSEngineManager.get_engine() で存在しない TTS エンジンを取得しようとするとエラーになる。"""
# Inputs
tts_engines = TTSEngineManager()
tts_engine1 = MockTTSEngine()
tts_engine2 = MockTTSEngine()
tts_engines.register_engine(tts_engine1, "0.0.1")
tts_engines.register_engine(tts_engine2, "0.0.2")
# Test
with pytest.raises(HTTPException) as _:
tts_engines.get_engine("0.0.3")
|