Paper Search MCP

import unittest import os import requests from paper_search_mcp.academic_platforms.google_scholar import GoogleScholarSearcher def check_scholar_accessible(): """检查 Google Scholar 是否可访问""" try: response = requests.get("https://scholar.google.com", timeout=5) return response.status_code == 200 except: return False class TestGoogleScholarSearcher(unittest.TestCase): @classmethod def setUpClass(cls): cls.scholar_accessible = check_scholar_accessible() if not cls.scholar_accessible: print("\nWarning: Google Scholar is not accessible, some tests will be skipped") def setUp(self): self.searcher = GoogleScholarSearcher() def test_search(self): if not self.scholar_accessible: self.skipTest("Google Scholar is not accessible") papers = self.searcher.search("machine learning", max_results=5) print(f"\nFound {len(papers)} papers for query 'machine learning':") for i, paper in enumerate(papers, 1): print(f"\n{i}. {paper.title}") print(f" Authors: {', '.join(paper.authors)}") print(f" Citations: {paper.citations}") self.assertTrue(len(papers) > 0) self.assertTrue(papers[0].title) def test_download_pdf_not_supported(self): with self.assertRaises(NotImplementedError): self.searcher.download_pdf("some_id", "./downloads") def test_read_paper_not_supported(self): message = self.searcher.read_paper("some_id") self.assertIn("Google Scholar doesn't support direct paper reading", message) if __name__ == '__main__': unittest.main()