tests/test_warmup.py

import os

import pytest

from src.embedding.embedding_service import EmbeddingService


@pytest.mark.integration
def test_embedding_warmup_basic():
    svc = EmbeddingService()
    emb = svc.embed_text("warmup")
    assert isinstance(emb, list)
    assert len(emb) > 10  # minimal dimensionality sanity check


@pytest.mark.integration
def test_embedding_warmup_quantized_toggle():
    # Ensure toggle behavior doesn't raise. We don't assert dimension to avoid coupling.
    original = os.environ.get("EMBEDDING_USE_QUANTIZED")
    try:
        os.environ["EMBEDDING_USE_QUANTIZED"] = "1"
        svc_q = EmbeddingService()
        emb_q = svc_q.embed_text("warmup")
        assert isinstance(emb_q, list)
        os.environ["EMBEDDING_USE_QUANTIZED"] = "0"
        svc_orig = EmbeddingService()
        emb_orig = svc_orig.embed_text("warmup")
        assert isinstance(emb_orig, list)
    finally:
        if original is not None:
            os.environ["EMBEDDING_USE_QUANTIZED"] = original
        else:
            os.environ.pop("EMBEDDING_USE_QUANTIZED", None)