File size: 1,353 Bytes
b0d0019 | 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 | """
Tests for model loader singleton behaviour.
"""
import pytest
from unittest.mock import patch, MagicMock
@patch("app.services.model_loader.LLM")
def test_get_llm_gemini_singleton(mock_llm_cls):
"""get_llm_gemini() must return the same instance on repeated calls."""
# Clear lru_cache between test runs
from app.services.model_loader import get_llm_gemini
get_llm_gemini.cache_clear()
mock_llm_cls.return_value = MagicMock(name="GeminiLLM")
first = get_llm_gemini()
second = get_llm_gemini()
assert first is second
assert mock_llm_cls.call_count == 1
@patch("app.services.model_loader.LLM")
def test_get_llm_groq_singleton(mock_llm_cls):
from app.services.model_loader import get_llm_groq
get_llm_groq.cache_clear()
mock_llm_cls.return_value = MagicMock(name="GroqLLM")
first = get_llm_groq()
second = get_llm_groq()
assert first is second
assert mock_llm_cls.call_count == 1
@patch("app.services.model_loader.LLM")
def test_get_llm_open_router_singleton(mock_llm_cls):
from app.services.model_loader import get_llm_open_router
get_llm_open_router.cache_clear()
mock_llm_cls.return_value = MagicMock(name="OpenRouterLLM")
first = get_llm_open_router()
second = get_llm_open_router()
assert first is second
assert mock_llm_cls.call_count == 1
|