From eb6f211f02bc184e7f92d6b4d53c91da34ab9f2f Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Mon, 21 Mar 2022 11:07:32 +0100 Subject: [PATCH] hardcoded test --- test/unit_tests/test_predictor.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/test/unit_tests/test_predictor.py b/test/unit_tests/test_predictor.py index e74f9df..851f4fe 100644 --- a/test/unit_tests/test_predictor.py +++ b/test/unit_tests/test_predictor.py @@ -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