diff options
Diffstat (limited to 'python/tests')
-rw-r--r-- | python/tests/import_crossref.py | 2 | ||||
-rw-r--r-- | python/tests/import_grobid_metadata.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/python/tests/import_crossref.py b/python/tests/import_crossref.py index a7826a73..193f78f6 100644 --- a/python/tests/import_crossref.py +++ b/python/tests/import_crossref.py @@ -78,7 +78,7 @@ def test_crossref_dict_parse(crossref_importer): assert r.contribs[0].raw_name == "Marcelo D. Radicioni" assert r.contribs[0].index == 0 assert r.contribs[1].raw_affiliation == "Some University" - assert r.contribs[1].extra['affiliations'] == ["Some Department"] + assert r.contribs[1].extra['more_affiliations'] == ["Some Department"] assert r.contribs[1].role == "author" assert r.contribs[3].role == "editor" assert r.contribs[3].index is None diff --git a/python/tests/import_grobid_metadata.py b/python/tests/import_grobid_metadata.py index feb604ce..4fed4aaa 100644 --- a/python/tests/import_grobid_metadata.py +++ b/python/tests/import_grobid_metadata.py @@ -26,7 +26,8 @@ def test_grobid_metadata_parse(grobid_metadata_importer): print(re.contribs) assert re.contribs[0].raw_name == "Wahyu Ary" assert re.publisher == None - assert re.extra.get('container_name') == None + if re.extra: + assert re.extra.get('container_name') == None assert len(re.refs) == 27 def test_file_metadata_parse(grobid_metadata_importer): |