def test_now_or_future(self):
assert simplify_datetime(now_or_future(None)) == \
simplify_datetime(datetime.now())
assert simplify_datetime(now_or_future(datetime.now() +
timedelta(days=-1))) == simplify_datetime(datetime.now())
assert simplify_datetime(now_or_future(datetime.now() +
timedelta(days=1))) == simplify_datetime(datetime.now() +
timedelta(days=1))
def test_now_or_future(self):
now = datetime(2015, 7, 17, tzinfo=utc)
past = datetime(2015, 7, 16, 23, 59, tzinfo=utc)
future = datetime(2015, 7, 17, 0, 1, tzinfo=utc)
assert now is now_or_future(None, now=now)
assert now is now_or_future(past, now=now)
assert future == now_or_future(future, now=now)
# Now let's perform the same tests with naive datetimes.
past = naive(past)
future = naive(future)
assert now is now_or_future(None, now=now)
assert naive(now) == now_or_future(past, now=now)
assert future == now_or_future(future, now=now)
