diff options
Diffstat (limited to 'tests/test_parse.py')
| -rw-r--r-- | tests/test_parse.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/tests/test_parse.py b/tests/test_parse.py index 7749201..25ffa64 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -5,8 +5,7 @@ import xml.etree.ElementTree import pytest -from grobid_tei_xml import (GrobidCitation, GrobidDocument, parse_citations_xml, - parse_document_xml) +from grobid_tei_xml import GrobidBiblio, GrobidDocument, parse_citations_xml, parse_document_xml from grobid_tei_xml.types import * @@ -21,7 +20,7 @@ def test_small_xml() -> None: grobid_version='0.5.1-SNAPSHOT', grobid_timestamp='2018-04-02T00:31+0000', language_code='en', - header=GrobidHeader( + header=GrobidBiblio( title="Dummy Example File", authors=[ GrobidAuthor(full_name="Brewster Kahle", @@ -43,14 +42,13 @@ def test_small_xml() -> None: surname="Doe", ), ], - journal=GrobidJournal( - name="Dummy Example File. Journal of Fake News. pp. 1-2. ISSN 1234-5678", ), + journal="Dummy Example File. Journal of Fake News. pp. 1-2. ISSN 1234-5678", date="2000", ), abstract="Everything you ever wanted to know about nothing", body=expected_body, citations=[ - GrobidCitation( + GrobidBiblio( index=0, id="b0", authors=[ @@ -62,7 +60,7 @@ def test_small_xml() -> None: volume="20", pages="1-11", ), - GrobidCitation( + GrobidBiblio( index=1, id="b1", authors=[], |