Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
ce89f2f
1
Parent(s): 7213715
Update test_tool_manager.py
Browse files
tests/tools/test_tool_manager.py
CHANGED
|
@@ -693,6 +693,7 @@ class TestCustomToolNames:
|
|
| 693 |
# The tool is stored under the custom name and its .name is also set to custom_name
|
| 694 |
assert manager.get_tool("custom_name") is not None
|
| 695 |
assert tool.name == "custom_name"
|
|
|
|
| 696 |
assert tool.fn.__name__ == "original_fn"
|
| 697 |
# The tool should not be accessible via its original function name
|
| 698 |
with pytest.raises(NotFoundError, match="Unknown tool: original_fn"):
|
|
@@ -764,6 +765,7 @@ class TestCustomToolNames:
|
|
| 764 |
assert stored_tool.name == "test_tool"
|
| 765 |
|
| 766 |
# But the function is different
|
|
|
|
| 767 |
assert stored_tool.fn.__name__ == "replacement_fn"
|
| 768 |
|
| 769 |
|
|
|
|
| 693 |
# The tool is stored under the custom name and its .name is also set to custom_name
|
| 694 |
assert manager.get_tool("custom_name") is not None
|
| 695 |
assert tool.name == "custom_name"
|
| 696 |
+
assert isinstance(tool, FunctionTool)
|
| 697 |
assert tool.fn.__name__ == "original_fn"
|
| 698 |
# The tool should not be accessible via its original function name
|
| 699 |
with pytest.raises(NotFoundError, match="Unknown tool: original_fn"):
|
|
|
|
| 765 |
assert stored_tool.name == "test_tool"
|
| 766 |
|
| 767 |
# But the function is different
|
| 768 |
+
assert isinstance(stored_tool, FunctionTool)
|
| 769 |
assert stored_tool.fn.__name__ == "replacement_fn"
|
| 770 |
|
| 771 |
|