diff options
Diffstat (limited to 'tests/test_grobid_unstructured.py')
-rw-r--r-- | tests/test_grobid_unstructured.py | 63 |
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) |