40 lines
1.2 KiB
Python
40 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()
|