"""Tests for model registries."""
import pytest
from ml_server.registry import TTSRegistry, ASRRegistry, SERRegistry
def test_tts_registry_available():
"""Test listing available TTS models."""
registry = TTSRegistry()
available = registry.available()
assert isinstance(available, list)
assert "pocket" in available
def test_asr_registry_available():
"""Test listing available ASR models."""
registry = ASRRegistry()
available = registry.available()
assert isinstance(available, list)
assert "parakeet" in available
def test_ser_registry_available():
"""Test listing available SER models."""
registry = SERRegistry()
available = registry.available()
assert isinstance(available, list)
assert "sensevoice" in available
def test_registry_load_unknown():
"""Test loading unknown model raises error."""
registry = TTSRegistry()
with pytest.raises(ValueError, match="Unknown model"):
registry.load("nonexistent_model")
def test_registry_active_initially_none():
"""Test active model is None before loading."""
registry = TTSRegistry()
assert registry.active is None
assert registry.active_name is None
