from grobid_tei_xml import parse_document_xml def test_small_xml_csl() -> None: with open('tests/files/small.xml', 'r') as f: tei_xml = f.read() d = parse_document_xml(tei_xml) assert d.to_csl_dict() == { "type": "article-journal", "title": "Dummy Example File", "author": [ { "given": "Brewster", "family": "Kahle" }, { "given": "J", "family": "Doe", }, ], "book-title": "Dummy Example File. Journal of Fake News. pp. 1-2. ISSN 1234-5678", "issued": [[2000]], } assert d.citations assert d.citations[0].to_csl_dict() == { "type": "article-journal", "title": "Everything is Wonderful", "author": [ { "given": "A", "family": "Seaperson" }, ], "container-title": "Letters in the Alphabet", "issued": [[2001]], "volume": 20, "page": "1-11", "page-first": "1", }