| from datetime import timedelta, date |
| |
| from django.template import Template, Context, add_to_builtins |
| from django.utils import unittest |
| from django.utils.dateformat import DateFormat |
| from django.utils.translation import ugettext as _ |
| from django.utils.html import escape |
| |
| add_to_builtins('django.contrib.humanize.templatetags.humanize') |
| |
| class HumanizeTests(unittest.TestCase): |
| |
| def humanize_tester(self, test_list, result_list, method): |
| # Using max below ensures we go through both lists |
| # However, if the lists are not equal length, this raises an exception |
| for index in xrange(max(len(test_list), len(result_list))): |
| test_content = test_list[index] |
| t = Template('{{ test_content|%s }}' % method) |
| rendered = t.render(Context(locals())).strip() |
| self.assertEqual(rendered, escape(result_list[index]), |
| msg="%s test failed, produced %s, should've produced %s" % (method, rendered, result_list[index])) |
| |
| def test_ordinal(self): |
| test_list = ('1','2','3','4','11','12', |
| '13','101','102','103','111', |
| 'something else', None) |
| result_list = ('1st', '2nd', '3rd', '4th', '11th', |
| '12th', '13th', '101st', '102nd', '103rd', |
| '111th', 'something else', None) |
| |
| self.humanize_tester(test_list, result_list, 'ordinal') |
| |
| def test_intcomma(self): |
| test_list = (100, 1000, 10123, 10311, 1000000, 1234567.25, |
| '100', '1000', '10123', '10311', '1000000', '1234567.1234567', |
| None) |
| result_list = ('100', '1,000', '10,123', '10,311', '1,000,000', '1,234,567.25', |
| '100', '1,000', '10,123', '10,311', '1,000,000', '1,234,567.1234567', |
| None) |
| |
| self.humanize_tester(test_list, result_list, 'intcomma') |
| |
| def test_intword(self): |
| test_list = ('100', '1000000', '1200000', '1290000', |
| '1000000000','2000000000','6000000000000', |
| None) |
| result_list = ('100', '1.0 million', '1.2 million', '1.3 million', |
| '1.0 billion', '2.0 billion', '6.0 trillion', |
| None) |
| |
| self.humanize_tester(test_list, result_list, 'intword') |
| |
| def test_apnumber(self): |
| test_list = [str(x) for x in range(1, 11)] |
| test_list.append(None) |
| result_list = (u'one', u'two', u'three', u'four', u'five', u'six', |
| u'seven', u'eight', u'nine', u'10', None) |
| |
| self.humanize_tester(test_list, result_list, 'apnumber') |
| |
| def test_naturalday(self): |
| from django.template import defaultfilters |
| today = date.today() |
| yesterday = today - timedelta(days=1) |
| tomorrow = today + timedelta(days=1) |
| someday = today - timedelta(days=10) |
| notdate = u"I'm not a date value" |
| |
| test_list = (today, yesterday, tomorrow, someday, notdate, None) |
| someday_result = defaultfilters.date(someday) |
| result_list = (_(u'today'), _(u'yesterday'), _(u'tomorrow'), |
| someday_result, u"I'm not a date value", None) |
| self.humanize_tester(test_list, result_list, 'naturalday') |
| |
| if __name__ == '__main__': |
| unittest.main() |
| |
