Clean up tests

Author: g-eoj

pydantic_ai_slim/pydantic_ai/_output.py

@@ -434,7 +434,7 @@ def build_json_schema(
         for key in object_keys:
             count = 1
             new_key = key
-            while new_key in unique_object_keys:  # pragma: no cover
+            while new_key in unique_object_keys:
                 count += 1
                 new_key = f'{key}_{count}'
             unique_object_keys.append(new_key)

tests/test_agent_output_schemas.py

@@ -46,7 +46,7 @@ async def test_tool_output_json_schema():
 
     agent = Agent(
         'test',
-        output_type=[ToolOutput(bool, name='alice', description='Dreaming'), ToolOutput(bool, name='bob')],
+        output_type=[ToolOutput(bool, name='alice'), ToolOutput(bool, name='bob')],
     )
     assert agent.output_json_schema() == snapshot(
         {
@@ -88,63 +88,6 @@ async def test_tool_output_json_schema():
         }
     )
 
-    agent = Agent(
-        'test',
-        output_type=[ToolOutput(bool, name='alice')] * 3,
-    )
-    assert agent.output_json_schema() == snapshot(
-        {
-            'type': 'object',
-            'properties': {
-                'result': {
-                    'anyOf': [
-                        {
-                            'type': 'object',
-                            'properties': {
-                                'kind': {'type': 'string', 'const': 'alice'},
-                                'data': {
-                                    'properties': {'response': {'type': 'boolean'}},
-                                    'required': ['response'],
-                                    'type': 'object',
-                                },
-                            },
-                            'required': ['kind', 'data'],
-                            'additionalProperties': False,
-                        },
-                        {
-                            'type': 'object',
-                            'properties': {
-                                'kind': {'type': 'string', 'const': 'alice_2'},
-                                'data': {
-                                    'properties': {'response': {'type': 'boolean'}},
-                                    'required': ['response'],
-                                    'type': 'object',
-                                },
-                            },
-                            'required': ['kind', 'data'],
-                            'additionalProperties': False,
-                        },
-                        {
-                            'type': 'object',
-                            'properties': {
-                                'kind': {'type': 'string', 'const': 'alice_3'},
-                                'data': {
-                                    'properties': {'response': {'type': 'boolean'}},
-                                    'required': ['response'],
-                                    'type': 'object',
-                                },
-                            },
-                            'required': ['kind', 'data'],
-                            'additionalProperties': False,
-                        },
-                    ]
-                }
-            },
-            'required': ['result'],
-            'additionalProperties': False,
-        }
-    )
-
 
 async def test_native_output_json_schema():
     agent = Agent(
@@ -372,7 +315,7 @@ async def test_image_output_json_schema():
 async def test_override_output_json_schema():
     agent = Agent('test')
     assert agent.output_json_schema() == snapshot({'type': 'string'})
-    output_type = [ToolOutput(bool, name='alice', description='Dreaming...')]
+    output_type = [ToolOutput(bool)]
     assert agent.output_json_schema(output_type=output_type) == snapshot(
         {'type': 'object', 'properties': {'response': {'type': 'boolean'}}, 'required': ['response']}
     )