Spaces:
Running on Zero
Running on Zero
File size: 646 Bytes
20db40d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | from fastapi.testclient import TestClient
from api.main import app
client = TestClient(app)
def test_health_endpoint():
response = client.get("/health")
assert response.status_code == 200
assert response.json() == {"status": "healthy"}
def test_generate_endpoint():
response = client.post(
"/generate",
json={"prompt": "Create a bar chart"},
)
assert response.status_code == 200
assert "matplotlib" in response.json()["generated_code"]
def test_generate_endpoint_rejects_blank_prompt():
response = client.post("/generate", json={"prompt": " "})
assert response.status_code == 400
|