tests/test_embeddings.py

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)