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

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()