aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_grobid_unstructured.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_grobid_unstructured.py')
-rw-r--r--tests/test_grobid_unstructured.py63
1 files changed, 1 insertions, 62 deletions
diff --git a/tests/test_grobid_unstructured.py b/tests/test_grobid_unstructured.py
index dd69936..b8d79ca 100644
--- a/tests/test_grobid_unstructured.py
+++ b/tests/test_grobid_unstructured.py
@@ -1,36 +1,6 @@
import pytest
-from fuzzycat.grobid_unstructured import grobid_api_process_citation, grobid_parse_unstructured, grobid_ref_to_release, transform_grobid_ref_xml
-
-
-def test_grobid_ref_to_release():
-
- d = {
- 'title':
- "some title",
- 'doi':
- '10.1234/5678',
- 'journal':
- 'some journal',
- 'authors': [
- {
- 'name': 'ahab sailor',
- 'given_name': 'ahab',
- 'surname': 'sailor'
- },
- {
- 'name': 'mary jane',
- 'given_name': 'mary',
- 'surname': 'jane'
- },
- ],
- }
- r = grobid_ref_to_release(d)
- assert r.title == d['title']
- assert r.ext_ids.doi == d['doi']
- assert r.extra['container_name'] == d['journal']
- assert r.contribs[0].surname == d['authors'][0]['surname']
- assert r.contribs[1].raw_name == d['authors'][1]['name']
+from grobid_tei_xml.grobid_unstructured import transform_grobid_ref_xml
def test_transform_grobid_ref_xml():
@@ -97,34 +67,3 @@ def test_transform_grobid_ref_xml():
assert d['journal'] == "Hernia"
-def test_grobid_parse_unstructured():
- """
- NOTE: this test makes live network requests to GROBID
- """
-
- r = grobid_parse_unstructured("blah")
- assert r is None
-
- r = grobid_parse_unstructured(
- """Cunningham HB, Weis JJ, Taveras LR, Huerta S. Mesh migration following abdominal hernia repair: a comprehensive review. Hernia. 2019 Apr;23(2):235-243. doi: 10.1007/s10029-019-01898-9. Epub 2019 Jan 30. PMID: 30701369."""
- )
- assert r.title == "Mesh migration following abdominal hernia repair: a comprehensive review"
- assert r.contribs[0].surname == "Cunningham"
- assert r.contribs[1].surname == "Weis"
- assert r.contribs[2].surname == "Taveras"
- assert r.contribs[3].surname == "Huerta"
- assert r.extra['container_name'] == "Hernia"
- assert r.release_year == 2019
- assert r.volume == "23"
- assert r.issue == "2"
- assert r.pages == "235-243"
- assert r.ext_ids.doi == "10.1007/s10029-019-01898-9"
- assert r.ext_ids.pmid == "30701369"
-
-
-def test_grobid_parse_unstructured_timeout():
- """
- NOTE: this test makes live network requests to GROBID
- """
- with pytest.raises(TimeoutError):
- grobid_parse_unstructured("blah", timeout=0.000001)