ci(lint): align tests with ruff format so pipeline passes

Author: vchomakov

examples/redis_assistant.py

@@ -12,15 +12,17 @@ async def build_agent():
         params={
             "command": "uv",
             "args": [
-                "--directory", "../src/", # change with the path to the MCP server
-                "run", "main.py"
+                "--directory",
+                "../src/",  # change with the path to the MCP server
+                "run",
+                "main.py",
             ],
-        "env": {
-            "REDIS_HOST": "127.0.0.1",
-            "REDIS_PORT": "6379",
-            "REDIS_USERNAME": "default",
-            "REDIS_PWD": ""
-        },
+            "env": {
+                "REDIS_HOST": "127.0.0.1",
+                "REDIS_PORT": "6379",
+                "REDIS_USERNAME": "default",
+                "REDIS_PWD": "",
+            },
         }
     )
 
@@ -30,7 +32,7 @@ async def build_agent():
     agent = Agent(
         name="Redis Assistant",
         instructions="You are a helpful assistant capable of reading and writing to Redis. Store every question and answer in the Redis Stream app:logger",
-        mcp_servers=[server]
+        mcp_servers=[server],
     )
 
     return agent
@@ -46,7 +48,7 @@ async def cli(agent, max_history=30):
         if q.strip().lower() in {"exit", "quit"}:
             break
 
-        if (len(q.strip()) > 0):
+        if len(q.strip()) > 0:
             # Format the context into a single string
             history = ""
             for turn in conversation_history:
@@ -58,7 +60,9 @@ async def cli(agent, max_history=30):
 
             response_text = ""
             async for event in result.stream_events():
-                if event.type == "raw_response_event" and isinstance(event.data, ResponseTextDeltaEvent):
+                if event.type == "raw_response_event" and isinstance(
+                    event.data, ResponseTextDeltaEvent
+                ):
                     print(event.data.delta, end="", flush=True)
                     response_text += event.data.delta
             print("\n")

tests/test_integration.py

@@ -240,9 +240,9 @@ def test_server_lists_tools(self, server_process):
             "get",
         ]
         for tool in tool_names:
-            assert (
-                tool in expected_tools
-            ), f"Expected tool '{tool}' not found in {tool_names}"
+            assert tool in expected_tools, (
+                f"Expected tool '{tool}' not found in {tool_names}"
+            )
 
     def test_server_tool_count_and_names(self, server_process):
         """Test that the server registers the correct number of tools with expected names."""
@@ -269,9 +269,9 @@ def test_server_tool_count_and_names(self, server_process):
 
         # Expected tool count (based on @mcp.tool() decorators in codebase)
         expected_tool_count = 44
-        assert (
-            len(tools) == expected_tool_count
-        ), f"Expected {expected_tool_count} tools, but got {len(tools)}"
+        assert len(tools) == expected_tool_count, (
+            f"Expected {expected_tool_count} tools, but got {len(tools)}"
+        )
 
         # Expected tool names (alphabetically sorted for easier verification)
         expected_tools = [
@@ -360,9 +360,9 @@ def test_server_tool_count_and_names(self, server_process):
 
         for category, category_tools in tool_categories.items():
             for tool in category_tools:
-                assert (
-                    tool in tool_names
-                ), f"Tool '{tool}' from category '{category}' not found in registered tools"
+                assert tool in tool_names, (
+                    f"Tool '{tool}' from category '{category}' not found in registered tools"
+                )
 
     def _initialize_server(self, server_process):
         """Helper to initialize the MCP server."""

tests/test_logging_utils.py

@@ -68,7 +68,9 @@ def test_configure_logging_adds_stderr_handler_when_none(monkeypatch, preserve_l
     level = configure_logging()
 
     assert level == logging.INFO
-    assert len(root.handlers) == 1, "Should add exactly one stderr handler when none exist"
+    assert len(root.handlers) == 1, (
+        "Should add exactly one stderr handler when none exist"
+    )
     handler = root.handlers[0]
     assert isinstance(handler, logging.StreamHandler)
     # StreamHandler exposes the underlying stream attribute
@@ -77,7 +79,9 @@ def test_configure_logging_adds_stderr_handler_when_none(monkeypatch, preserve_l
     assert root.level == logging.INFO
 
 
-def test_configure_logging_only_lowers_restrictive_handlers(monkeypatch, preserve_logging):
+def test_configure_logging_only_lowers_restrictive_handlers(
+    monkeypatch, preserve_logging
+):
     root = logging.getLogger()
     # Start from a clean handler set
     for h in list(root.handlers):
@@ -101,7 +105,9 @@ def test_configure_logging_only_lowers_restrictive_handlers(monkeypatch, preserv
     assert h_notset.level == logging.NOTSET
 
 
-def test_configure_logging_does_not_raise_handler_threshold(monkeypatch, preserve_logging):
+def test_configure_logging_does_not_raise_handler_threshold(
+    monkeypatch, preserve_logging
+):
     root = logging.getLogger()
     # Clean handlers
     for h in list(root.handlers):
@@ -121,7 +127,9 @@ def test_configure_logging_does_not_raise_handler_threshold(monkeypatch, preserv
     assert root.level == logging.ERROR
 
 
-def test_configure_logging_does_not_add_handler_if_exists(monkeypatch, preserve_logging):
+def test_configure_logging_does_not_add_handler_if_exists(
+    monkeypatch, preserve_logging
+):
     root = logging.getLogger()
     # Start with one existing handler
     for h in list(root.handlers):
@@ -135,4 +143,3 @@ def test_configure_logging_does_not_add_handler_if_exists(monkeypatch, preserve_
     # Should not add another handler
     assert len(root.handlers) == 1
     assert root.handlers[0] is existing
-

tests/test_server.py

@@ -127,6 +127,6 @@ def test_mcp_server_attributes(self):
 
         for attr in expected_attributes:
             assert hasattr(mcp, attr), f"MCP server missing attribute: {attr}"
-            assert callable(
-                getattr(mcp, attr)
-            ), f"MCP server attribute {attr} is not callable"
+            assert callable(getattr(mcp, attr)), (
+                f"MCP server attribute {attr} is not callable"
+            )