WorldDisasterLM-8B / tests /test_risk_engine.py
drdeveloper88's picture
Upload WorldDisasterLM-8B source code: FastAPI backend, training pipeline, 11-language support
495526b
Raw
History Blame Contribute Delete
300 Bytes
from backend.app.services.risk_engine import compute_risk
def test_compute_risk_range() -> None:
response = compute_risk("flood", vulnerability_index=0.7, exposure_index=0.8)
assert 0.0 <= response.risk_score <= 1.0
assert response.risk_level in {"low", "moderate", "high", "critical"}