format update

tests/prompts/test_prompt_integration.py

@@ -3,39 +3,40 @@
 import pytest
 from arxiv_mcp_server.prompts.handlers import list_prompts, get_prompt
 
+
 @pytest.mark.asyncio
 async def test_server_list_prompts():
     """Test server list_prompts endpoint."""
     prompts = await list_prompts()
     assert len(prompts) == 1
-    
+
     # Check that all prompts have required fields
     for prompt in prompts:
         assert prompt.name
         assert prompt.description
         assert prompt.arguments is not None
 
+
 @pytest.mark.asyncio
 async def test_server_get_analysis_prompt():
     """Test server get_prompt endpoint with analysis prompt."""
-    result = await get_prompt(
+    result = await get_prompt("deep-paper-analysis", {"paper_id": "2401.00123"})
-        "deep-paper-analysis",
+
-        {"paper_id": "2401.00123"}
-    )
-    
     assert len(result.messages) == 1
     message = result.messages[0]
     assert message.role == "user"
     assert "2401.00123" in message.content.text
 
+
 @pytest.mark.asyncio
 async def test_server_get_prompt_invalid_name():
     """Test server get_prompt endpoint with invalid prompt name."""
     with pytest.raises(ValueError, match="Prompt not found"):
         await get_prompt("invalid-prompt", {})
 
+
 @pytest.mark.asyncio
 async def test_server_get_prompt_missing_args():
     """Test server get_prompt endpoint with missing required arguments."""
     with pytest.raises(ValueError, match="Missing required argument"):
-        await get_prompt("deep-paper-analysis", {})
+        await get_prompt("deep-paper-analysis", {})

tests/prompts/test_prompts.py

@@ -5,47 +5,49 @@ from typing import Dict
 from arxiv_mcp_server.prompts.handlers import list_prompts, get_prompt
 from mcp.types import GetPromptResult, PromptMessage, TextContent
 
+
 @pytest.mark.asyncio
 async def test_list_prompts():
     """Test listing available prompts."""
     prompts = await list_prompts()
     assert len(prompts) == 1
-    
+
     prompt_names = {p.name for p in prompts}
     expected_names = {"deep-paper-analysis"}
     assert prompt_names == expected_names
 
+
 @pytest.mark.asyncio
 async def test_get_paper_analysis_prompt():
     """Test getting paper analysis prompt."""
-    result = await get_prompt(
+    result = await get_prompt("deep-paper-analysis", {"paper_id": "2401.00123"})
-        "deep-paper-analysis",
+
-        {"paper_id": "2401.00123"}
-    )
-    
     assert isinstance(result, GetPromptResult)
     assert len(result.messages) == 1
     message = result.messages[0]
-    
+
     assert isinstance(message, PromptMessage)
     assert message.role == "user"
     assert isinstance(message.content, TextContent)
     assert "2401.00123" in message.content.text
 
+
 @pytest.mark.asyncio
 async def test_get_prompt_with_invalid_name():
     """Test getting prompt with invalid name."""
     with pytest.raises(ValueError, match="Prompt not found"):
         await get_prompt("invalid-prompt", {})
 
+
 @pytest.mark.asyncio
 async def test_get_prompt_with_no_arguments():
     """Test getting prompt with no arguments."""
     with pytest.raises(ValueError, match="No arguments provided"):
         await get_prompt("deep-paper-analysis", None)
 
+
 @pytest.mark.asyncio
 async def test_get_prompt_with_missing_required_argument():
     """Test getting prompt with missing required argument."""
     with pytest.raises(ValueError, match="Missing required argument"):
-        await get_prompt("deep-paper-analysis", {})
+        await get_prompt("deep-paper-analysis", {})