mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

added test to verify the cells in the pages (2)

Browse Source 
Signed-off-by: Peter Staar <taa@zurich.ibm.com>
This commit is contained in:
Peter Staar 2024-08-28 10:58:19 +02:00
parent e6ed6f4793
commit c6440c8911
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_toplevel_functions.py
View File

@ -56,6 +56,13 @@ def verify_cells(doc_pred_json, doc_true_json):
for pid, page_true_item in enumerate(doc_true_json["pages"]): 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"]): for cid, cell_true_item in enumerate(page_true_item["cells"]):
cell_pred_item = doc_pred_json["pages"][pid]["cells"][cid] cell_pred_item = doc_pred_json["pages"][pid]["cells"][cid]