Jeremiah Lowin commited on
Commit
ce89f2f
·
1 Parent(s): 7213715

Update test_tool_manager.py

Browse files
Files changed (1) hide show
  1. tests/tools/test_tool_manager.py +2 -0
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