def test_set_current_date_with_negative_shift(self, now_mock):
date = datetime(2017, 12, 23, 12, 34, 0)
now_mock.return_value = pytz.timezone(
timezone.get_default_timezone_name()).localize(date)
self.assertEqual("12/23/2017 08:34", set_current_date(shift_hours=-4))
self.assertEqual("12/22/2017 12:34", set_current_date(shift_hours=-24))
def test_set_current_date_with_positive_shift(self, now_mock):
date = datetime(2017, 12, 23, 12, 34, 0)
now_mock.return_value = timezone.get_default_timezone().localize(date)
self.assertEqual("12/23/2017 17:34", set_current_date(shift_hours=5))
self.assertEqual("12/24/2017 12:34", set_current_date(shift_hours=24))