tests/test_api.py

import requests

BASE_URL = "http://localhost:8000"

def test_generate():
    resp = requests.post(f"{BASE_URL}/generate", json={
        "n_samples": 2,
        "temperature": 1.0,
        "random_seed": 42
    })
    print("/generate status:", resp.status_code)
    print("/generate response:", resp.json())

def test_encode():
    resp = requests.post(f"{BASE_URL}/encode", json={
        "age": 45,
        "gender": "Male",
        "diagnosis": "Diabetes",
        "blood_type": "A+",
        "length_of_stay": 7
    })
    print("/encode status:", resp.status_code)
    print("/encode response:", resp.json())

if __name__ == "__main__":
    test_generate()
    test_encode()