| """Integration test for POST /fusion/predict.""" |
| from __future__ import annotations |
|
|
| from fastapi.testclient import TestClient |
|
|
| from src.api.main import app |
|
|
|
|
| client = TestClient(app) |
|
|
|
|
| class TestFusionRoute: |
| def test_happy_path_mri_only(self) -> None: |
| body = { |
| "mri": { |
| "label_text": "alzheimers", |
| "label": 1, |
| "confidence": 0.88, |
| "probabilities": [ |
| {"label_text": "control", "probability": 0.12}, |
| {"label_text": "alzheimers", "probability": 0.88}, |
| ], |
| }, |
| } |
| r = client.post("/fusion/predict", json=body) |
| assert r.status_code == 200, r.text |
| data = r.json() |
| assert "diseases" in data |
| assert any(d["disease"] == "alzheimers" for d in data["diseases"]) |
| assert data["top_disease"] in {"alzheimers", "parkinsons", "other"} |
|
|
| def test_empty_input_returns_baseline(self) -> None: |
| r = client.post("/fusion/predict", json={}) |
| assert r.status_code == 200 |
| data = r.json() |
| for d in data["diseases"]: |
| assert abs(d["probability"] - 0.5) < 1e-6 |
| assert "mri" in data["missing_inputs"] |
| assert data["top_disease"] is None |
|
|
| def test_invalid_probability_returns_422(self) -> None: |
| body = { |
| "mri": { |
| "label_text": "x", |
| "label": 0, |
| "confidence": 1.5, |
| "probabilities": [{"label_text": "x", "probability": 1.5}], |
| }, |
| } |
| r = client.post("/fusion/predict", json=body) |
| assert r.status_code == 422 |
|
|
