mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

reformat code

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

32
tests/test_toplevel_functions.py
View File

@ -21,28 +21,32 @@ def get_pdf_paths():
def verify_cells(doc_pred_json, doc_true_json):
assert len(doc_pred_json["input"]["pages"])==len(doc_true_json["input"]["pages"]), \
"pred- and true-doc do not have the same number of pages"
assert len(doc_pred_json["input"]["pages"]) == len(
doc_true_json["input"]["pages"]
), "pred- and true-doc do not have the same number of pages"
for pid, page_true_item in enumerate(doc_true_json["input"]["pages"]):
for cid, cell_true_item in enumerate(page_true_json["cells"]):
cell_pred_item = doc_pred_json["input"]["pages"][pid]["cells"][cid]
true_text = cell_true_item["text"]
pred_text = cell_pred_item["text"]
assert true_text==pred_text, f"{true_text}!={pred_text}"
assert true_text == pred_text, f"{true_text}!={pred_text}"
for _ in ["t", "b", "l", "r"]:
true_val = np.round(cell_true_item["bbox"][_])
pred_val = np.round(cell_pred_item["bbox"][_])
assert pred_val==true_val, f"bbox for {_} is not the same: {true_val} != {pred_val}"
assert (
pred_val == true_val
), f"bbox for {_} is not the same: {true_val} != {pred_val}"
return True
def verify_maintext(doc_pred_json, doc_true_json):
for l, true_item in enumerate(doc_true_json["output"]["main_text"]):
@ -53,6 +57,7 @@ def verify_maintext(doc_pred_json, doc_true_json):
assert "text" in pred_item, f"`text` is in {pred_item}"
assert true_item["text"] == pred_item["text"]
def verify_tables(doc_pred_json, doc_true_json):
for l, true_item in enumerate(doc_true_json["output"]["tables"]):
@ -82,7 +87,7 @@ def verify_tables(doc_pred_json, doc_true_json):
return True
def verify_json(doc_pred_json, doc_true_json):
if doc_pred_json.keys() != doc_true_json.keys():
@ -91,12 +96,17 @@ def verify_json(doc_pred_json, doc_true_json):
if doc_pred_json["output"].keys() != doc_true_json["output"].keys():
return False
assert verify_maintext(doc_pred_json, doc_true_json), "verify_maintext(doc_pred_json, doc_true_json)"
assert verify_maintext(
doc_pred_json, doc_true_json
), "verify_maintext(doc_pred_json, doc_true_json)"
assert verify_tables(doc_pred_json, doc_true_json), "verify_tables(doc_pred_json, doc_true_json)"
assert verify_tables(
doc_pred_json, doc_true_json
), "verify_tables(doc_pred_json, doc_true_json)"
return True
def verify_md(doc_pred_md, doc_true_md):
return doc_pred_md == doc_true_md