34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
from augmentation import CounselorAugmentationEngine
|
|
|
|
|
|
def test_explicit_intent_forces_critical_sidebar_guidance():
|
|
engine = CounselorAugmentationEngine()
|
|
result = engine.build_augmented_guidance(
|
|
"I want to kill myself tonight. I already wrote a note.",
|
|
assistant_text="I'm here with you."
|
|
)
|
|
|
|
assert result.risk_level == "CRITICAL"
|
|
assert result.risk_score >= 90
|
|
assert result.local_only is True
|
|
assert result.advisory_only is True
|
|
assert "Explicit self-harm intent" in result.signals
|
|
assert result.suggested_talking_points
|
|
assert result.deescalation_techniques
|
|
assert "You said" in result.follow_up_prompt
|
|
assert "never replaces human judgment" in result.operator_notice.lower()
|
|
|
|
|
|
def test_hopelessness_signal_produces_follow_up_and_talking_points():
|
|
engine = CounselorAugmentationEngine()
|
|
result = engine.build_augmented_guidance(
|
|
"I feel so hopeless about my life and I can't go on.",
|
|
assistant_text=""
|
|
)
|
|
|
|
assert result.risk_level in {"HIGH", "CRITICAL"}
|
|
assert result.signals
|
|
assert result.suggested_talking_points
|
|
assert result.deescalation_techniques
|
|
assert result.follow_up_prompt
|