fix: unit-tests (#3)

Signed-off-by: Peter Staar <taa@zurich.ibm.com>
DS4SD · Aug 6, 2024 · fa7bef7 · fa7bef7
1 parent ccb7675
commit fa7bef7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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"]):