| 1 | import unittest |
| 2 | from django.template import Template, Context, add_to_builtins |
| 3 | |
| 4 | add_to_builtins('django.contrib.humanize.templatetags.humanize') |
| 5 | |
| 6 | class HumanizeTests(unittest.TestCase): |
| 7 | |
| 8 | def humanize_tester(self, test_list, result_list, method): |
| 9 | # Using max below ensures we go through both lists |
| 10 | # However, if the lists are not equal length, this raises an exception |
| 11 | for index in xrange(len(max(test_list,result_list))): |
| 12 | test_content = test_list[index] |
| 13 | t = Template('{{ test_content|%s }}' % method) |
| 14 | rendered = t.render(Context(locals())).strip() |
| 15 | self.assertEqual(rendered, result_list[index], |
| 16 | msg="""%s test failed, produced %s, |
| 17 | should've produced %s""" % (method, rendered, result_list[index])) |
| 18 | |
| 19 | def test_ordinal(self): |
| 20 | test_list = ('1','2','3','4','11','12', |
| 21 | '13','101','102','103','111', |
| 22 | 'something else') |
| 23 | result_list = ('1st', '2nd', '3rd', '4th', '11th', |
| 24 | '12th', '13th', '101st', '102nd', '103rd', |
| 25 | '111th', 'something else') |
| 26 | |
| 27 | self.humanize_tester(test_list, result_list, 'ordinal') |
| 28 | |
| 29 | def test_intcomma(self): |
| 30 | test_list = ('100','1000','10123','10311','1000000') |
| 31 | result_list = ('100', '1,000', '10,123', '10,311', '1,000,000') |
| 32 | |
| 33 | self.humanize_tester(test_list, result_list, 'intcomma') |
| 34 | |
| 35 | def test_intword(self): |
| 36 | test_list = ('100', '1000000', '1200000', '1290000', |
| 37 | '1000000000','2000000000','6000000000000') |
| 38 | result_list = ('100', '1.0 million', '1.2 million', '1.3 million', |
| 39 | '1.0 billion', '2.0 billion', '6.0 trillion') |
| 40 | |
| 41 | self.humanize_tester(test_list, result_list, 'intword') |
| 42 | |
| 43 | def test_apnumber(self): |
| 44 | test_list = [str(x) for x in xrange(1,11)] |
| 45 | result_list = ('one', 'two', 'three', 'four', 'five', 'six', |
| 46 | 'seven', 'eight', 'nine', '10') |
| 47 | |
| 48 | self.humanize_tester(test_list, result_list, 'apnumber') |
| 49 | |
| 50 | if __name__ == '__main__': |
| 51 | unittest.main() |
| 52 | |