Files
t2a_v2/tests/test_severity.py
dom 9df4465fef feat: règles métier T2A Phase 1 — exclusions diagnostiques, sévérité CMA et alertes codage
Ajout des règles d'exclusion symptôme (R00-R99) vs diagnostic précis (Chapitres I-XIV),
détection heuristique de sévérité CMA sur 25 racines CIM-10, et affichage des alertes
de codage dans le viewer Flask. 153 tests, 0 régression.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-11 08:53:14 +01:00

110 lines
3.8 KiB
Python

"""Tests pour le module de sévérité heuristique (CMA/CMS)."""
import pytest
from src.config import Diagnostic
from src.medical.severity import (
evaluate_severity,
enrich_dossier_severity,
_detect_severity_markers,
_is_heuristic_cma,
)
class TestDetectSeverityMarkers:
def test_severe_markers(self):
niveau, marqueurs = _detect_severity_markers("Pancréatite aiguë sévère")
assert niveau == "severe"
assert any(m in ("severe", "aigue") for m in marqueurs)
def test_moderate_markers(self):
niveau, marqueurs = _detect_severity_markers("Insuffisance rénale modérée")
assert niveau == "modere"
assert "modere" in marqueurs or "moderee" in marqueurs
def test_mild_markers(self):
niveau, marqueurs = _detect_severity_markers("Anémie chronique bénigne")
assert niveau == "leger"
assert any(m in ("chronique", "benin", "benigne") for m in marqueurs)
def test_no_markers(self):
niveau, marqueurs = _detect_severity_markers("Hypertension artérielle")
assert niveau == "non_evalue"
assert marqueurs == []
def test_severe_overrides_mild(self):
"""Si 'sévère' et 'chronique' sont présents, 'severe' l'emporte."""
niveau, marqueurs = _detect_severity_markers("Insuffisance cardiaque chronique décompensée")
assert niveau == "severe"
class TestHeuristicCMA:
def test_e11_is_cma(self):
assert _is_heuristic_cma("E11.9") is True
def test_i48_is_cma(self):
assert _is_heuristic_cma("I48.9") is True
def test_a41_is_cma(self):
assert _is_heuristic_cma("A41.9") is True
def test_k85_not_cma(self):
assert _is_heuristic_cma("K85.1") is False
def test_i10_not_cma(self):
assert _is_heuristic_cma("I10") is False
def test_empty(self):
assert _is_heuristic_cma("") is False
assert _is_heuristic_cma(None) is False
class TestEvaluateSeverity:
def test_cma_code_detected(self):
diag = Diagnostic(texte="Diabète type 2", cim10_suggestion="E11.9")
info = evaluate_severity(diag)
assert info.est_cma_probable is True
def test_non_cma_code(self):
diag = Diagnostic(texte="Pancréatite aiguë biliaire", cim10_suggestion="K85.1")
info = evaluate_severity(diag)
assert info.est_cma_probable is False
def test_severity_from_text(self):
diag = Diagnostic(texte="Sepsis sévère", cim10_suggestion="A41.9")
info = evaluate_severity(diag)
assert info.niveau_severite == "severe"
assert info.est_cma_probable is True
def test_combined_text_and_dict_label(self):
"""Le label du dictionnaire CIM-10 enrichit la détection de sévérité."""
diag = Diagnostic(texte="Embolie pulmonaire", cim10_suggestion="I26.9")
info = evaluate_severity(diag)
assert info.est_cma_probable is True
class TestEnrichDossierSeverity:
def test_enriches_das_in_place(self):
dp = Diagnostic(texte="Pancréatite aiguë biliaire", cim10_suggestion="K85.1")
das = [
Diagnostic(texte="Fibrillation auriculaire", cim10_suggestion="I48.9"),
Diagnostic(texte="Obésité", cim10_suggestion="E66.0"),
]
alertes = enrich_dossier_severity(dp, das)
# I48.9 = CMA probable
assert das[0].est_cma is True
assert das[0].niveau_severite is not None
# E66.0 = non CMA
assert das[1].est_cma is None
# Au moins une alerte CMA
assert any("CMA" in a for a in alertes)
def test_dp_severity_set(self):
dp = Diagnostic(texte="Sepsis sévère", cim10_suggestion="A41.9")
alertes = enrich_dossier_severity(dp, [])
assert dp.niveau_severite == "severe"
assert dp.est_cma is True