You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
879 B
Python
24 lines
879 B
Python
1 month ago
|
import pytest
|
||
|
|
|
||
|
|
from marker.schema import BlockTypes
|
||
|
|
from marker.schema.text.line import Line
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.config({"page_range": [0]})
|
||
|
|
def test_document_builder(pdf_document):
|
||
|
|
first_page = pdf_document.pages[0]
|
||
|
|
assert first_page.structure[0] == '/page/0/SectionHeader/0'
|
||
|
|
|
||
|
|
first_block = first_page.get_block(first_page.structure[0])
|
||
|
|
assert first_block.block_type == BlockTypes.SectionHeader
|
||
|
|
assert first_block.text_extraction_method == 'pdftext'
|
||
|
|
|
||
|
|
first_text_block: Line = first_page.get_block(first_block.structure[0])
|
||
|
|
assert first_text_block.block_type == BlockTypes.Line
|
||
|
|
|
||
|
|
first_span = first_page.get_block(first_text_block.structure[0])
|
||
|
|
assert first_span.block_type == BlockTypes.Span
|
||
|
|
assert first_span.text == 'Subspace Adversarial Training'
|
||
|
|
assert first_span.font == 'NimbusRomNo9L-Medi'
|
||
|
|
assert first_span.formats == ['plain']
|