refactor: update formatting in test_assistant_messages_in_history

mjunaidca · mjunaidca · commit c09a2258af56 · 2025-03-13T00:20:59.000+05:00
- Enhanced the readability of the test case by reformatting the expected output and input message structures.
- This change maintains the same functionality while making the test code cleaner and easier to understand.
diff --git a/tests/test_openai_chatcompletions_converter.py b/tests/test_openai_chatcompletions_converter.py
@@ -400,24 +400,27 @@ def test_assistant_messages_in_history():
     Test that assistant messages are added to the history.
     """
     messages = _Converter.items_to_messages(
-            [
-                {
-                    "role": "user",
-                    "content": "Hello",
-                },
-                {
-                    "role": "assistant",
-                    "content": "Hello?",
-                },
-                {
-                    "role": "user",
-                    "content": "What was my Name?",
-                },
-            ]
-        )
-    
-    # OUTPUT is [{'role': 'user', 'content': 'Hello'}, {'role': 'assistant', 'content': 'Hello?'}, {'role': 'user', 'content': 'What was my Name?'}]
-    assert messages == [{'role': 'user', 'content': 'Hello'}, {'role': 'assistant', 'content': 'Hello?'}, {'role': 'user', 'content': 'What was my Name?'}]
+        [
+            {
+                "role": "user",
+                "content": "Hello",
+            },
+            {
+                "role": "assistant",
+                "content": "Hello?",
+            },
+            {
+                "role": "user",
+                "content": "What was my Name?",
+            },
+        ]
+    )
+
+    assert messages == [
+        {"role": "user", "content": "Hello"},
+        {"role": "assistant", "content": "Hello?"},
+        {"role": "user", "content": "What was my Name?"},
+    ]
     assert len(messages) == 3
     assert messages[0]["role"] == "user"
     assert messages[0]["content"] == "Hello"
