import unittest from solutions.day_21 import Solution class Day21TestCase(unittest.TestCase): def setUp(self): self.solution = Solution() def test_something(self): puzzle_input = """ ../.# => ##./#../... .#./..#/### => #..#/..../..../#..# """.strip() assert self.solution.solve(puzzle_input, 2) == 12 if __name__ == "__main__": unittest.main()