import pytest from main import get_highest, get_highest_dynamic @pytest.mark.parametrize('given, expected', [ ('987654321111111', 98), ('811111111111119', 89), ('234234234234278', 78), ('818181911112111', 92) ]) def test_get_highest(given, expected): assert get_highest(given) == expected @pytest.mark.parametrize('given, expected', [ ('987654321111111', 987654321111), ('811111111111119', 811111111119), ('234234234234278', 434234234278), ('818181911112111', 888911112111), ('4123535244222342322334342233754335452333242522124322242423331132232242422443224231234323332243364522', 755554464522) ]) def test_get_even_highester(given, expected): assert get_highest_dynamic(12, given) == expected