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 |
|