diff options
author | Bryan Newbold <bnewbold@archive.org> | 2021-10-21 14:00:10 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@archive.org> | 2021-10-21 14:00:10 -0700 |
commit | 45870aa17b5dec6d63df8b7a2eb7839feac9afec (patch) | |
tree | b7e6918da2420cbbe3e7fcbe5d14ec3d2de3cc53 /tests | |
parent | 2f3fd33c33f12b4426072a0279de85ff797611fe (diff) | |
download | grobid_tei_xml-45870aa17b5dec6d63df8b7a2eb7839feac9afec.tar.gz grobid_tei_xml-45870aa17b5dec6d63df8b7a2eb7839feac9afec.zip |
rip out API interaction code
Diffstat (limited to 'tests')
-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) |