42 lines
1.1 KiB
Python
42 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()
|