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']