From 79932b7d69c53c4e70c1a45dbc9a9cf96ef48072 Mon Sep 17 00:00:00 2001 From: Michele Dolfi <97102151+dolfim-ibm@users.noreply.github.com> Date: Wed, 11 Sep 2024 12:53:59 +0200 Subject: [PATCH] test: check for stable obj_type (#70) Signed-off-by: Michele Dolfi --- tests/verify_utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/verify_utils.py b/tests/verify_utils.py index 0d0566d..448b7b6 100644 --- a/tests/verify_utils.py +++ b/tests/verify_utils.py @@ -51,9 +51,14 @@ def verify_maintext(doc_pred: DsDocument, doc_true: DsDocument): ), f"document has different length of main-text than expected. {len(doc_true.main_text)}!={len(doc_pred.main_text)}" for l, true_item in enumerate(doc_true.main_text): - if isinstance(true_item, BaseText): - pred_item = doc_pred.main_text[l] + pred_item = doc_pred.main_text[l] + # Validate type + assert ( + true_item.obj_type == pred_item.obj_type + ), f"Item[{l}] type does not match. expected[{true_item.obj_type}] != predicted [{pred_item.obj_type}]" + # Validate text ceels + if isinstance(true_item, BaseText): assert isinstance( pred_item, BaseText ), f"{pred_item} is not a BaseText element, but {true_item} is."