format update

This commit is contained in:
blazickjp
2025-04-22 04:08:35 -07:00
parent 26fa8f3b0b
commit 4510855fc5
2 changed files with 18 additions and 15 deletions

View File

@@ -3,6 +3,7 @@
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."""
@@ -15,25 +16,25 @@ async def test_server_list_prompts():
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(
"deep-paper-analysis",
{"paper_id": "2401.00123"}
)
result = await get_prompt("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."""

View File

@@ -5,6 +5,7 @@ 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."""
@@ -15,13 +16,11 @@ async def test_list_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(
"deep-paper-analysis",
{"paper_id": "2401.00123"}
)
result = await get_prompt("deep-paper-analysis", {"paper_id": "2401.00123"})
assert isinstance(result, GetPromptResult)
assert len(result.messages) == 1
@@ -32,18 +31,21 @@ async def test_get_paper_analysis_prompt():
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."""