Moved test files from agentic_security.probe_actor to tests.probe_actor

This commit is contained in:
doa
2025-03-08 20:18:54 +03:00
parent d20c1a3d0d
commit 21b43b18e7
2 changed files with 0 additions and 0 deletions
+13
View File
@@ -0,0 +1,13 @@
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)