43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
"""자살사고 단계 절대 상한의 출력 fail-closed 회귀 테스트."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from .services import guardrail, state_machine
|
|
|
|
|
|
class GuardrailIdeationCapTest(unittest.TestCase):
|
|
def test_over_cap_is_blocked_and_requires_regeneration(self) -> None:
|
|
result = guardrail.sanitize_client_reply(
|
|
"요즘 버티기 어렵다는 생각이 자주 들어요.",
|
|
ideation_stage=5,
|
|
turn_seq=4,
|
|
)
|
|
|
|
self.assertTrue(result.blocked)
|
|
self.assertTrue(result.needs_regeneration)
|
|
self.assertIn("ideation_over_cap:5>3", result.reasons)
|
|
|
|
def test_at_cap_without_other_violation_is_unchanged(self) -> None:
|
|
text = "요즘 버티기 어렵다는 생각이 자주 들어요."
|
|
|
|
result = guardrail.sanitize_client_reply(
|
|
text,
|
|
ideation_stage=state_machine.IDEATION_STAGE_CAP,
|
|
turn_seq=4,
|
|
)
|
|
|
|
self.assertFalse(result.blocked)
|
|
self.assertFalse(result.needs_regeneration)
|
|
self.assertEqual(result.text, text)
|
|
self.assertEqual(result.reasons, [])
|
|
|
|
def test_guardrail_reuses_state_machine_cap_and_clamp(self) -> None:
|
|
self.assertEqual(guardrail.IDEATION_STAGE_CAP, state_machine.IDEATION_STAGE_CAP)
|
|
self.assertEqual(guardrail.clamp_ideation(5), state_machine.IDEATION_STAGE_CAP)
|
|
self.assertEqual(guardrail.clamp_ideation(0), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|