Spaces:
Sleeping
Sleeping
File size: 327 Bytes
10e948d |
1 2 3 4 5 6 7 8 9 10 11 |
import pytest
from app.core.embeddings import EmbeddingGenerator
def test_embedding_generation():
embedder = EmbeddingGenerator("sentence-transformers/all-MiniLM-L6-v2")
embeddings = embedder.generate(["test text"])
assert len(embeddings) == 1
assert len(embeddings[0]) == 384
# Run tests with: uv run pytest
|