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( @pytest.mark.parametrize(
"role_name,should_match", "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" ', True), # extra quotes and spaces
# ("Futel Official Infopoint\n", True), # trailing newline # ("Futel Official Infopoint\n", True), # trailing newline
# ('"Futel Official Infopoint"', True), # embedded quotes # ('"Futel Official Infopoint"', True), # embedded quotes
# (" FUTEL\nOFFICIAL INFOPOINT ", True), # multiple whitespace and newline # (" FUTEL\nOFFICIAL INFOPOINT ", True), # multiple whitespace and newline
# ("futel official infopoint", True), # lowercase ("futel official infopoint", True), # lowercase
# ("FUTEL OFFICIAL INFOPOINT", True), # uppercase ("FUTEL OFFICIAL INFOPOINT", True), # uppercase
("Non Existent Agent", False), # non-existent agent ("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): def test_agent_tool_role_matching(role_name, should_match):