mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 12:48:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply ruff formatting to test file

Browse Source 
Co-authored-by: cau-git <60343111+cau-git@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2025-11-19 13:28:01 +00:00
parent 0788e714a9
commit 6fe6aae91a
1 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_page_range_bug.py
View File

@@ -65,7 +65,9 @@ def test_page_range_beyond_32():
# Verify that the page numbers are correct # Verify that the page numbers are correct
page_numbers2 = [p.page_no for p in result2.pages] page_numbers2 = [p.page_no for p in result2.pages]
expected_page_nos = list(range(29, 29 + expected_pages2)) # 0-indexed: 29, 30, 31, 32, 33, 34 expected_page_nos = list(
range(29, 29 + expected_pages2)
) # 0-indexed: 29, 30, 31, 32, 33, 34
assert page_numbers2 == expected_page_nos, ( assert page_numbers2 == expected_page_nos, (
f"Expected page numbers {expected_page_nos}, got {page_numbers2}" f"Expected page numbers {expected_page_nos}, got {page_numbers2}"
) )
@@ -119,13 +121,17 @@ def test_page_range_edge_cases():
if total_pages >= 32: if total_pages >= 32:
result = converter.convert(test_pdf, page_range=(32, 32)) result = converter.convert(test_pdf, page_range=(32, 32))
assert len(result.pages) == 1, f"Expected 1 page, got {len(result.pages)}" assert len(result.pages) == 1, f"Expected 1 page, got {len(result.pages)}"
assert result.pages[0].page_no == 31, f"Expected page_no=31, got {result.pages[0].page_no}" assert result.pages[0].page_no == 31, (
f"Expected page_no=31, got {result.pages[0].page_no}"
)
# Edge case 2: Single page after boundary (page 33) # Edge case 2: Single page after boundary (page 33)
if total_pages >= 33: if total_pages >= 33:
result = converter.convert(test_pdf, page_range=(33, 33)) result = converter.convert(test_pdf, page_range=(33, 33))
assert len(result.pages) == 1, f"Expected 1 page, got {len(result.pages)}" assert len(result.pages) == 1, f"Expected 1 page, got {len(result.pages)}"
assert result.pages[0].page_no == 32, f"Expected page_no=32, got {result.pages[0].page_no}" assert result.pages[0].page_no == 32, (
f"Expected page_no=32, got {result.pages[0].page_no}"
)
# Edge case 3: Range crossing the boundary (31-34) # Edge case 3: Range crossing the boundary (31-34)
if total_pages >= 34: if total_pages >= 34: