import pytest
from shell_history_search.embeddings import EmbeddingService


class TestEmbeddingService:
    @pytest.fixture
    def service(self):
        return EmbeddingService()

    def test_get_embedding(self, service):
        embedding = service.get_embedding("test command")
        assert isinstance(embedding, list)
        assert len(embedding) == 384

    def test_get_embedding_consistency(self, service):
        emb1 = service.get_embedding("test command")
        emb2 = service.get_embedding("test command")
        assert emb1 == emb2

    def test_get_embedding_different_commands(self, service):
        emb1 = service.get_embedding("command one")
        emb2 = service.get_embedding("command two")
        assert emb1 != emb2