add in more role tests

This commit is contained in:
Brandon Hancock
2025-01-09 16:04:48 -05:00
parent 8833ccc123
commit 0b70b3c8b4

View File

@@ -16,15 +16,15 @@ class InternalAgentTool(BaseAgentTool):
@pytest.mark.parametrize(
"role_name,should_match",
[
# ("Futel Official Infopoint", True), # exact match
("Futel Official Infopoint", True), # exact match
# (' "Futel Official Infopoint" ', True), # extra quotes and spaces
# ("Futel Official Infopoint\n", True), # trailing newline
# ('"Futel Official Infopoint"', True), # embedded quotes
# (" FUTEL\nOFFICIAL INFOPOINT ", True), # multiple whitespace and newline
# ("futel official infopoint", True), # lowercase
# ("FUTEL OFFICIAL INFOPOINT", True), # uppercase
("futel official infopoint", True), # lowercase
("FUTEL OFFICIAL INFOPOINT", True), # uppercase
("Non Existent Agent", False), # non-existent agent
# (None, False), # None agent name
(None, False), # None agent name
],
)
def test_agent_tool_role_matching(role_name, should_match):