ago/test_ago.py

106 lines
3 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, ms
FUTURE = PRESENT + timedelta( 2, 12447, 963 ) # days, secs, ms
# 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_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['microsecond'] == 963
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()