File size: 4,075 Bytes
327b23d | 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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | """Integration tests for /predict/bbb_permeability_map and /research/drug_dose_adjustment."""
from __future__ import annotations
from pathlib import Path
import numpy as np
import pytest
from fastapi.testclient import TestClient
from PIL import Image
from src.api.main import app
from tests.fixtures.build_dummy_resnet18_2d import build as build_dummy_2d
def _png(path: Path) -> Path:
arr = (np.random.RandomState(0).rand(170, 170, 3) * 255).astype(np.uint8)
Image.fromarray(arr, mode="RGB").save(str(path))
return path
@pytest.fixture()
def client_proxy(monkeypatch, tmp_path):
ckpt = build_dummy_2d(tmp_path / "best.pt")
monkeypatch.setenv("MRI_MODEL_PATH_2D", str(ckpt))
return TestClient(app), tmp_path
class TestBBBPermeabilityMapRoute:
def test_heuristic_proxy_happy_path(self, client_proxy) -> None:
client, tmp_path = client_proxy
img = _png(tmp_path / "scan.png")
r = client.post(
"/predict/bbb_permeability_map",
json={"input_path": str(img), "mode": "heuristic_proxy"},
)
assert r.status_code == 200, r.text
data = r.json()
assert 0.0 <= data["permeability_score"] <= 1.0
assert data["interpretation"] in {
"BBB intact", "mild leakage", "moderate leakage", "severe leakage",
}
assert data["method"] == "heuristic_proxy"
assert data["voxel_map_available"] is False
def test_unknown_mode_returns_400(self, client_proxy) -> None:
client, tmp_path = client_proxy
img = _png(tmp_path / "scan.png")
r = client.post(
"/predict/bbb_permeability_map",
json={"input_path": str(img), "mode": "bogus_mode"},
)
assert r.status_code == 400
def test_missing_input_returns_404(self, client_proxy) -> None:
client, tmp_path = client_proxy
r = client.post(
"/predict/bbb_permeability_map",
json={"input_path": str(tmp_path / "missing.png"), "mode": "heuristic_proxy"},
)
assert r.status_code == 404
class TestDrugDoseAdjustmentRoute:
def test_intact_bbb_returns_baseline(self) -> None:
client = TestClient(app)
r = client.post("/research/drug_dose_adjustment", json={
"baseline_dose_mg": 100.0,
"bbb_permeability_score": 0.05,
"drug_bbb_permeable": True,
})
assert r.status_code == 200, r.text
data = r.json()
assert data["recommended_dose_mg"] == pytest.approx(100.0)
assert data["risk_level"] == "low"
def test_leaky_bbb_permeable_drug_reduced(self) -> None:
client = TestClient(app)
r = client.post("/research/drug_dose_adjustment", json={
"baseline_dose_mg": 100.0,
"bbb_permeability_score": 0.5,
"drug_bbb_permeable": True,
})
assert r.status_code == 200
data = r.json()
assert data["recommended_dose_mg"] == pytest.approx(65.0)
assert data["risk_level"] == "moderate"
assert "research suggestion" in data["rationale"].lower()
def test_severe_leakage_high_risk(self) -> None:
client = TestClient(app)
r = client.post("/research/drug_dose_adjustment", json={
"baseline_dose_mg": 100.0,
"bbb_permeability_score": 0.85,
"drug_bbb_permeable": True,
})
data = r.json()
assert data["risk_level"] == "high"
def test_negative_baseline_returns_422(self) -> None:
client = TestClient(app)
r = client.post("/research/drug_dose_adjustment", json={
"baseline_dose_mg": -1.0,
"bbb_permeability_score": 0.5,
})
assert r.status_code == 422 # pydantic gt=0.0 validation
def test_score_out_of_range_returns_422(self) -> None:
client = TestClient(app)
r = client.post("/research/drug_dose_adjustment", json={
"baseline_dose_mg": 100.0,
"bbb_permeability_score": 1.5,
})
assert r.status_code == 422
|