diff --git a/2024/src/aoc/days/day5.py b/2024/src/aoc/days/day5.py index cd9c30f..3458a3b 100644 --- a/2024/src/aoc/days/day5.py +++ b/2024/src/aoc/days/day5.py @@ -1,5 +1,5 @@ -from collections import defaultdict import functools +from collections import defaultdict from . import CombinedRunner @@ -21,7 +21,7 @@ def is_correct(update: list[int], must_after: dict[int, set[int]]) -> bool: return False forbidden |= must_after.get(entry, set()) - + return True diff --git a/2024/tests/test_day5.py b/2024/tests/test_day5.py index b00e7db..0677b64 100644 --- a/2024/tests/test_day5.py +++ b/2024/tests/test_day5.py @@ -2,6 +2,7 @@ import os from aoc.days.day5 import DayRunner + def get_data() -> str: sample = os.path.dirname(__file__) + "/samples/05.txt" with open(sample, mode="rt", encoding="utf-8") as f: @@ -13,6 +14,7 @@ def test_sample_part1() -> None: assert DayRunner.part1(data) == 143 + def test_sample_part2() -> None: data = get_data()