def test_window_matches_default_window(self): """ Assert that the window matches the default window """ # Create allocation with no start_date and no end_date allocation = create_allocation(self.increase_date) end_date = datetime(1990, 1, 1, tzinfo=pytz.utc) # Override the default end date to a testable value (start, end) = engine.get_allocation_window( allocation, default_end_date=end_date) self.assertEquals(start, engine._get_zero_date_utc()) self.assertEquals(end, end_date)
def test_window_matches_default_window(self): """ Assert that the window matches the default window """ # Create allocation with no start_date and no end_date allocation = create_allocation(self.increase_date) end_date = datetime(1990, 1, 1, tzinfo=pytz.utc) # Override the default end date to a testable value (start, end) = engine.get_allocation_window(allocation, default_end_date=end_date) self.assertEquals(start, engine._get_zero_date_utc()) self.assertEquals(end, end_date)
def test_get_zero_date_time_is_valid(self): """ Assert that the date time is utc timezone """ zero_datetime = engine._get_zero_date_utc() # Validate that the time corresponds to the desired result self.assertEqual(zero_datetime.tzinfo, pytz.UTC) self.assertEqual(zero_datetime.year, 1970) self.assertEqual(zero_datetime.month, 1) self.assertEqual(zero_datetime.day, 1) self.assertEqual(zero_datetime.hour, 0) self.assertEqual(zero_datetime.minute, 0) self.assertEqual(zero_datetime.second, 0)