Fixes https://github.com/BeehiveInnovations/gemini-mcp-server/issues/6

tests/test_collaboration.py

@@ -235,9 +235,9 @@ class TestCollaborationWorkflow:
         )
 
         response = json.loads(result[0].text)
-        assert (
-            response["status"] == "requires_clarification"
-        ), "Should request clarification when asked about dependencies without package files"
+        assert response["status"] == "requires_clarification", (
+            "Should request clarification when asked about dependencies without package files"
+        )
 
         clarification = json.loads(response["content"])
         assert "package.json" in str(clarification["files_needed"]), "Should specifically request package.json"

tests/test_conversation_memory.py

@@ -300,7 +300,7 @@ class TestConversationFlow:
         # REQUEST 6: Try to exceed MAX_CONVERSATION_TURNS limit - should fail
         turns_at_limit = [
             ConversationTurn(
-                role="assistant" if i % 2 == 0 else "user", content=f"Turn {i+1}", timestamp="2023-01-01T00:00:30Z"
+                role="assistant" if i % 2 == 0 else "user", content=f"Turn {i + 1}", timestamp="2023-01-01T00:00:30Z"
             )
             for i in range(MAX_CONVERSATION_TURNS)
         ]
@@ -423,7 +423,9 @@ class TestConversationFlow:
             # Mock context with current turns
             turns = [
                 ConversationTurn(
-                    role="user" if i % 2 == 0 else "assistant", content=f"Turn {i+1}", timestamp="2023-01-01T00:00:00Z"
+                    role="user" if i % 2 == 0 else "assistant",
+                    content=f"Turn {i + 1}",
+                    timestamp="2023-01-01T00:00:00Z",
                 )
                 for i in range(turn_num)
             ]
@@ -445,7 +447,7 @@ class TestConversationFlow:
         # Now we should be at the limit - create final context
         final_turns = [
             ConversationTurn(
-                role="user" if i % 2 == 0 else "assistant", content=f"Turn {i+1}", timestamp="2023-01-01T00:00:00Z"
+                role="user" if i % 2 == 0 else "assistant", content=f"Turn {i + 1}", timestamp="2023-01-01T00:00:00Z"
             )
             for i in range(MAX_CONVERSATION_TURNS)
         ]

tests/test_thinking_modes.py

@@ -32,9 +32,9 @@ class TestThinkingModes:
         ]
 
         for tool, expected_default in tools:
-            assert (
-                tool.get_default_thinking_mode() == expected_default
-            ), f"{tool.__class__.__name__} should default to {expected_default}"
+            assert tool.get_default_thinking_mode() == expected_default, (
+                f"{tool.__class__.__name__} should default to {expected_default}"
+            )
 
     @pytest.mark.asyncio
     @patch("tools.base.BaseTool.create_model")