Formatting

Author: holtskinner

tests/utils/test_message.py

@@ -1,13 +1,14 @@
-from unittest.mock import patch
 import uuid
 
+from unittest.mock import patch
+
 from a2a.types import (
     Message,
     Part,
     Role,
     TextPart,
 )
-from a2a.utils import new_agent_text_message, get_text_parts, get_message_text
+from a2a.utils import get_message_text, get_text_parts, new_agent_text_message
 
 
 class TestNewAgentTextMessage:
@@ -16,7 +17,10 @@ def test_new_agent_text_message_basic(self):
         text = "Hello, I'm an agent"
 
         # Exercise - with a fixed uuid for testing
-        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+        with patch(
+            'uuid.uuid4',
+            return_value=uuid.UUID('12345678-1234-5678-1234-567812345678'),
+        ):
             message = new_agent_text_message(text)
 
         # Verify
@@ -29,11 +33,14 @@ def test_new_agent_text_message_basic(self):
 
     def test_new_agent_text_message_with_context_id(self):
         # Setup
-        text = "Message with context"
-        context_id = "test-context-id"
+        text = 'Message with context'
+        context_id = 'test-context-id'
 
         # Exercise
-        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+        with patch(
+            'uuid.uuid4',
+            return_value=uuid.UUID('12345678-1234-5678-1234-567812345678'),
+        ):
             message = new_agent_text_message(text, context_id=context_id)
 
         # Verify
@@ -45,11 +52,14 @@ def test_new_agent_text_message_with_context_id(self):
 
     def test_new_agent_text_message_with_task_id(self):
         # Setup
-        text = "Message with task id"
-        task_id = "test-task-id"
+        text = 'Message with task id'
+        task_id = 'test-task-id'
 
         # Exercise
-        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+        with patch(
+            'uuid.uuid4',
+            return_value=uuid.UUID('12345678-1234-5678-1234-567812345678'),
+        ):
             message = new_agent_text_message(text, task_id=task_id)
 
         # Verify
@@ -61,13 +71,18 @@ def test_new_agent_text_message_with_task_id(self):
 
     def test_new_agent_text_message_with_both_ids(self):
         # Setup
-        text = "Message with both ids"
-        context_id = "test-context-id"
-        task_id = "test-task-id"
+        text = 'Message with both ids'
+        context_id = 'test-context-id'
+        task_id = 'test-task-id'
 
         # Exercise
-        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
-            message = new_agent_text_message(text, context_id=context_id, task_id=task_id)
+        with patch(
+            'uuid.uuid4',
+            return_value=uuid.UUID('12345678-1234-5678-1234-567812345678'),
+        ):
+            message = new_agent_text_message(
+                text, context_id=context_id, task_id=task_id
+            )
 
         # Verify
         assert message.role == Role.agent
@@ -78,42 +93,45 @@ def test_new_agent_text_message_with_both_ids(self):
 
     def test_new_agent_text_message_empty_text(self):
         # Setup
-        text = ""
+        text = ''
 
         # Exercise
-        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+        with patch(
+            'uuid.uuid4',
+            return_value=uuid.UUID('12345678-1234-5678-1234-567812345678'),
+        ):
             message = new_agent_text_message(text)
 
         # Verify
         assert message.role == Role.agent
-        assert message.parts[0].root.text == ""
+        assert message.parts[0].root.text == ''
         assert message.messageId == '12345678-1234-5678-1234-567812345678'
 
 
 class TestGetTextParts:
     def test_get_text_parts_single_text_part(self):
         # Setup
-        parts = [Part(root=TextPart(text="Hello world"))]
+        parts = [Part(root=TextPart(text='Hello world'))]
 
         # Exercise
         result = get_text_parts(parts)
 
         # Verify
-        assert result == ["Hello world"]
+        assert result == ['Hello world']
 
     def test_get_text_parts_multiple_text_parts(self):
         # Setup
         parts = [
-            Part(root=TextPart(text="First part")),
-            Part(root=TextPart(text="Second part")),
-            Part(root=TextPart(text="Third part")),
+            Part(root=TextPart(text='First part')),
+            Part(root=TextPart(text='Second part')),
+            Part(root=TextPart(text='Third part')),
         ]
 
         # Exercise
         result = get_text_parts(parts)
 
         # Verify
-        assert result == ["First part", "Second part", "Third part"]
+        assert result == ['First part', 'Second part', 'Third part']
 
     def test_get_text_parts_empty_list(self):
         # Setup
@@ -131,62 +149,62 @@ def test_get_message_text_single_part(self):
         # Setup
         message = Message(
             role=Role.agent,
-            parts=[Part(root=TextPart(text="Hello world"))],
-            messageId="test-message-id",
+            parts=[Part(root=TextPart(text='Hello world'))],
+            messageId='test-message-id',
         )
 
         # Exercise
         result = get_message_text(message)
 
         # Verify
-        assert result == "Hello world"
+        assert result == 'Hello world'
 
     def test_get_message_text_multiple_parts(self):
         # Setup
         message = Message(
             role=Role.agent,
             parts=[
-                Part(root=TextPart(text="First line")),
-                Part(root=TextPart(text="Second line")),
-                Part(root=TextPart(text="Third line")),
+                Part(root=TextPart(text='First line')),
+                Part(root=TextPart(text='Second line')),
+                Part(root=TextPart(text='Third line')),
             ],
-            messageId="test-message-id",
+            messageId='test-message-id',
         )
 
         # Exercise
         result = get_message_text(message)
 
         # Verify - default delimiter is newline
-        assert result == "First line\nSecond line\nThird line"
+        assert result == 'First line\nSecond line\nThird line'
 
     def test_get_message_text_custom_delimiter(self):
         # Setup
         message = Message(
             role=Role.agent,
             parts=[
-                Part(root=TextPart(text="First part")),
-                Part(root=TextPart(text="Second part")),
-                Part(root=TextPart(text="Third part")),
+                Part(root=TextPart(text='First part')),
+                Part(root=TextPart(text='Second part')),
+                Part(root=TextPart(text='Third part')),
             ],
-            messageId="test-message-id",
+            messageId='test-message-id',
         )
 
         # Exercise
-        result = get_message_text(message, delimiter=" | ")
+        result = get_message_text(message, delimiter=' | ')
 
         # Verify
-        assert result == "First part | Second part | Third part"
+        assert result == 'First part | Second part | Third part'
 
     def test_get_message_text_empty_parts(self):
         # Setup
         message = Message(
             role=Role.agent,
             parts=[],
-            messageId="test-message-id",
+            messageId='test-message-id',
         )
 
         # Exercise
         result = get_message_text(message)
 
         # Verify
-        assert result == ""
+        assert result == ''