Fix the block count and test for images in test_chunk_renderer.

tests/renderers/test_chunk_renderer.py

@@ -4,13 +4,26 @@ from marker.renderers.chunk import ChunkRenderer
 
 
 @pytest.mark.config({"page_range": [0]})
-def test_markdown_renderer_pagination(pdf_document):
+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) == 15
+    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