import unittest from solutions.day_16 import Solution class Day16TestCase(unittest.TestCase): def setUp(self): self.solution = Solution() def test_something(self): puzzle_input = """s1,x3/4,pe/b""".strip() assert self.solution.solve(puzzle_input, 5) == "baedc" if __name__ == "__main__": unittest.main()