MCPcopy Create free account
hub / github.com/docling-project/docling-parse / test_render_single_document

Function test_render_single_document

tests/test_threaded_render.py:119–146  ·  view source on GitHub ↗

Render all pages of one document and verify each result is a valid RGBA image.

()

Source from the content-addressed store, hash-verified

117
118
119def test_render_single_document():
120 """Render all pages of one document and verify each result is a valid RGBA image."""
121 filename = SAMPLE_PDF
122
123 parser = _make_parser()
124 key = parser.load(filename)
125
126 count = 0
127 for result in parser.iterate_results():
128 assert result.doc_key == key
129 assert result.page_number >= 1
130 assert result.success, (
131 f"Render failed page {result.page_number}: {result.error_message}"
132 )
133 assert result.has_image
134
135 image = result.get_image()
136 assert isinstance(image, PILImage.Image)
137 assert image.mode == "RGBA"
138 assert image.width > 0
139 assert image.height > 0
140 assert result.get_page().dimension.rect is not None
141 assert result.timings.total_s > 0
142 assert result.timings.render_page_s >= 0
143
144 count += 1
145
146 assert count == parser.page_count(key)
147
148
149def test_render_image_dimensions_are_consistent():

Callers

nothing calls this directly

Calls 6

iterate_resultsMethod · 0.80
get_imageMethod · 0.80
page_countMethod · 0.80
_make_parserFunction · 0.70
loadMethod · 0.45
get_pageMethod · 0.45

Tested by

no test coverage detected