diff options
author | Bryan Newbold <bnewbold@archive.org> | 2021-10-27 15:33:29 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@archive.org> | 2021-10-27 18:25:58 -0700 |
commit | 33211915773a0c77d064c55c1b02ceed6f455feb (patch) | |
tree | 1828505db917686e7223d41e97c6446223f2da32 /tests/test_refs_transform.py | |
parent | 6c103e4dc48e7e0c0f6cdedc18b0afe33babf1ac (diff) | |
download | fatcat-scholar-33211915773a0c77d064c55c1b02ceed6f455feb.tar.gz fatcat-scholar-33211915773a0c77d064c55c1b02ceed6f455feb.zip |
replace grobid2json with grobid_tei_xml
This first iteration uses the .to_legacy_dict() helpers for backwards
compatibility
Diffstat (limited to 'tests/test_refs_transform.py')
-rw-r--r-- | tests/test_refs_transform.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_refs_transform.py b/tests/test_refs_transform.py index 5f15adf..2fc210f 100644 --- a/tests/test_refs_transform.py +++ b/tests/test_refs_transform.py @@ -1,8 +1,8 @@ import json from fatcat_openapi_client import ReleaseEntity +from grobid_tei_xml import parse_document_xml -from fatcat_scholar.grobid2json import teixml2json from fatcat_scholar.transform import refs_from_crossref, refs_from_grobid @@ -19,7 +19,8 @@ def test_transform_refs_grobid() -> None: ext_ids={}, ) - tei_dict = teixml2json(blob, True) + tei_doc = parse_document_xml(blob) + tei_dict = tei_doc.to_legacy_dict() refs = refs_from_grobid(dummy_release, tei_dict) ref = refs[12] |