diff --git a/tests/test_utils.py b/tests/test_utils.py index 88ff72c..5195fd3 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -1,6 +1,11 @@ # test_utils.py import pytest -from prompt_systematic_review.utils import * +from prompt_systematic_review.utils import search_arxiv, count_articles -def test_assert(): - assert True \ No newline at end of file +def test_search_arxiv(): + data = search_arxiv("covid",max_results=10) + assert len(data) > 0 + +def test_count_articles(): + data = search_arxiv("covid",max_results=10) + assert count_articles(data) == 10 \ No newline at end of file