Add tests for fetch_semanticscholar

Author: stijnh

tests/test_sources_semanticscholar.py

@@ -1,6 +1,20 @@
-from litstudy.sources.semanticscholar import search_semanticscholar
+from litstudy.sources.semanticscholar import search_semanticscholar, fetch_semanticscholar
 
 
 def test_load_s2_file():
     search_semanticscholar('exascale', limit=10)
     # assert 1==2
+
+def test_fetch_semanticscholar():
+    doc = fetch_semanticscholar("arXiv:1705.10311")
+    assert doc.title == "Optimal Multi-Object Segmentation with Novel Gradient Vector Flow Based Shape Priors"
+
+    doc = fetch_semanticscholar("MAG:112218234")
+    assert doc.title == "Schistosomus Reflexus Syndrome in Olive Ridley Sea Turtles ( Lepidochelys olivacea)"
+
+    doc = fetch_semanticscholar("ACL:W12-3903")
+    assert doc.title == "The Study of Effect of Length in Morphological Segmentation of Agglutinative Languages"
+
+    doc = fetch_semanticscholar("PMID:19872477")
+    assert doc.title == "THE COMBINATION OF GELATIN WITH HYDROCHLORIC ACID : II. NEW DETERMINATIONS OF THE ISOELECTRIC POINT AND COMBINING CAPACITY OF A PURIFIED GELATIN"
+