test: fix ruff issues

Author: didier-durand

tests/server/test_id_generator.py

@@ -1,20 +1,27 @@
 """Tests for IDGenerator abstract base class."""
 import uuid
+
 from unittest.mock import patch
 
 import pytest
 
-from a2a.server.id_generator import IDGeneratorContext, IDGenerator, UUIDGenerator
+from pydantic import ValidationError
+
+from a2a.server.id_generator import (
+    IDGenerator,
+    IDGeneratorContext,
+    UUIDGenerator,
+)
 
 
 class TestIDGeneratorContext:
     """Tests for IDGeneratorContext."""
 
     def test_context_creation_with_all_fields(self):
         """Test creating context with all fields populated."""
-        context = IDGeneratorContext(task_id="task_123", context_id="context_456")
-        assert context.task_id == "task_123"
-        assert context.context_id == "context_456"
+        context = IDGeneratorContext(task_id='task_123', context_id='context_456')
+        assert context.task_id == 'task_123'
+        assert context.context_id == 'context_456'
 
     def test_context_creation_with_defaults(self):
         """Test creating context with default None values."""
@@ -24,20 +31,18 @@ def test_context_creation_with_defaults(self):
 
     def test_context_creation_with_partial_fields(self):
         """Test creating context with only some fields populated."""
-        context = IDGeneratorContext(task_id="task_123")
-        assert context.task_id == "task_123"
+        context = IDGeneratorContext(task_id='task_123')
+        assert context.task_id == 'task_123'
         assert context.context_id is None
 
     def test_context_mutability(self):
         """Test that context fields can be updated (Pydantic models are mutable by default)."""
-        context = IDGeneratorContext(task_id="task_123")
-        context.task_id = "task_456"
-        assert context.task_id == "task_456"
+        context = IDGeneratorContext(task_id='task_123')
+        context.task_id = 'task_456'
+        assert context.task_id == 'task_456'
 
     def test_context_validation(self):
         """Test that context raises validation error for invalid types."""
-        from pydantic import ValidationError  # pylint: disable=C0415
-
         with pytest.raises(ValidationError):
             IDGeneratorContext(task_id={"not": "a string"})  # noqa
 
@@ -48,26 +53,26 @@ class TestIDGenerator:
     def test_cannot_instantiate_abstract_class(self):
         """Test that IDGenerator cannot be instantiated directly."""
         with pytest.raises(TypeError):
-            IDGenerator()  # noqa pylint: disable=E0110
+            IDGenerator()
 
     def test_subclass_must_implement_generate(self):
         """Test that subclasses must implement the generate method."""
 
-        class IncompleteGenerator(IDGenerator):  # noqa pylint: disable=C0115,R0903
+        class IncompleteGenerator(IDGenerator):
             pass
 
         with pytest.raises(TypeError):
-            IncompleteGenerator()  # noqa pylint: disable=E0110
+            IncompleteGenerator()
 
     def test_valid_subclass_implementation(self):
         """Test that a valid subclass can be instantiated."""
 
         class ValidGenerator(IDGenerator):  # pylint: disable=C0115,R0903
             def generate(self, context: IDGeneratorContext) -> str:
-                return "test_id"
+                return 'test_id'
 
         generator = ValidGenerator()
-        assert generator.generate(IDGeneratorContext()) == "test_id"
+        assert generator.generate(IDGeneratorContext()) == 'test_id'
 
 
 class TestUUIDGenerator:
@@ -98,8 +103,8 @@ def test_generate_returns_uuid_version_4(self):
     def test_generate_ignores_context(self):
         """Test that generate ignores the context parameter."""
         generator = UUIDGenerator()
-        context1 = IDGeneratorContext(task_id="task_1", context_id="context_1")
-        context2 = IDGeneratorContext(task_id="task_2", context_id="context_2")
+        context1 = IDGeneratorContext(task_id='task_1', context_id='context_1')
+        context2 = IDGeneratorContext(task_id='task_2', context_id='context_2')
         result1 = generator.generate(context1)
         result2 = generator.generate(context2)
         uuid.UUID(result1)
@@ -125,7 +130,7 @@ def test_generate_with_empty_context(self):
     def test_generate_with_populated_context(self):
         """Test that generate works with a populated context."""
         generator = UUIDGenerator()
-        context = IDGeneratorContext(task_id="task_123", context_id="context_456")
+        context = IDGeneratorContext(task_id='task_123', context_id='context_456')
         result = generator.generate(context)
         assert isinstance(result, str)
         uuid.UUID(result)