Files
agentic_security/langalf/probe_actor/test_refusal.py
T
Alexander Myasoedov 15106c1c23 feat(Init):
2024-04-13 18:04:24 +03:00

15 lines
449 B
Python

from langalf.probe_actor.refusal import check_refusal
class TestCheckRefusal:
# The function correctly identifies a refusal phrase in the response.
def test_identify_refusal_phrase(self):
response = "I'm sorry, but I cannot provide that information."
assert check_refusal(response)
# The response is an empty string.
def test_empty_response(self):
response = ""
assert not check_refusal(response)