Function test_threaded_single_document

tests/test_threaded_parse.py:178–208

Test threaded parsing with a single document.

()

def test_threaded_single_document():
    """Test threaded parsing with a single document."""
    filename = SAMPLE_PDF

    parser = DoclingThreadedPdfParser(
        parser_config=ThreadedPdfParserConfig(
            loglevel="fatal",
            threads=2,
            max_concurrent_results=4,
            boundary_type=PdfPageBoundaryType.CROP_BOX,
        ),
        decode_config=_make_decode_config(),
    )

    key = parser.load(filename)
    assert parser.page_count(key) > 0

    count = 0
    for result in parser.iterate_results():
        assert result.success, (
            f"Failed to decode page {result.page_number}: {result.error_message}"
        )
        assert result.doc_key == key
        assert result.page_width > 0
        assert result.page_height > 0
        assert result.get_timings().total() > 0
        assert result.timings.total_s > 0
        assert result.timings.decode_page_s > 0
        count += 1

    assert count == parser.page_count(key)


def test_threaded_results_match_sequential():