From a0e275a4bad46ef41585f0207d6dfa1e3c38bc35 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Wed, 27 Oct 2021 19:10:35 -0700 Subject: remove grobid2json helper file, replace with grobid_tei_xml --- python/tests/test_grobid2json.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'python/tests') diff --git a/python/tests/test_grobid2json.py b/python/tests/test_grobid2json.py index 98888e8..b00a88d 100644 --- a/python/tests/test_grobid2json.py +++ b/python/tests/test_grobid2json.py @@ -2,23 +2,27 @@ import json import xml import pytest - -from grobid2json import * +from grobid_tei_xml import parse_document_xml def test_small_xml(): + """ + This used to be a test of grobid2json; now it is a compatability test for + the to_legacy_dict() feature of grobid_tei_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()) - assert teixml2json(tei_xml) == json_form + tei_doc = parse_document_xml(tei_xml) + assert tei_doc.to_legacy_dict() == json_form def test_invalid_xml(): with pytest.raises(xml.etree.ElementTree.ParseError): - teixml2json("this is not XML") + parse_document_xml("this is not XML") with pytest.raises(ValueError): - teixml2json("") + parse_document_xml("") -- cgit v1.2.3