Python DateUtils.sum_one_month_to_date示例

示例#1

文件： dates_generator.py 项目： nicomin/Desafio_Uno

class DatesGenerator:
    def __init__(self, begin_date: datetime, end_date: datetime):
        self.date_utils = DateUtils()
        self.begin_date = begin_date
        self.end_date = end_date

    def generate(self):
        if not self.begin_date or not self.end_date:
            raise DatesApplicationException(
                DatesApplicationException.ErrorType.BAD_PARAMETERS)
        current_date = self.begin_date
        period_dates = []
        while current_date <= self.end_date:
            period_dates.append(current_date)
            current_date = self.date_utils.sum_one_month_to_date(current_date)
        return period_dates

示例#2

class TestDateUtils(TestCase):
    def setUp(self):
        self.utils = DateUtils()

    def test_get_date_from_date_str_when_date_str_then_return_datetime_obj(
            self):
        date_str = '1970-02-01'
        result = self.utils.get_date_from_date_str(date_str)
        self.assertEqual(1, result.day)
        self.assertEqual(2, result.month)
        self.assertEqual(1970, result.year)

    def test_sum_one_month_to_date_when_date_then_return_that_date_one_month_on_future(
            self):
        input_date = datetime(1970, 1, 1)
        result = self.utils.sum_one_month_to_date(input_date)

        self.assertEqual(1, result.day)
        self.assertEqual(2, result.month)
        self.assertEqual(1970, result.year)