diff options
Diffstat (limited to 'python/tests/test_grobid.py')
| -rw-r--r-- | python/tests/test_grobid.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/python/tests/test_grobid.py b/python/tests/test_grobid.py index 8c5e080..24ea40e 100644 --- a/python/tests/test_grobid.py +++ b/python/tests/test_grobid.py @@ -64,8 +64,12 @@ def test_grobid_worker_cdx(grobid_client, wayback_client): body=REAL_TEI_XML, content_type='text/xml') with open('tests/files/example.cdx', 'r') as cdx_file: - pusher = CdxLinePusher(worker, cdx_file, - filter_http_statuses=[200], filter_mimetypes=['application/pdf']) + pusher = CdxLinePusher( + worker, + cdx_file, + filter_http_statuses=[200, 226], + filter_mimetypes=['application/pdf'], + ) pusher_counts = pusher.run() assert pusher_counts['total'] assert pusher_counts['pushed'] == 7 |