tests/test_tts_asset_setup.py

from app.benchmarks.tts_asset_setup import download_specs


def test_asset_setup_contains_only_researchmate_tts_models():
    specs = download_specs()

    assert {spec.name for spec in specs} == {"pocket-tts", "piper-en-us"}


def test_asset_setup_limits_piper_to_lessac_medium():
    piper = next(spec for spec in download_specs() if spec.name == "piper-en-us")

    assert piper.allow_patterns
    assert all("en/en_US/lessac/medium/" in pattern for pattern in piper.allow_patterns)
    assert {pattern.rsplit("/", 1)[-1] for pattern in piper.allow_patterns} == {
        "en_US-lessac-medium.onnx",
        "en_US-lessac-medium.onnx.json",
        "MODEL_CARD",
    }