ago/test_ago.py
2013-01-12 14:27:23 -05:00

81 lines
2.2 KiB
Python

from datetime import datetime
from datetime import timedelta
from types import *
from ago import human
from ago import delta2human
from ago import delta2dict
PRESENT = datetime.now()
PAST = PRESENT - timedelta( 492, 58711, 45 ) # days, secs, ms
FUTURE = PRESENT + timedelta( 2, 12447, 963 ) # days, secs, ms
PAST_DELTA = PRESENT - PAST
FUTURE_DELTA = PRESENT - FUTURE
def test_human_is_string():
assert type(human( PAST )) is StringType
def test_delta2human_is_string():
assert type(delta2human( PAST_DELTA )) is StringType
def test_delta2dict_is_dict():
assert type(delta2dict( PAST_DELTA )) is DictType
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_one_day_singular():
assert 's' not in delta2human( timedelta(1) )
def test_two_day_plural():
assert 's' in delta2human( 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_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
def test_output():
"""Test and example usage"""
print '\nTesting past:\n'
print delta2dict( PAST_DELTA )
print 'Commented ' + delta2human( PAST_DELTA, 3 )
print human( PAST, past_tense = "Commented {} ago" )
print '\nTesting future:\n'
print delta2dict( FUTURE_DELTA )
print 'Shutdown ' + delta2human( FUTURE_DELTA, 3 )
print human( FUTURE, future_tense = 'Shutdown in {} from now' )
print ''