File size: 565 Bytes
187966e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from __future__ import annotations
import pytest
@pytest.mark.slow
async def test_real_embedder_dimensions_and_determinism():
from app.embeddings import cosine, embed_query, embed_texts
vecs = await embed_texts(["hola mundo", "adios"])
assert len(vecs) == 2
assert len(vecs[0]) == 384
assert len(vecs[0]) == len(vecs[1])
q = await embed_query("hola mundo")
assert len(q) == 384
same = await embed_texts(["repetido", "repetido"])
assert same[0] == same[1]
assert cosine(vecs[0], vecs[0]) == pytest.approx(1.0, abs=1e-6)
|