| import unittest | |
| from erniebot_agent.tools.calculator_tool import CalculatorTool | |
| class TestCalculatorTool(unittest.TestCase): | |
| def test_schema(self): | |
| calculator = CalculatorTool() | |
| function_call_schema = calculator.function_call_schema() | |
| self.assertEqual(function_call_schema["description"], "CalculatorTool用于执行数学公式计算") | |
| self.assertIn("math_formula", function_call_schema["parameters"]["properties"]) | |
| self.assertEqual(function_call_schema["parameters"]["properties"]["math_formula"]["type"], "string") | |
| self.assertEqual(function_call_schema["parameters"]["properties"]["math_formula"]["type"], "string") | |
| self.assertEqual( | |
| function_call_schema["responses"]["properties"]["formula_result"]["type"], | |
| "number", | |
| ) | |