]>
Commit | Line | Data |
---|---|---|
4710c53d | 1 | import warnings\r |
2 | warnings.filterwarnings("ignore", "strop functions are obsolete;",\r | |
3 | DeprecationWarning,\r | |
4 | r'test.test_strop|unittest')\r | |
5 | import strop\r | |
6 | import unittest\r | |
7 | from test import test_support\r | |
8 | \r | |
9 | \r | |
10 | class StropFunctionTestCase(unittest.TestCase):\r | |
11 | \r | |
12 | def test_atoi(self):\r | |
13 | self.assertTrue(strop.atoi(" 1 ") == 1)\r | |
14 | self.assertRaises(ValueError, strop.atoi, " 1x")\r | |
15 | self.assertRaises(ValueError, strop.atoi, " x1 ")\r | |
16 | \r | |
17 | def test_atol(self):\r | |
18 | self.assertTrue(strop.atol(" 1 ") == 1L)\r | |
19 | self.assertRaises(ValueError, strop.atol, " 1x")\r | |
20 | self.assertRaises(ValueError, strop.atol, " x1 ")\r | |
21 | \r | |
22 | def test_atof(self):\r | |
23 | self.assertTrue(strop.atof(" 1 ") == 1.0)\r | |
24 | self.assertRaises(ValueError, strop.atof, " 1x")\r | |
25 | self.assertRaises(ValueError, strop.atof, " x1 ")\r | |
26 | \r | |
27 | def test_capitalize(self):\r | |
28 | self.assertTrue(strop.capitalize(" hello ") == " hello ")\r | |
29 | self.assertTrue(strop.capitalize("hello ") == "Hello ")\r | |
30 | \r | |
31 | def test_find(self):\r | |
32 | self.assertTrue(strop.find("abcdefghiabc", "abc") == 0)\r | |
33 | self.assertTrue(strop.find("abcdefghiabc", "abc", 1) == 9)\r | |
34 | self.assertTrue(strop.find("abcdefghiabc", "def", 4) == -1)\r | |
35 | \r | |
36 | def test_rfind(self):\r | |
37 | self.assertTrue(strop.rfind("abcdefghiabc", "abc") == 9)\r | |
38 | \r | |
39 | def test_lower(self):\r | |
40 | self.assertTrue(strop.lower("HeLLo") == "hello")\r | |
41 | \r | |
42 | def test_upper(self):\r | |
43 | self.assertTrue(strop.upper("HeLLo") == "HELLO")\r | |
44 | \r | |
45 | def test_swapcase(self):\r | |
46 | self.assertTrue(strop.swapcase("HeLLo cOmpUteRs") == "hEllO CoMPuTErS")\r | |
47 | \r | |
48 | def test_strip(self):\r | |
49 | self.assertTrue(strop.strip(" \t\n hello \t\n ") == "hello")\r | |
50 | \r | |
51 | def test_lstrip(self):\r | |
52 | self.assertTrue(strop.lstrip(" \t\n hello \t\n ") == "hello \t\n ")\r | |
53 | \r | |
54 | def test_rstrip(self):\r | |
55 | self.assertTrue(strop.rstrip(" \t\n hello \t\n ") == " \t\n hello")\r | |
56 | \r | |
57 | def test_replace(self):\r | |
58 | replace = strop.replace\r | |
59 | self.assertTrue(replace("one!two!three!", '!', '@', 1)\r | |
60 | == "one@two!three!")\r | |
61 | self.assertTrue(replace("one!two!three!", '!', '@', 2)\r | |
62 | == "one@two@three!")\r | |
63 | self.assertTrue(replace("one!two!three!", '!', '@', 3)\r | |
64 | == "one@two@three@")\r | |
65 | self.assertTrue(replace("one!two!three!", '!', '@', 4)\r | |
66 | == "one@two@three@")\r | |
67 | \r | |
68 | # CAUTION: a replace count of 0 means infinity only to strop,\r | |
69 | # not to the string .replace() method or to the\r | |
70 | # string.replace() function.\r | |
71 | \r | |
72 | self.assertTrue(replace("one!two!three!", '!', '@', 0)\r | |
73 | == "one@two@three@")\r | |
74 | self.assertTrue(replace("one!two!three!", '!', '@')\r | |
75 | == "one@two@three@")\r | |
76 | self.assertTrue(replace("one!two!three!", 'x', '@')\r | |
77 | == "one!two!three!")\r | |
78 | self.assertTrue(replace("one!two!three!", 'x', '@', 2)\r | |
79 | == "one!two!three!")\r | |
80 | \r | |
81 | def test_split(self):\r | |
82 | split = strop.split\r | |
83 | self.assertTrue(split("this is the split function")\r | |
84 | == ['this', 'is', 'the', 'split', 'function'])\r | |
85 | self.assertTrue(split("a|b|c|d", '|') == ['a', 'b', 'c', 'd'])\r | |
86 | self.assertTrue(split("a|b|c|d", '|', 2) == ['a', 'b', 'c|d'])\r | |
87 | self.assertTrue(split("a b c d", None, 1) == ['a', 'b c d'])\r | |
88 | self.assertTrue(split("a b c d", None, 2) == ['a', 'b', 'c d'])\r | |
89 | self.assertTrue(split("a b c d", None, 3) == ['a', 'b', 'c', 'd'])\r | |
90 | self.assertTrue(split("a b c d", None, 4) == ['a', 'b', 'c', 'd'])\r | |
91 | self.assertTrue(split("a b c d", None, 0) == ['a', 'b', 'c', 'd'])\r | |
92 | self.assertTrue(split("a b c d", None, 2) == ['a', 'b', 'c d'])\r | |
93 | \r | |
94 | def test_join(self):\r | |
95 | self.assertTrue(strop.join(['a', 'b', 'c', 'd']) == 'a b c d')\r | |
96 | self.assertTrue(strop.join(('a', 'b', 'c', 'd'), '') == 'abcd')\r | |
97 | self.assertTrue(strop.join(Sequence()) == 'w x y z')\r | |
98 | \r | |
99 | # try a few long ones\r | |
100 | self.assertTrue(strop.join(['x' * 100] * 100, ':')\r | |
101 | == (('x' * 100) + ":") * 99 + "x" * 100)\r | |
102 | self.assertTrue(strop.join(('x' * 100,) * 100, ':')\r | |
103 | == (('x' * 100) + ":") * 99 + "x" * 100)\r | |
104 | \r | |
105 | def test_maketrans(self):\r | |
106 | self.assertTrue(strop.maketrans("abc", "xyz") == transtable)\r | |
107 | self.assertRaises(ValueError, strop.maketrans, "abc", "xyzq")\r | |
108 | \r | |
109 | def test_translate(self):\r | |
110 | self.assertTrue(strop.translate("xyzabcdef", transtable, "def")\r | |
111 | == "xyzxyz")\r | |
112 | \r | |
113 | def test_data_attributes(self):\r | |
114 | strop.lowercase\r | |
115 | strop.uppercase\r | |
116 | strop.whitespace\r | |
117 | \r | |
118 | @test_support.precisionbigmemtest(size=test_support._2G - 1, memuse=5)\r | |
119 | def test_stropjoin_huge_list(self, size):\r | |
120 | a = "A" * size\r | |
121 | try:\r | |
122 | r = strop.join([a, a], a)\r | |
123 | except OverflowError:\r | |
124 | pass\r | |
125 | else:\r | |
126 | self.assertEqual(len(r), len(a) * 3)\r | |
127 | \r | |
128 | @test_support.precisionbigmemtest(size=test_support._2G - 1, memuse=1)\r | |
129 | def test_stropjoin_huge_tup(self, size):\r | |
130 | a = "A" * size\r | |
131 | try:\r | |
132 | r = strop.join((a, a), a)\r | |
133 | except OverflowError:\r | |
134 | pass # acceptable on 32-bit\r | |
135 | else:\r | |
136 | self.assertEqual(len(r), len(a) * 3)\r | |
137 | \r | |
138 | transtable = '\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037 !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`xyzdefghijklmnopqrstuvwxyz{|}~\177\200\201\202\203\204\205\206\207\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374\375\376\377'\r | |
139 | \r | |
140 | \r | |
141 | # join() now works with any sequence type.\r | |
142 | class Sequence:\r | |
143 | def __init__(self): self.seq = 'wxyz'\r | |
144 | def __len__(self): return len(self.seq)\r | |
145 | def __getitem__(self, i): return self.seq[i]\r | |
146 | \r | |
147 | \r | |
148 | def test_main():\r | |
149 | test_support.run_unittest(StropFunctionTestCase)\r | |
150 | \r | |
151 | \r | |
152 | if __name__ == "__main__":\r | |
153 | test_main()\r |