test_arxiv.py•597 B
# tests/test_arxiv.py
import unittest
from paper_search_mcp.academic_platforms.arxiv import ArxivSearcher
class TestArxivSearcher(unittest.TestCase):
def test_search(self):
searcher = ArxivSearcher()
papers = searcher.search("machine learning", max_results=10)
print(f"Found {len(papers)} papers for query 'machine learning':")
for i, paper in enumerate(papers, 1):
print(f"{i}. {paper.title} (ID: {paper.paper_id})")
self.assertEqual(len(papers), 10)
self.assertTrue(papers[0].title)
if __name__ == '__main__':
unittest.main()