from aoc.days.day2 import DayRunner, is_savable

from . import get_data


def test_individual_samples() -> None:
    assert is_savable("7 6 4 2 1")
    assert not is_savable("1 2 7 8 9")
    assert not is_savable("9 7 6 2 1")
    assert is_savable("1 3 2 4 5")
    assert is_savable("8 6 4 4 1")
    assert is_savable("1 3 6 7 9")


def test_sample_part1() -> None:
    assert DayRunner.part1(get_data(2)) == 2


def test_sample_part2() -> None:
    assert DayRunner.part2(get_data(2)) == 4