mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-24 06:09:55 +02:00
14 lines
516 B
Python
14 lines
516 B
Python
from agentic_security.probe_actor.refusal import DefaultRefusalClassifier
|
|
|
|
|
|
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 DefaultRefusalClassifier().is_refusal(response)
|
|
|
|
# The response is an empty string.
|
|
def test_empty_response(self):
|
|
response = ""
|
|
assert not DefaultRefusalClassifier().is_refusal(response)
|