177 lines
4.6 KiB
Python
177 lines
4.6 KiB
Python
from __future__ import print_function
|
|
|
|
from datetime import datetime, timedelta, tzinfo
|
|
|
|
from ago import human
|
|
from ago import delta2dict
|
|
|
|
# datetime objects
|
|
PRESENT = datetime.now()
|
|
PAST = PRESENT - timedelta(492, 58711, 45) # days, secs, microseconds
|
|
FUTURE = PRESENT + timedelta(2, 12447, 967742) # days, secs, microseconds
|
|
|
|
# timedelta objects
|
|
PAST_DELTA = PRESENT - PAST
|
|
FUTURE_DELTA = PRESENT - FUTURE
|
|
ONE_YEAR_FOUR_HOURS_DELTA = timedelta(365, 14400, 0)
|
|
|
|
|
|
class FixedOffset(tzinfo):
|
|
"""Fixed offset in minutes east from UTC.
|
|
|
|
Adapted from Python's documentation about tzinfo objects:
|
|
https://docs.python.org/2.7/library/datetime.html#tzinfo-objects
|
|
"""
|
|
def __init__(self, offset):
|
|
self._offset = timedelta(minutes=offset)
|
|
|
|
def utcoffset(self, dt):
|
|
return self._offset
|
|
|
|
def tzname(self, dt):
|
|
return "Test"
|
|
|
|
def dst(self, dt):
|
|
return timedelta(0)
|
|
|
|
|
|
def test_human_passed_datetime_is_string():
|
|
assert isinstance(human(PAST), str)
|
|
|
|
|
|
def test_human_passed_timedelta_is_string():
|
|
assert isinstance(human(PAST_DELTA), str)
|
|
|
|
|
|
def test_delta2dict_is_dict():
|
|
assert isinstance(delta2dict(PAST_DELTA), dict)
|
|
|
|
|
|
def test_ago_in_past_human():
|
|
assert "ago" in human(PAST)
|
|
|
|
|
|
def test_in_in_future_human():
|
|
assert "in" in human(FUTURE)
|
|
|
|
|
|
def test_no_coma_in_one_precision():
|
|
assert "," not in human(PAST, precision=1)
|
|
assert "," not in human(FUTURE, precision=1)
|
|
|
|
|
|
def test_coma_in_three_precision():
|
|
assert "," in human(PAST, precision=3)
|
|
assert "," in human(FUTURE, precision=3)
|
|
|
|
|
|
def test_coma_in_out_of_bounds_precision():
|
|
assert "," in human(PAST, precision=10)
|
|
assert "," in human(FUTURE, precision=10)
|
|
|
|
|
|
def test_zero_day_is_skipped_display_hour():
|
|
_result = human(ONE_YEAR_FOUR_HOURS_DELTA, precision=2)
|
|
assert "year" in _result
|
|
# day is 0 so it is skipped, so we should show hours ...
|
|
assert "hour" in _result
|
|
|
|
|
|
def test_one_day_singular():
|
|
assert "s" not in human(timedelta(1))
|
|
|
|
|
|
def test_two_day_plural():
|
|
assert "s" in human(timedelta(2))
|
|
|
|
|
|
def test_abbreviation():
|
|
assert "2d ago" in human(timedelta(2), abbreviate=True)
|
|
assert "3d, 12h ago" in human(timedelta(3.5), abbreviate=True)
|
|
assert "2h, 24m ago" in human(timedelta(.1), abbreviate=True)
|
|
assert "1y, 35d ago" in human(timedelta(400), abbreviate=True)
|
|
|
|
|
|
def test_timestamp_integer():
|
|
result = human(1474485467933 / 1000, precision=6)
|
|
assert "minute" in result
|
|
assert "second" in result
|
|
assert "ago" in result
|
|
|
|
|
|
def test_timestamp_float():
|
|
result = human(1474485467933 / 1000.0, precision=6)
|
|
assert "minute" in result
|
|
assert "second" in result
|
|
assert "ago" in result
|
|
|
|
|
|
def test_past_tense():
|
|
output = human(
|
|
PAST,
|
|
past_tense="titanic sunk {} ago",
|
|
future_tense="titanic will sink in {} from now",
|
|
)
|
|
assert "titanic sunk" in output
|
|
|
|
|
|
def test_future_tense():
|
|
output = human(
|
|
FUTURE,
|
|
past_tense="titanic sunk {} ago",
|
|
future_tense="titanic will sink in {} from now",
|
|
)
|
|
assert "titanic will sink in" in output
|
|
|
|
|
|
def test_valid_past_dict():
|
|
past_dict = delta2dict(PAST_DELTA)
|
|
assert past_dict["year"] == 1
|
|
assert past_dict["day"] == 127
|
|
assert past_dict["hour"] == 16
|
|
assert past_dict["minute"] == 18
|
|
assert past_dict["microsecond"] == 45
|
|
|
|
|
|
def test_valid_future_dict():
|
|
past_dict = delta2dict(FUTURE_DELTA)
|
|
assert past_dict["year"] == 0
|
|
assert past_dict["day"] == 2
|
|
assert past_dict["hour"] == 3
|
|
assert past_dict["minute"] == 27
|
|
assert past_dict["millisecond"] == 967
|
|
assert past_dict["microsecond"] == 742
|
|
|
|
|
|
def test_timezone_support():
|
|
# test a naive datetime with no timezone
|
|
# (add an extra minute to ensure it stays over an hour ahead during test)
|
|
dt = datetime.now() + timedelta(minutes=61)
|
|
output = human(dt)
|
|
assert output.startswith("in 1 hour")
|
|
|
|
# test a timezone-aware datetime with a UTC-2 timezone
|
|
dt = datetime.now(tz=FixedOffset(-120)) + timedelta(minutes=61)
|
|
output = human(dt)
|
|
assert output.startswith("in 1 hour")
|
|
|
|
|
|
def example_usage():
|
|
"""Test and example usage"""
|
|
|
|
print("\nTest past tense:\n")
|
|
print(delta2dict(PAST_DELTA))
|
|
print("Commented " + human(PAST_DELTA, 1))
|
|
print(human(PAST, past_tense="Commented {} ago"))
|
|
|
|
print(human(ONE_YEAR_FOUR_HOURS_DELTA, past_tense="Posted {} ago"))
|
|
|
|
print("\nTest future tense:\n")
|
|
print(delta2dict(FUTURE_DELTA))
|
|
print("Shutdown " + human(FUTURE_DELTA, 5))
|
|
print(human(FUTURE, future_tense="Shutdown in {} from now"))
|
|
print("")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
example_usage()
|