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

test

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-03-21 11:01:21 +01:00
parent 3d9c4d8856
commit 3e9bfac5cf
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/unit_tests/test_predictor.py
View File

@ -1,11 +1,11 @@
def test_predict_pdf_works(predictor, test_pdf):
# FIXME strange outputs, with '\n's in the dict
predictions, metadata = predictor.predict_pdf(test_pdf)
predictions = [p for p in predictions]
assert predictions[0] == {'class': 'formula',
'probabilities': {'formula': 1.0, 'other': 1.576210689757148e-17,
'signature': 5.364939675629517e-24,
'logo': 8.815339177836606e-25}}
predictions = [p for p in predictions][0]
assert predictions["class"] == "formula"
assert dict(**predictions["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