governance-risk-classifier / tests /test_inference.py
MonikaDvorackova's picture
Add governance risk inference and tests
0f16e07 unverified
Raw
History Blame Contribute Delete
1.01 kB
import sys
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))
from inference import predict_governance_risk
def test_high_risk_scenario():
result = predict_governance_risk({
"sector": "financial_services",
"impact": "high",
"decision_autonomy": "automated",
"human_oversight": "none",
"monitoring": "none",
"traceability": "none",
"technical_documentation": "partial",
})
assert result["risk_tier"] == "unacceptable"
assert "class_probabilities" in result
def test_lower_risk_scenario():
result = predict_governance_risk({
"sector": "enterprise_productivity",
"impact": "low",
"decision_autonomy": "assistive",
"human_oversight": "strong",
"monitoring": "strong",
"traceability": "strong",
"technical_documentation": "strong",
})
assert result["risk_tier"] == "lower"
assert "class_probabilities" in result