tests/test_embedder.py

import pytest
from ingestion.embedder import embedder


@pytest.mark.unit
class TestEmbedder:
    
    def test_embed_query(self, sample_query):
        embedding = embedder.embed_query(sample_query)
        
        assert embedding is not None
        assert len(embedding) == 384
        assert all(isinstance(x, float) for x in embedding)
    
    def test_embed_documents(self, sample_document_text):
        texts = [sample_document_text, "Another document for testing."]
        embeddings = embedder.embed_documents(texts)
        
        assert len(embeddings) == 2
        assert all(len(emb) == 384 for emb in embeddings)
        assert all(isinstance(x, float) for emb in embeddings for x in emb)
    
    def test_get_embeddings(self):
        model = embedder.get_embeddings()
        
        assert model is not None
        assert hasattr(model, 'embed_query')
        assert hasattr(model, 'embed_documents')