From 94548c409d9bd2d84d9a143fdd6baed9c675e6be Mon Sep 17 00:00:00 2001 From: Peter Staar Date: Wed, 18 Sep 2024 07:13:29 +0200 Subject: [PATCH] added the tests for tables Signed-off-by: Peter Staar --- tests/verify_utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/verify_utils.py b/tests/verify_utils.py index 4dae7dda..0e1a7d9e 100644 --- a/tests/verify_utils.py +++ b/tests/verify_utils.py @@ -162,9 +162,13 @@ def verify_conversion_result( doc_pred_pages, doc_true_pages ), f"Mismatch in PDF cell prediction for {input_path}" - assert verify_output( + # assert verify_output( + # doc_pred, doc_true + # ), f"Mismatch in JSON prediction for {input_path}" + + assert verify_tables( doc_pred, doc_true - ), f"Mismatch in JSON prediction for {input_path}" + ), f"verify_tables(doc_pred, doc_true) mismatch for {input_path}" assert verify_md( doc_pred_md, doc_true_md