diff options
Diffstat (limited to 'tests/test_grobid2json.py')
| -rw-r--r-- | tests/test_grobid2json.py | 35 |
1 files changed, 14 insertions, 21 deletions
diff --git a/tests/test_grobid2json.py b/tests/test_grobid2json.py index a1c975e..47ab293 100644 --- a/tests/test_grobid2json.py +++ b/tests/test_grobid2json.py @@ -1,12 +1,12 @@ -import xml import json +import xml + import pytest -from grobid_tei_xml import teixml2json, parse_document_xml, GrobidDocument, GrobidCitation -from grobid_tei_xml.grobid2json import transform_grobid_ref_xml +from grobid_tei_xml.grobid2json import teixml2json, transform_grobid_ref_xml -def test_small_xml(): +def test_small_xml() -> None: with open('tests/files/small.xml', 'r') as f: tei_xml = f.read() @@ -16,7 +16,7 @@ def test_small_xml(): assert teixml2json(tei_xml) == json_form -def test_invalid_xml(): +def test_invalid_xml() -> None: with pytest.raises(xml.etree.ElementTree.ParseError): teixml2json("this is not XML") @@ -31,29 +31,21 @@ def test_grobid_teixml2json() -> None: obj = teixml2json(blob, True) - assert ( - obj["title"] == - "Changes of patients' satisfaction with the health care services in Lithuanian Health Promoting Hospitals network" - ) + assert obj[ + "title"] == """Changes of patients' satisfaction with the health care services in Lithuanian Health Promoting Hospitals network""" ref = [c for c in obj["citations"] if c["id"] == "b12"][0] - assert ref["authors"][0] == { - "given_name": "K", - "name": "K Tasa", - "surname": "Tasa" - } + assert ref["authors"][0] == {"given_name": "K", "name": "K Tasa", "surname": "Tasa"} assert ref["journal"] == "Quality Management in Health Care" assert ref["title"] == "Using patient feedback for quality improvement" assert ref["date"] == "1996" assert ref["pages"] == "206-225" assert ref["volume"] == "8" - assert ( - ref["unstructured"] == - "Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19." - ) + assert ref["unstructured"] == \ + """Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19.""" -def test_transform_grobid_ref_xml(): +def test_transform_grobid_ref_xml() -> None: citation_xml = """ <biblStruct > <analytic> @@ -104,8 +96,9 @@ def test_transform_grobid_ref_xml(): </biblStruct>""" d = transform_grobid_ref_xml(citation_xml) - assert d[ - 'title'] == "Mesh migration following abdominal hernia repair: a comprehensive review" + assert d + assert d['title'] == \ + "Mesh migration following abdominal hernia repair: a comprehensive review" assert d['authors'][2]['given_name'] == "L" assert d['authors'][2]['surname'] == "Taveras" assert d['authors'][2]['name'] == "L R Taveras" |