added tests for search_arxiv and count_articles

trigaten · Oct 26, 2023 · 9e0c2c8 · 9e0c2c8
1 parent a5029af
commit 9e0c2c8
Showing 1 changed file with 8 additions and 3 deletions.
diff --git 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
+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