def test_sub_month_keep_leap(self): base_date = datetime.datetime(2016, 3, 31) date = datetime.datetime(2016, 2, 29) trans_dt = ck_utils.sub_month(base_date) self.assertEqual(date, trans_dt)