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

hardcoded test

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

8
test/unit_tests/test_predictor.py
View File

@ -3,9 +3,11 @@ def test_predict_pdf_works(predictor, test_pdf):
predictions, metadata = predictor.predict_pdf(test_pdf)
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}
probabilities = predictions["probabilities"]
assert probabilities["formula"] == 1.0
assert probabilities['other'] == 1.576210689757148e-17
assert probabilities['signature'] == 5.364939675629517e-24
assert probabilities['logo'] == 8.815339177836606e-25
metadata = list(metadata)
metadata = dict(**metadata[0])
metadata.pop("document_filename") # temp filename cannot be tested