mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-26 20:14:47 +00:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import glob
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from docling.backend.docling_parse_backend import DoclingParseDocumentBackend, DoclingParsePageBackend
|
|
from docling.datamodel.base_models import BoundingBox
|
|
|
|
from docling.document_converter import DocumentConverter
|
|
|
|
@pytest.fixture
|
|
def test_doc_path():
|
|
return Path("./data/2206.01062.pdf")
|
|
|
|
def test_get_text_from_rect(test_doc_path):
|
|
doc_backend = DoclingParseDocumentBackend(test_doc_path)
|
|
page_backend: DoclingParsePageBackend = doc_backend.load_page(0)
|
|
|
|
# Get the title text of the DocLayNet paper
|
|
textpiece = page_backend.get_text_in_rect(bbox=BoundingBox(l=102,t=77,r=511,b=124))
|
|
ref = "DocLayNet: A Large Human-Annotated Dataset for Document-Layout Analysis"
|
|
|
|
assert textpiece.strip() == ref
|
|
|
|
def test_crop_page_image(test_doc_path):
|
|
doc_backend = DoclingParseDocumentBackend(test_doc_path)
|
|
page_backend: DoclingParsePageBackend = doc_backend.load_page(0)
|
|
|
|
# Crop out "Figure 1" from the DocLayNet paper
|
|
im = page_backend.get_page_image(scale=2, cropbox=BoundingBox(l=317,t=246,r=574,b=527))
|
|
# im.show()
|
|
|
|
def test_num_pages(test_doc_path):
|
|
doc_backend = DoclingParseDocumentBackend(test_doc_path)
|
|
doc_backend.page_count() == 9
|
|
|