test_day07.py

import unittest

import aoc
import day07


class TestDay07(unittest.TestCase):

    def test_part1_example1(self):
        input_list = [
            "3,15,3,16,1002,16,10,16,1,16,15,15,4,15,99,0,0"
        ]

        result = day07.part1(input_list)
        self.assertEqual(43210, result)

    def test_part1_example2(self):
        input_list = [
            "3, 23, 3, 24, 1002, 24, 10, 24, 1002, 23, -1, 23, 101, 5, 23, 23, 1, 24, 23, 23, 4,"
            "23, 99, 0, 0"
        ]

        result = day07.part1(input_list)
        self.assertEqual(54321, result)

    def test_part1_example3(self):
        input_list = [
            "3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,"
            "1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0"
        ]

        result = day07.part1(input_list)
        self.assertEqual(65210, result)

    def test_part1_input(self):
        result = day07.part1(aoc.read_input('day07.input'))
        self.assertEqual(11828, result)

    def test_part2_example1(self):
        input_list = [
            "3,26,1001,26,-4,26,3,27,1002,27,2,27,1,27,26,"
            "27,4,27,1001,28,-1,28,1005,28,6,99,0,0,5"
        ]

        result = day07.part2(input_list)
        self.assertEqual(139629729, result)

    def test_part2_example2(self):
        input_list = [
            "3,52,1001,52,-5,52,3,53,1,52,56,54,1007,54,5,55,1005,55,26,1001,54,"
            "-5,54,1105,1,12,1,53,54,53,1008,54,0,55,1001,55,1,55,2,53,55,53,4,"
            "53,1001,56,-1,56,1005,56,6,99,0,0,0,0,10"
        ]

        result = day07.part2(input_list)
        self.assertEqual(18216, result)

    def test_part2_input(self):
        result = day07.part2(aoc.read_input('day07.input'))
        self.assertEqual(1714298, result)


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