From 9a5e29f5d74238a40a659f016405ea51c885c5c7 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Wed, 11 Sep 2024 12:22:14 +0200 Subject: [PATCH] test: check for stable obj_type 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 0d0566d9..448b7b61 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."