medilang-tech / tests /test_api_endpoints.py
Dama03's picture
first push of the AI
411a994
raw
history blame
568 Bytes
from fastapi.testclient import TestClient
from main import app
client = TestClient(app)
def test_unified_chat_text():
r = client.post("/api/chat/unified", json={
"message": "J'ai de la fièvre et des frissons",
"message_type": "text",
"language": "fr"
})
assert r.status_code in (200, 500)
if r.status_code == 200:
data = r.json()
assert "response" in data and "context" in data
def test_cameroon_overview():
r = client.get("/api/cameroon-data/stats/overview")
assert r.status_code in (200, 500)