Spaces:
Sleeping
Sleeping
File size: 1,069 Bytes
159faf0 |
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 |
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)
|