Spaces:
Sleeping
Sleeping
| from tests.conftest import VALID_PAYLOAD, TEST_API_KEY | |
| # Tester GET /health | |
| def test_health_check(client): | |
| response = client.get("/health") | |
| assert response.status_code == 200 | |
| assert response.json()["status"] == "ok" | |
| # Tester POST /predict avec données valides | |
| def test_predict_valid(client): | |
| response = client.post( | |
| "/predict", | |
| json=VALID_PAYLOAD, | |
| headers={"X-API-Key": TEST_API_KEY}, | |
| ) | |
| assert response.status_code == 200 | |
| body = response.json() | |
| assert "label" in body | |
| assert body["label"] in ("Reste", "Quitte") | |
| assert "probabilite" in body | |
| assert 0.0 <= body["probabilite"] <= 1.0 | |
| # Tester POST /predict avec données invalides | |
| def test_predict_invalid_data(client): | |
| invalid_payload = {"age": 999, "genre": 99} # hors bornes + champs manquants | |
| response = client.post( | |
| "/predict", | |
| json=invalid_payload, | |
| headers={"X-API-Key": TEST_API_KEY}, | |
| ) | |
| assert response.status_code == 422 # Pydantic validation error | |
| # Tester POST /predict sans API Key → 401 | |
| def test_predict_no_api_key(client): | |
| response = client.post("/predict", json=VALID_PAYLOAD) | |
| assert response.status_code == 401 | |