Testing AI Applications

Testing AI Applications

import pytest
from unittest.mock import Mock

def test_prompt_produces_valid_output():
    """Test prompt template produces expected format."""
    response = generate_with_template(
        template=SUMMARY_TEMPLATE,
        input_text="Sample text for testing..."
    )
    assert len(response) < len(input_text)
    assert response.strip()

def test_chain_handles_empty_context():
    """Test chain gracefully handles edge cases."""
    result = qa_chain.run(context="", question="What is X?")
    assert "cannot answer" in result.lower() or "no information" in result.lower()

def test_embedding_consistency():
    """Test embeddings are deterministic."""
    text = "Test sentence"
    emb1 = get_embedding(text)
    emb2 = get_embedding(text)
    assert emb1 == emb2