def test_no_delta_type_ago_exception(self):
extend_date_time_obj = ExtendedDateTime(2)
with self.assertRaises(Exception) as context:
extend_date_time_obj.ago
self.assertTrue(
'Need days or weeks or months or years to compute datetime delta',
str(context.exception))
def test_years_after_delta(self):
years_gap = ExtendedDateTime(1)
current_date = datetime.now().date()
delta = relativedelta(years=1)
expected_date = current_date + delta
calculated_date = years_gap.years.after.date()
self.assertEqual(calculated_date, expected_date)
def test_months_after_delta(self):
months_gap = ExtendedDateTime(4)
current_date = datetime.now().date()
delta = relativedelta(months=4)
expected_date = current_date + delta
calculated_date = months_gap.months.after.date()
self.assertEqual(calculated_date, expected_date)
def test_weeks_after_delta(self):
weeks_gap = ExtendedDateTime(3)
current_date = datetime.now().date()
delta = relativedelta(weeks=3)
expected_date = current_date + delta
calculated_date = weeks_gap.weeks.after.date()
self.assertEqual(calculated_date, expected_date)
def test_days_after_delta(self):
days_gap = ExtendedDateTime(2)
current_date = datetime.now().date()
delta = relativedelta(days=2)
expected_date = current_date + delta
calculated_date = days_gap.days.after.date()
self.assertEqual(calculated_date, expected_date)
def test_instance_type(self):
extend_date_time_obj = ExtendedDateTime(2)
self.assertIsInstance(extend_date_time_obj.days.ago, datetime)