advent-of-code/2017-python/tests/day_19_tests.py

34 lines
768 B
Python

import unittest
from solutions.day_19 import Solution
class Day19TestCase(unittest.TestCase):
def setUp(self):
self.solution = Solution()
def test_seen(self):
puzzle_input = """ |
| +--+
A | C
F---|----E|--+
| | | D
+B-+ +--+
"""
assert self.solution.solve(puzzle_input) == "ABCDEF"
def test_steps(self):
puzzle_input = """ |
| +--+
A | C
F---|----E|--+
| | | D
+B-+ +--+
"""
assert self.solution.solve_again(puzzle_input) == 38
if __name__ == "__main__":
unittest.main()