day_05_test.py

import unittest

from day_05 import parse_almanac, part_1, part_2


class TestDay05(unittest.TestCase):
    _seeds: list[int]
    _maps: list[list[tuple[range, range]]]

    def setUp(self):
        self._seeds, self._maps = parse_almanac('''seeds: 79 14 55 13

seed-to-soil map:
50 98 2
52 50 48

soil-to-fertilizer map:
0 15 37
37 52 2
39 0 15

fertilizer-to-water map:
49 53 8
0 11 42
42 0 7
57 7 4

water-to-light map:
88 18 7
18 25 70

light-to-temperature map:
45 77 23
81 45 19
68 64 13

temperature-to-humidity map:
0 69 1
1 0 69

humidity-to-location map:
60 56 37
56 93 4''')

    def test_part_1(self):
        self.assertEqual(part_1(self._seeds, self._maps), 35)

    def test_part_2(self):
        self.assertEqual(part_2(self._seeds, self._maps), 46)


if __name__ == '__main__':
    unittest.main()