tests/renderers/test_chunk_renderer.py

import pytest

from marker.renderers.chunk import ChunkRenderer


@pytest.mark.config({"page_range": [0]})
def test_chunk_renderer(pdf_document):
    renderer = ChunkRenderer()
    chunk_output = renderer(pdf_document)
    blocks = chunk_output.blocks
    page_info = chunk_output.page_info

    assert len(blocks) == 14
    assert blocks[0].block_type == "SectionHeader"
    assert page_info[0]["bbox"] is not None
    assert page_info[0]["polygon"] is not None

    figure_groups = [block for block in blocks if block.block_type == "FigureGroup"]
    figures = [block for block in blocks if block.block_type == "Figure"]
    captions = [block for block in blocks if block.block_type == "Caption"]

    assert len(figure_groups) == 1
    assert len(figures) == 0
    assert len(captions) == 0

    figure_group = figure_groups[0]
    assert figure_group.images is not None
    assert len(figure_group.images) == 1
    assert "<img src='/page/0/Figure/9'>" in figure_group.html