From 0b70b3c8b4211eae1e647b63e4cb07ff2084331a Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Thu, 9 Jan 2025 16:04:48 -0500 Subject: [PATCH] add in more role tests --- tests/test_manager_llm_delegation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_manager_llm_delegation.py b/tests/test_manager_llm_delegation.py index ff0e6a47f..5dbb90c21 100644 --- a/tests/test_manager_llm_delegation.py +++ b/tests/test_manager_llm_delegation.py @@ -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):