ago/test_ago.py
russellballestrini b2f6af9d70 The result of running black .
changed ago.py
changed setup.py
changed test_ago.py
2018-05-25 17:33:10 -04:00

146 lines
3.7 KiB
Python

from __future__ import print_function
from datetime import datetime
from datetime import timedelta
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)
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 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()