test_embedding_generator.py

from embedding_generator import EmbeddingGenerator

def test_embedding_generator():
    embedding_generator = EmbeddingGenerator()
    
    # Test generating embeddings
    embeddings = embedding_generator.generate("test query")
    assert isinstance(embeddings, list)
    assert len(embeddings) == 768