diff options
author | Bryan Newbold <bnewbold@archive.org> | 2021-10-21 14:05:26 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@archive.org> | 2021-10-21 14:05:26 -0700 |
commit | d58721ddb034363db10eb013986f560a87df7a19 (patch) | |
tree | 2f25cf50a638aa6db08d9470ecc8243db869b00c /tests/test_grobid2json.py | |
parent | c1059b97772e91468edbcacab6bd830ca1ffaa81 (diff) | |
download | grobid_tei_xml-d58721ddb034363db10eb013986f560a87df7a19.tar.gz grobid_tei_xml-d58721ddb034363db10eb013986f560a87df7a19.zip |
make fmt
Diffstat (limited to 'tests/test_grobid2json.py')
-rw-r--r-- | tests/test_grobid2json.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/test_grobid2json.py b/tests/test_grobid2json.py index 7ee423c..6e3dac2 100644 --- a/tests/test_grobid2json.py +++ b/tests/test_grobid2json.py @@ -1,4 +1,3 @@ - import xml import json import pytest @@ -6,14 +5,15 @@ from grobid_tei_xml.grobid2json import teixml2json def test_small_xml(): - + with open('tests/files/small.xml', 'r') as f: tei_xml = f.read() with open('tests/files/small.json', 'r') as f: - json_form = json.loads(f.read()) + json_form = json.loads(f.read()) assert teixml2json(tei_xml) == json_form + def test_invalid_xml(): with pytest.raises(xml.etree.ElementTree.ParseError): @@ -30,18 +30,22 @@ 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" + 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." + ref["unstructured"] == + "Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19." ) |