import pytest @pytest.mark.parametrize('raw,expected', [ ('1', 1), (' 2 ', 2), ('0', None), ('abc', None), ]) def test_parse_page(raw, expected): assert parse_page(raw) == expected