Files
adventofcode/2024/tests/test_day2.py
2024-12-02 08:55:57 +01:00

27 lines
641 B
Python

import os
from aoc.days.day2 import DayRunner, is_savable
def get_data() -> str:
sample = os.path.dirname(__file__) + "/samples/02.txt"
with open(sample, mode="rt", encoding="utf-8") as f:
return f.read()
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
def test_sample_part2() -> None:
assert DayRunner.part2(get_data()) == 4