From 544aeae538c9693c48fbf5dd7bbc0ae51845ba76 Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Tue, 6 Aug 2024 13:17:28 +0200 Subject: [PATCH] fixed unit-tests Signed-off-by: Peter Staar --- tests/test_parse.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_parse.py b/tests/test_parse.py index f0904077..846f0f6c 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -28,7 +28,10 @@ def test_reference_documents(): with open(pdf_doc+".json", "r") as fr: true_doc = json.load(fr) - assert pred_doc["pages"]==true_doc["pages"], "pred_doc[\"pages\"]!=true_doc[\"pages\"]" + num_true_pages=len(true_doc["pages"]) + num_pred_pages=len(pred_doc["pages"]) + + assert num_true_pages==num_pred_pages, f"len(pred_doc[\"pages\"])!=len(true_doc[\"pages\"]) => {num_true_pages}!={num_pred_pages}" for pred_page,true_page in zip(pred_doc["pages"], true_doc["pages"]):