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

import unittest

from solutions.day_01 import Solution


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

    def test_sums_equal_pairs(self):
        assert self.solution.solve("1122") == 3
        assert self.solution.solve("1111") == 4
        assert self.solution.solve("1234") == 0
        assert self.solution.solve("91212129") == 9

    def test_sums_equal_pairs_halvway_around(self):
        assert self.solution.solve_again("1212") == 6
        assert self.solution.solve_again("1221") == 0
        assert self.solution.solve_again("123425") == 4
        assert self.solution.solve_again("123123") == 12
        assert self.solution.solve_again("12131415") == 4


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