test_day03.py

import day03
import unittest

class Day03(unittest.TestCase):
    def test_part1(self):
        lines = """467..114..
...*......
..35..633.
......#...
617*......
.....+.58.
..592.....
......755.
...$.*....
.664.598.."""
        self.assertEqual(day03.solvePart1(lines.splitlines()), 4361)
    def test_part2(self):
        lines = """467+.114..
+..*......
..35..633.
......#...
617*......
.....+.58.
..592.....
......755.
...$.*....
.664.598.."""
        self.assertEqual(day03.solvePart2(lines.splitlines()), 467835)

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