From f853d0afa1c3c181ad45b95ff7c07d48b9de8237 Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Wed, 28 Aug 2024 09:10:41 +0200 Subject: [PATCH] reformat code Signed-off-by: Peter Staar --- tests/test_toplevel_functions.py | 32 +++++++++++++++++++++----------- 1 file changed, 21 insertions(+), 11 deletions(-) diff --git a/tests/test_toplevel_functions.py b/tests/test_toplevel_functions.py index 8254914a..6e0b1ac1 100644 --- a/tests/test_toplevel_functions.py +++ b/tests/test_toplevel_functions.py @@ -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