def month_add(value, arg): if isinstance(value, datetime.date): try: arg = int(arg) months = value.month + arg if months <= 12: return datetime.date(year=value.year, month=months, day=value.day) else: years = months//12 months = months % 12 or 12 return datetime.date(year=value.year+years, month=months, day=value.day) except (ValueError, TypeError): pass result = add(value, arg) if isinstance(result, int): return result % 12 or 12 else: return result
def test_add(self): self.assertEqual(add('1', '2'), 3)
def test_add(self): self.assertEqual(add("1", "2"), 3)