def test_day_1(): import day_01 assert day_01.day_01(1122) == 3 assert day_01.day_01(1111) == 4 assert day_01.day_01(1234) == 0 assert day_01.day_01(91212129) == 9
def test_sum_multiple(self): # Replacement for example 2, this one does not assume wraparound self.assertEqual(2, day_01("1112", -1))
def test_sum_consecutive(self): self.assertEqual(3, day_01("1122", -1))
def test_part2(self): self.assertEqual(6, day_01("1212", 2)) self.assertEqual(0, day_01("1221", 2)) self.assertEqual(4, day_01("123425", 3)) self.assertEqual(12, day_01("123123", 3)) self.assertEqual(4, day_01("12131415", 4))
def test_wraparound(self): self.assertEqual(9, day_01("91212129", -1)) self.assertEqual(4, day_01("1111", -1))
def test_none(self): self.assertEqual(0, day_01("1234", -1))