43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from solutions.day_07 import Solution, Program
|
||
|
|
|
||
|
|
|
||
|
|
class Day7TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.puzzle_input = '''
|
||
|
|
pbga (66)
|
||
|
|
xhth (57)
|
||
|
|
ebii (61)
|
||
|
|
havc (66)
|
||
|
|
ktlj (57)
|
||
|
|
fwft (72) -> ktlj, cntj, xhth
|
||
|
|
qoyq (66)
|
||
|
|
padx (45) -> pbga, havc, qoyq
|
||
|
|
tknk (41) -> ugml, padx, fwft
|
||
|
|
jptl (61)
|
||
|
|
ugml (68) -> gyxo, ebii, jptl
|
||
|
|
gyxo (61)
|
||
|
|
cntj (57)
|
||
|
|
'''.strip()
|
||
|
|
self.solution = Solution()
|
||
|
|
|
||
|
|
def test_find_bottom_tower(self):
|
||
|
|
p = Program('ugml (68) -> gyxo, ebii, jptl')
|
||
|
|
assert p.name == 'ugml'
|
||
|
|
assert p.weight == 68
|
||
|
|
assert p.disc == ('gyxo', 'ebii', 'jptl')
|
||
|
|
p = Program('jptl (61)')
|
||
|
|
assert p.name == 'jptl'
|
||
|
|
assert p.weight == 61
|
||
|
|
assert p.disc == ()
|
||
|
|
assert self.solution.solve(self.puzzle_input).name == 'tknk'
|
||
|
|
|
||
|
|
def test_find_weight_correction(self):
|
||
|
|
corrected = self.solution.solve_again(self.puzzle_input)
|
||
|
|
assert corrected == 60
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|