diff --git a/tests/unittests/tools/mcp_tool/test_mcp_toolset.py b/tests/unittests/tools/mcp_tool/test_mcp_toolset.py index 0cdb72c96b..e05c0a48a9 100644 --- a/tests/unittests/tools/mcp_tool/test_mcp_toolset.py +++ b/tests/unittests/tools/mcp_tool/test_mcp_toolset.py @@ -196,9 +196,8 @@ def test_init_with_tool_filter_list(self): connection_params=self.mock_stdio_params, tool_filter=tool_filter ) - # The tool filter is stored in the parent BaseToolset class - # We can verify it by checking the filtering behavior in get_tools - assert toolset._is_tool_selected is not None + # The tool filter is stored on the parent BaseToolset class. + assert toolset.tool_filter == tool_filter def test_init_with_auth(self): """Test initialization with authentication."""