Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

fix test

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-03-21 10:21:38 +01:00
parent 6faad5ad5b
commit 58ca784d6c
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
test/unit_tests/test_predictor.py
View File

@ -1,11 +1,16 @@
def test_predict_pdf_works(predictor, test_pdf):
# FIXME strange outputs, with '\n's in the dict
predictions, metadata = predictor.predict_pdf(test_pdf)
assert list(predictions) == [{'class': 'formula',
'probabilities': {'formula': 1.0, 'other': 1.576210689757148e-17,
'signature': 5.364939675629517e-24,
'logo': 8.815339177836606e-25}}]
# FIXME strange output for metadata, with '\n's in the dict
predictions = list(predictions)
predictions = dict(**predictions[0])
assert predictions == {'class': 'formula',
'probabilities': {'formula': 1.0, 'other': 1.576210689757148e-17,
'signature': 5.364939675629517e-24,
'logo': 8.815339177836606e-25}}
metadata = list(metadata)
metadata = dict(**metadata[0])
metadata.pop("document_filename") # temp filename cannot be tested
assert metadata == {'px_width': 389.0, 'px_height': 389.0, 'width': 194.49999000000003, 'height': 194.49998999999997, 'x1': 320.861, 'x2': 515.36099, 'y1': 347.699, 'y2': 542.19899, 'page_width': 595.2800000000001, 'page_height': 841.89, 'page_rotation': 0, 'page_idx': 1, 'n_pages': 3}
assert metadata == {'px_width': 389.0, 'px_height': 389.0, 'width': 194.49999000000003,
'height': 194.49998999999997, 'x1': 320.861, 'x2': 515.36099, 'y1': 347.699, 'y2': 542.19899,
'page_width': 595.2800000000001, 'page_height': 841.89, 'page_rotation': 0, 'page_idx': 1,
'n_pages': 3}