def test_regular_number(self): initial_value = re.match(r'\d+', '12') expected_result = '13' self.assertEqual(increment_match(initial_value), expected_result)
def test_starting_with_zero(self): initial_value = re.match(r'[0]\d+', '012') expected_result = '013' self.assertEqual(increment_match(initial_value), expected_result)