test_day05.py

import day05
import unittest

class Day05(unittest.TestCase):
    lines = """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_part1(self):
        self.assertEqual(day05.solvePart1(self.lines.splitlines()), 35)
    def test_part2(self):
        self.assertEqual(day05.solvePart2(self.lines.splitlines()), 46)

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