Skip to content

Commit 07a4cc6

Browse files
committed
revert temp test changes
Signed-off-by: Panos Vagenas <pva@zurich.ibm.com>
1 parent b7d6aef commit 07a4cc6

File tree

1 file changed

+2
-34
lines changed

1 file changed

+2
-34
lines changed

test/test_docling_doc.py

Lines changed: 2 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def test_reference_doc():
416416
_test_serialize_and_reload(doc)
417417

418418
# Call Export methods
419-
_test_export_methods(doc, filename=filename, test_annotations=True)
419+
_test_export_methods(doc, filename=filename)
420420

421421

422422
def test_parse_doc():
@@ -497,10 +497,7 @@ def _verify_regression_test(pred: str, filename: str, ext: str):
497497

498498

499499
def _test_export_methods(
500-
doc: DoclingDocument,
501-
filename: str,
502-
page_break_placeholder: Optional[str] = None,
503-
test_annotations: bool = False,
500+
doc: DoclingDocument, filename: str, page_break_placeholder: Optional[str] = None
504501
):
505502
# Iterate all elements
506503
et_pred = doc.export_to_element_tree()
@@ -510,27 +507,6 @@ def _test_export_methods(
510507
md_pred = doc.export_to_markdown()
511508
_verify_regression_test(md_pred, filename=filename, ext="md")
512509

513-
# if test_annotations:
514-
# # include_annotations=True and mark_annotations=False is the default
515-
# # include_annotations=False and mark_annotations=True does not make sense
516-
517-
# _verify_regression_test(
518-
# pred=doc.export_to_markdown(
519-
# include_annotations=False,
520-
# mark_annotations=False,
521-
# ),
522-
# filename=filename,
523-
# ext="include_annotations_false.md",
524-
# )
525-
# _verify_regression_test(
526-
# pred=doc.export_to_markdown(
527-
# include_annotations=True,
528-
# mark_annotations=True,
529-
# ),
530-
# filename=filename,
531-
# ext="mark_annotations_true.md",
532-
# )
533-
534510
if page_break_placeholder is not None:
535511
md_pred = doc.export_to_markdown(page_break_placeholder=page_break_placeholder)
536512
_verify_regression_test(md_pred, filename=filename, ext="paged.md")
@@ -539,14 +515,6 @@ def _test_export_methods(
539515
html_pred = doc.export_to_html()
540516
_verify_regression_test(html_pred, filename=filename, ext="html")
541517

542-
# if test_annotations:
543-
# # include_annotations=True is the default
544-
# _verify_regression_test(
545-
# pred=doc.export_to_html(include_annotations=False),
546-
# filename=filename,
547-
# ext="include_annotations_false.html",
548-
# )
549-
550518
# Test DocTags export ...
551519
dt_pred = doc.export_to_doctags()
552520
_verify_regression_test(dt_pred, filename=filename, ext="dt")

0 commit comments

Comments
 (0)