| import pytest | |
| from spi.renderers.mock import MockVideoRenderer | |
| from spi.renderers.registry import ModelRouter | |
| def test_unknown_mode_rejected(): | |
| with pytest.raises(ValueError): | |
| ModelRouter("hyper") | |
| def test_routes_mock_video(): | |
| r = ModelRouter("mock") | |
| renderer = r.route("video") | |
| assert isinstance(renderer, MockVideoRenderer) | |
| caps = renderer.capabilities() | |
| assert caps["vram_gb"] == 0 and "text_to_video" in caps["tasks"] | |
| def test_register_and_prefer(): | |
| r = ModelRouter("mock") | |
| class MyRenderer(MockVideoRenderer): | |
| name = "my_renderer" | |
| r.register("video", MyRenderer()) | |
| assert r.route("video", prefer="my_renderer").name == "my_renderer" | |
| assert r.list_renderers("video")[0]["name"] in ("mock_video", "my_renderer") | |
| def test_no_lane_raises(): | |
| r = ModelRouter("mock") | |
| with pytest.raises(Exception): | |
| r.route("threed") |