PP-OCRv6_Demo/tests/test_page_spec.py

14 lines
311 B
Python

import pytest
from ocr_app.pdf import parse_page_spec
def test_page_ranges():
assert parse_page_spec("1-2,4,6-", 7) == [0, 1, 3, 5, 6]
@pytest.mark.parametrize("value", ["0", "3-2", "1-a", "8"])
def test_invalid_page_ranges(value):
with pytest.raises(ValueError):
parse_page_spec(value, 7)