updated tests to use snapshot

pydantic · Feb 12, 2025 · a0b36dd · a0b36dd
1 parent 6547863
commit a0b36dd
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/tests/test_tools.py b/tests/test_tools.py
@@ -358,8 +358,14 @@ def y_tool(y: str) -> None:
     agent = Agent('test', tools=[Tool(x_tool, function_schema=y_fs)])
 
     # make sure the function schema for y_tool is used instead of the default of x_tool.
-    assert 'x' not in agent._function_tools['x_tool']._parameters_json_schema['properties']
-    assert 'y' in agent._function_tools['x_tool']._parameters_json_schema['properties']
+    assert agent._function_tools['x_tool']._parameters_json_schema == snapshot(
+        {
+            'additionalProperties': False,
+            'properties': {'y': {'title': 'Y', 'type': 'string'}},
+            'required': ['y'],
+            'type': 'object',
+        }
+    )
 
 
 def test_init_tool_ctx_with_function_schema():
@@ -374,9 +380,14 @@ def y_tool(ctx: RunContext[int], y: str) -> None:
     )
     agent = Agent('test', tools=[Tool(x_tool, function_schema=y_fs, takes_ctx=True)], deps_type=int)
 
-    # make sure the function schema for y_tool is used instead of the default of x_tool.
-    assert 'x' not in agent._function_tools['x_tool']._parameters_json_schema['properties']
-    assert 'y' in agent._function_tools['x_tool']._parameters_json_schema['properties']
+    assert agent._function_tools['x_tool']._parameters_json_schema == snapshot(
+        {
+            'additionalProperties': False,
+            'properties': {'y': {'title': 'Y', 'type': 'string'}},
+            'required': ['y'],
+            'type': 'object',
+        }
+    )
 
 
 def ctx_tool(ctx: RunContext[int], x: int) -> int: