import unittest

from solutions.day_09 import Solution


class Day9TestCase(unittest.TestCase):
    def setUp(self):
        self.solution = Solution()

    def test_calculates_score(self):
        assert self.solution.solve("{}") == 1
        assert self.solution.solve("{{{}}}") == 6
        assert self.solution.solve("{{},{}}") == 5
        assert self.solution.solve("{{{},{},{{}}}}") == 16
        assert self.solution.solve("{<a>,<a>,<a>,<a>}") == 1
        assert self.solution.solve("{{<ab>},{<ab>},{<ab>},{<ab>}}") == 9
        assert self.solution.solve("{{<!!>},{<!!>},{<!!>},{<!!>}}") == 9
        assert self.solution.solve("{{<a!>},{<a!>},{<a!>},{<ab>}}") == 3

    def test_count_garbage(self):
        assert self.solution.solve_again("<>") == 0
        assert self.solution.solve_again("<random characters>") == 17
        assert self.solution.solve_again("<<<<>") == 3
        assert self.solution.solve_again("<{!>}>") == 2
        assert self.solution.solve_again("<!!>") == 0
        assert self.solution.solve_again("<!!!>>") == 0
        assert self.solution.solve_again('<{o"i!a,<{i<a>') == 10


if __name__ == "__main__":
    unittest.main()