code_collection/advent_of_code/2025(python)/test_day6.py at master · Allianzcortex/code_collection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import unittest

from day6 import solve_day6_part1, solve_day6_part2

question_input = [
    "123 328  51 64",
    " 45 64  387 23",
    "  6 98  215 314",
    "*   +   *   +  ",
]


class TestSample(unittest.TestCase):

    def test_part1(self):
        self.assertEqual(solve_day6_part1(question_input), 4277556)

    def test_part2(self):
        self.assertEqual(solve_day6_part2(question_input), 3263827)


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