ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-21 21:14:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update test_tool_manager.py

Browse source
This commit is contained in:
Jeremiah Lowin 2025-06-04 10:24:03 -04:00
commit 6014e880ef
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/tools/test_tool_manager.py
View file

@ -693,6 +693,7 @@ class TestCustomToolNames:
# The tool is stored under the custom name and its .name is also set to custom_name
assert manager.get_tool("custom_name") is not None
assert tool.name == "custom_name"
assert isinstance(tool, FunctionTool)
assert tool.fn.__name__ == "original_fn"
# The tool should not be accessible via its original function name
with pytest.raises(NotFoundError, match="Unknown tool: original_fn"):
@ -764,6 +765,7 @@ class TestCustomToolNames:
assert stored_tool.name == "test_tool"
# But the function is different
assert isinstance(stored_tool, FunctionTool)
assert stored_tool.fn.__name__ == "replacement_fn"