tests/test_api.py

from fastapi.testclient import TestClient
from src.main import app

client = TestClient(app)

def test_coding_endpoint():
    provider_notes = {
        "notes": "Patient has a fever and cough."
    }
    
    response = client.post("/api/coding", json=provider_notes)
    
    assert response.status_code == 200
    data = response.json()
    
    assert "CPT" in data
    assert "ICD" in data
    assert isinstance(data["CPT"], list)
    assert isinstance(data["ICD"], list)
    assert "explanation" in data["CPT"][0]
    assert "explanation" in data["ICD"][0]