from paperscraper_mcpserver.server import search_pubmed_papers, search_scholar_papers

def main():
    print("Testing PubMed tool...")
    pubmed_result = search_pubmed_papers("COVID-19", 2)
    print(pubmed_result)

    print("\nTesting Scholar tool...")
    scholar_result = search_scholar_papers("Attention is all you need")
    print(scholar_result)

if __name__ == "__main__":
    main()
