File size: 1,489 Bytes
f9569f4 | 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 | import pytest
from server.models.base import (
Lang,
ModelAdapter,
ParamSpec,
is_valid_adapter,
)
class FakeOk(ModelAdapter):
id = "fake-ok"
label = "Fake OK"
description = "Test"
languages = [Lang(code="en", label="English")]
paralinguistic_tags: list[str] = []
supports_voice_clone = True
params = [ParamSpec(name="t", label="T", type="float", default=0.5, min=0.0, max=1.0)]
def __init__(self, device: str): self.device = device
def load(self): ...
def unload(self): ...
def generate(self, text, reference_wav_path, language, params):
return (b"fake", 24000)
def test_is_valid_adapter_accepts_fake():
assert is_valid_adapter(FakeOk) is True
def test_is_valid_adapter_rejects_missing_id():
class Bad(ModelAdapter):
id = ""
label = "X"
description = "X"
languages: list[Lang] = []
paralinguistic_tags: list[str] = []
supports_voice_clone = False
params: list[ParamSpec] = []
def __init__(self, device): ...
def load(self): ...
def unload(self): ...
def generate(self, *a, **k): return (b"", 0)
assert is_valid_adapter(Bad) is False
def test_param_spec_defaults_validated():
with pytest.raises(ValueError):
ParamSpec(name="t", label="T", type="float", default=2.0, min=0.0, max=1.0)
def test_lang_dataclass():
l = Lang(code="hi", label="Hindi")
assert (l.code, l.label) == ("hi", "Hindi")
|