From 8a34f3747a6881eab66775388a9b6d7878c00a6c Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Wed, 30 Jun 2021 18:39:48 -0700 Subject: refs: include (source) release_stage in output --- tests/test_refs_transform.py | 27 ++++++++++++++++++--------- 1 file changed, 18 insertions(+), 9 deletions(-) (limited to 'tests') diff --git a/tests/test_refs_transform.py b/tests/test_refs_transform.py index c26ee1e..3fa490b 100644 --- a/tests/test_refs_transform.py +++ b/tests/test_refs_transform.py @@ -13,21 +13,30 @@ def test_transform_refs_grobid() -> None: ident="releasedummy22222222222222", work_id="workdummy22222222222222222", release_year=1234, + release_stage="accepted", ext_ids={}, ) tei_dict = teixml2json(blob, True) refs = refs_from_grobid(dummy_release, tei_dict) - ref = refs[12].biblio - assert ref.contrib_raw_names is not None - assert ref.contrib_raw_names[0] == "K Tasa" - assert ref.container_name == "Quality Management in Health Care" - assert ref.title == "Using patient feedback for quality improvement" - assert ref.year == 1996 - assert ref.pages == "206-225" - assert ref.volume == "8" + ref = refs[12] + assert ref.release_ident == "releasedummy22222222222222" + assert ref.work_ident == "workdummy22222222222222222" + assert ref.release_stage == "accepted" + assert ref.release_year == 1234 + assert ref.ref_source == "grobid" + assert ref.key == "b12" + assert ref.index == 12 + assert ref.locator == None + assert ref.biblio.contrib_raw_names is not None + assert ref.biblio.contrib_raw_names[0] == "K Tasa" + assert ref.biblio.container_name == "Quality Management in Health Care" + assert ref.biblio.title == "Using patient feedback for quality improvement" + assert ref.biblio.year == 1996 + assert ref.biblio.pages == "206-225" + assert ref.biblio.volume == "8" assert ( - ref.unstructured + ref.biblio.unstructured == "Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19." ) -- cgit v1.2.3