advent-of-code/2015-python/tests/test_day_12.py

39 lines
1.2 KiB
Python

import unittest
from solutions.day_12 import Solution
class Day12TestCase(unittest.TestCase):
def setUp(self):
self.solution = Solution()
self.puzzle_input = self.solution.parse_input(
"""
<REPLACE ME>
"""
)
def test_parse_puzzle_input(self):
data = """
<REPLACE ME>
""".strip()
assert self.solution.parse_input(data) == "<REPLACE ME>"
def test_solve_first_part(self):
assert self.solution.solve("[1,2,3]") == 6
assert self.solution.solve('{"a":2,"b":4}') == 6
assert self.solution.solve("[[[3]]]") == 3
assert self.solution.solve('{"a":{"b":4},"c":-1}') == 3
assert self.solution.solve('{"a":[-1,1]}') == 0
assert self.solution.solve('-1,{"a":1}]') == 0
assert self.solution.solve("{}") == 0
assert self.solution.solve("[]") == 0
def test_solve_second_part(self):
assert self.solution.solve_again("[1,2,3]") == 6
assert self.solution.solve_again('[1,{"c":"red","b":2},3]') == 4
assert self.solution.solve_again('{"d":"red","e":[1,2,3,4],"f":5}') == 0
assert self.solution.solve_again('[1,"red",5]') == 6
if __name__ == "__main__":
unittest.main()