def test_add_month_keep_leap(self): base_date = datetime.datetime(2016, 1, 31) date = datetime.datetime(2016, 2, 29) trans_dt = ck_utils.add_month(base_date) self.assertEqual(date, trans_dt)