From c6440c891154c1b64aa31eb58c257b7fc0342a9b Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Wed, 28 Aug 2024 10:58:19 +0200 Subject: [PATCH] added test to verify the cells in the pages (2) Signed-off-by: Peter Staar --- tests/test_toplevel_functions.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_toplevel_functions.py b/tests/test_toplevel_functions.py index 33552f65..9b98f54f 100644 --- a/tests/test_toplevel_functions.py +++ b/tests/test_toplevel_functions.py @@ -56,6 +56,13 @@ def verify_cells(doc_pred_json, doc_true_json): for pid, page_true_item in enumerate(doc_true_json["pages"]): + num_true_cells = len(page_true_item["cells"]) + num_pred_cells = len(doc_pred_json["pages"][pid]["cells"]) + + assert ( + num_true_cells == num_pred_cells + ), f"num_true_cells!=num_pred_cells {num_true_cells}!={num_pred_cells}" + for cid, cell_true_item in enumerate(page_true_item["cells"]): cell_pred_item = doc_pred_json["pages"][pid]["cells"][cid]