Formatting

tests/test_registry_edge_cases.py

@@ -1,6 +1,7 @@
 """Edge case tests for tool registry."""
+
+
 import pytest
-from unittest.mock import Mock
 
 from agent.registry import Tool, make_tools
 
@@ -182,7 +183,9 @@ class TestMakeToolsEdgeCases:
             params = tool.parameters
             if "required" in params and "properties" in params:
                 for req in params["required"]:
-                    assert req in params["properties"], f"Required param {req} not in properties for {tool.name}"
+                    assert (
+                        req in params["properties"]
+                    ), f"Required param {req} not in properties for {tool.name}"
 
     def test_make_tools_descriptions_not_empty(self, memory):
         """Should have non-empty descriptions."""
@@ -233,7 +236,9 @@ class TestMakeToolsEdgeCases:
             if "properties" in tool.parameters:
                 for prop_name, prop_schema in tool.parameters["properties"].items():
                     if "type" in prop_schema:
-                        assert prop_schema["type"] in valid_types, f"Invalid type for {tool.name}.{prop_name}"
+                        assert (
+                            prop_schema["type"] in valid_types
+                        ), f"Invalid type for {tool.name}.{prop_name}"
 
     def test_make_tools_enum_values(self, memory):
         """Should have valid enum values."""