mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed the tests

Browse Source 
Signed-off-by: Peter Staar <taa@zurich.ibm.com>
This commit is contained in:
Peter Staar 2024-09-18 06:48:44 +02:00
parent 57a34b6936
commit 25dbb4e2c5
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/data/redp5110.json
View File

File diff suppressed because one or more lines are too long

2
tests/data/redp5110.pages.json
View File

File diff suppressed because one or more lines are too long

6
tests/verify_utils.py
View File

@ -96,15 +96,15 @@ def verify_tables(doc_pred: DsDocument, doc_true: DsDocument):
for i, row in enumerate(true_item.data): for i, row in enumerate(true_item.data):
for j, col in enumerate(true_item.data[i]): for j, col in enumerate(true_item.data[i]):
print("true: ", true_item.data[i][j]) #print("true: ", true_item.data[i][j])
print("pred: ", pred_item.data[i][j]) #print("pred: ", pred_item.data[i][j])
assert ( assert (
true_item.data[i][j].text == pred_item.data[i][j].text true_item.data[i][j].text == pred_item.data[i][j].text
), "table-cell does not have the same text" ), "table-cell does not have the same text"
assert ( assert (
true_item.data[i][j].type == pred_item.data[i][j].type true_item.data[i][j].obj_type == pred_item.data[i][j].obj_type
), "table-cell does not have the same type" ), "table-cell does not have the same type"
return True return True