tests/test_api.py

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

client = TestClient(app)

def test_root():
    response = client.get("/")
    assert response.status_code == 200
    assert response.json() == {"response":"Hello"}


def test_context_analysis():
    payload = {
        "message": "Hello"
    }

    response = client.post("/context-analysis", json=payload)

    assert response.status_code == 200
    assert response.json()