import pytest

from marker.schema import BlockTypes
from marker.schema.text.line import Line


@pytest.mark.filename("thinkpython.pdf")
@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 == "Think Python"
    assert first_span.font == "URWPalladioL-Roma"
    assert first_span.formats == ["plain"]


@pytest.mark.config({"page_range": [0]})
def test_document_builder_inline_eq(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 == "surya"

    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.strip() == "Subspace Adversarial Training"
    assert "bold" in first_span.formats