tests/python/unit/test_tts.py

import pytest

from omniff.models.tts import TTSModel


def test_tts_interface():
    model = TTSModel(model_id="suno/bark-small")
    assert not model.is_loaded
    assert model.model_id == "suno/bark-small"


def test_tts_infer_not_loaded():
    model = TTSModel()
    with pytest.raises(RuntimeError, match="not loaded"):
        model.infer({"text": "hello"})