| 1 | import warnings
|
|---|
| 2 | warnings.filterwarnings("ignore", "strop functions are obsolete;",
|
|---|
| 3 | DeprecationWarning,
|
|---|
| 4 | r'test.test_strop|unittest')
|
|---|
| 5 | import strop
|
|---|
| 6 | import unittest
|
|---|
| 7 | from test import test_support
|
|---|
| 8 |
|
|---|
| 9 |
|
|---|
| 10 | class StropFunctionTestCase(unittest.TestCase):
|
|---|
| 11 |
|
|---|
| 12 | def test_atoi(self):
|
|---|
| 13 | self.assertTrue(strop.atoi(" 1 ") == 1)
|
|---|
| 14 | self.assertRaises(ValueError, strop.atoi, " 1x")
|
|---|
| 15 | self.assertRaises(ValueError, strop.atoi, " x1 ")
|
|---|
| 16 |
|
|---|
| 17 | def test_atol(self):
|
|---|
| 18 | self.assertTrue(strop.atol(" 1 ") == 1L)
|
|---|
| 19 | self.assertRaises(ValueError, strop.atol, " 1x")
|
|---|
| 20 | self.assertRaises(ValueError, strop.atol, " x1 ")
|
|---|
| 21 |
|
|---|
| 22 | def test_atof(self):
|
|---|
| 23 | self.assertTrue(strop.atof(" 1 ") == 1.0)
|
|---|
| 24 | self.assertRaises(ValueError, strop.atof, " 1x")
|
|---|
| 25 | self.assertRaises(ValueError, strop.atof, " x1 ")
|
|---|
| 26 |
|
|---|
| 27 | def test_capitalize(self):
|
|---|
| 28 | self.assertTrue(strop.capitalize(" hello ") == " hello ")
|
|---|
| 29 | self.assertTrue(strop.capitalize("hello ") == "Hello ")
|
|---|
| 30 |
|
|---|
| 31 | def test_find(self):
|
|---|
| 32 | self.assertTrue(strop.find("abcdefghiabc", "abc") == 0)
|
|---|
| 33 | self.assertTrue(strop.find("abcdefghiabc", "abc", 1) == 9)
|
|---|
| 34 | self.assertTrue(strop.find("abcdefghiabc", "def", 4) == -1)
|
|---|
| 35 |
|
|---|
| 36 | def test_rfind(self):
|
|---|
| 37 | self.assertTrue(strop.rfind("abcdefghiabc", "abc") == 9)
|
|---|
| 38 |
|
|---|
| 39 | def test_lower(self):
|
|---|
| 40 | self.assertTrue(strop.lower("HeLLo") == "hello")
|
|---|
| 41 |
|
|---|
| 42 | def test_upper(self):
|
|---|
| 43 | self.assertTrue(strop.upper("HeLLo") == "HELLO")
|
|---|
| 44 |
|
|---|
| 45 | def test_swapcase(self):
|
|---|
| 46 | self.assertTrue(strop.swapcase("HeLLo cOmpUteRs") == "hEllO CoMPuTErS")
|
|---|
| 47 |
|
|---|
| 48 | def test_strip(self):
|
|---|
| 49 | self.assertTrue(strop.strip(" \t\n hello \t\n ") == "hello")
|
|---|
| 50 |
|
|---|
| 51 | def test_lstrip(self):
|
|---|
| 52 | self.assertTrue(strop.lstrip(" \t\n hello \t\n ") == "hello \t\n ")
|
|---|
| 53 |
|
|---|
| 54 | def test_rstrip(self):
|
|---|
| 55 | self.assertTrue(strop.rstrip(" \t\n hello \t\n ") == " \t\n hello")
|
|---|
| 56 |
|
|---|
| 57 | def test_replace(self):
|
|---|
| 58 | replace = strop.replace
|
|---|
| 59 | self.assertTrue(replace("one!two!three!", '!', '@', 1)
|
|---|
| 60 | == "one@two!three!")
|
|---|
| 61 | self.assertTrue(replace("one!two!three!", '!', '@', 2)
|
|---|
| 62 | == "one@two@three!")
|
|---|
| 63 | self.assertTrue(replace("one!two!three!", '!', '@', 3)
|
|---|
| 64 | == "one@two@three@")
|
|---|
| 65 | self.assertTrue(replace("one!two!three!", '!', '@', 4)
|
|---|
| 66 | == "one@two@three@")
|
|---|
| 67 |
|
|---|
| 68 | # CAUTION: a replace count of 0 means infinity only to strop,
|
|---|
| 69 | # not to the string .replace() method or to the
|
|---|
| 70 | # string.replace() function.
|
|---|
| 71 |
|
|---|
| 72 | self.assertTrue(replace("one!two!three!", '!', '@', 0)
|
|---|
| 73 | == "one@two@three@")
|
|---|
| 74 | self.assertTrue(replace("one!two!three!", '!', '@')
|
|---|
| 75 | == "one@two@three@")
|
|---|
| 76 | self.assertTrue(replace("one!two!three!", 'x', '@')
|
|---|
| 77 | == "one!two!three!")
|
|---|
| 78 | self.assertTrue(replace("one!two!three!", 'x', '@', 2)
|
|---|
| 79 | == "one!two!three!")
|
|---|
| 80 |
|
|---|
| 81 | def test_split(self):
|
|---|
| 82 | split = strop.split
|
|---|
| 83 | self.assertTrue(split("this is the split function")
|
|---|
| 84 | == ['this', 'is', 'the', 'split', 'function'])
|
|---|
| 85 | self.assertTrue(split("a|b|c|d", '|') == ['a', 'b', 'c', 'd'])
|
|---|
| 86 | self.assertTrue(split("a|b|c|d", '|', 2) == ['a', 'b', 'c|d'])
|
|---|
| 87 | self.assertTrue(split("a b c d", None, 1) == ['a', 'b c d'])
|
|---|
| 88 | self.assertTrue(split("a b c d", None, 2) == ['a', 'b', 'c d'])
|
|---|
| 89 | self.assertTrue(split("a b c d", None, 3) == ['a', 'b', 'c', 'd'])
|
|---|
| 90 | self.assertTrue(split("a b c d", None, 4) == ['a', 'b', 'c', 'd'])
|
|---|
| 91 | self.assertTrue(split("a b c d", None, 0) == ['a', 'b', 'c', 'd'])
|
|---|
| 92 | self.assertTrue(split("a b c d", None, 2) == ['a', 'b', 'c d'])
|
|---|
| 93 |
|
|---|
| 94 | def test_join(self):
|
|---|
| 95 | self.assertTrue(strop.join(['a', 'b', 'c', 'd']) == 'a b c d')
|
|---|
| 96 | self.assertTrue(strop.join(('a', 'b', 'c', 'd'), '') == 'abcd')
|
|---|
| 97 | self.assertTrue(strop.join(Sequence()) == 'w x y z')
|
|---|
| 98 |
|
|---|
| 99 | # try a few long ones
|
|---|
| 100 | self.assertTrue(strop.join(['x' * 100] * 100, ':')
|
|---|
| 101 | == (('x' * 100) + ":") * 99 + "x" * 100)
|
|---|
| 102 | self.assertTrue(strop.join(('x' * 100,) * 100, ':')
|
|---|
| 103 | == (('x' * 100) + ":") * 99 + "x" * 100)
|
|---|
| 104 |
|
|---|
| 105 | def test_maketrans(self):
|
|---|
| 106 | self.assertTrue(strop.maketrans("abc", "xyz") == transtable)
|
|---|
| 107 | self.assertRaises(ValueError, strop.maketrans, "abc", "xyzq")
|
|---|
| 108 |
|
|---|
| 109 | def test_translate(self):
|
|---|
| 110 | self.assertTrue(strop.translate("xyzabcdef", transtable, "def")
|
|---|
| 111 | == "xyzxyz")
|
|---|
| 112 |
|
|---|
| 113 | def test_data_attributes(self):
|
|---|
| 114 | strop.lowercase
|
|---|
| 115 | strop.uppercase
|
|---|
| 116 | strop.whitespace
|
|---|
| 117 |
|
|---|
| 118 | @test_support.precisionbigmemtest(size=test_support._2G - 1, memuse=5)
|
|---|
| 119 | def test_stropjoin_huge_list(self, size):
|
|---|
| 120 | a = "A" * size
|
|---|
| 121 | try:
|
|---|
| 122 | r = strop.join([a, a], a)
|
|---|
| 123 | except OverflowError:
|
|---|
| 124 | pass
|
|---|
| 125 | else:
|
|---|
| 126 | self.assertEqual(len(r), len(a) * 3)
|
|---|
| 127 |
|
|---|
| 128 | @test_support.precisionbigmemtest(size=test_support._2G - 1, memuse=1)
|
|---|
| 129 | def test_stropjoin_huge_tup(self, size):
|
|---|
| 130 | a = "A" * size
|
|---|
| 131 | try:
|
|---|
| 132 | r = strop.join((a, a), a)
|
|---|
| 133 | except OverflowError:
|
|---|
| 134 | pass # acceptable on 32-bit
|
|---|
| 135 | else:
|
|---|
| 136 | self.assertEqual(len(r), len(a) * 3)
|
|---|
| 137 |
|
|---|
| 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'
|
|---|
| 139 |
|
|---|
| 140 |
|
|---|
| 141 | # join() now works with any sequence type.
|
|---|
| 142 | class Sequence:
|
|---|
| 143 | def __init__(self): self.seq = 'wxyz'
|
|---|
| 144 | def __len__(self): return len(self.seq)
|
|---|
| 145 | def __getitem__(self, i): return self.seq[i]
|
|---|
| 146 |
|
|---|
| 147 |
|
|---|
| 148 | def test_main():
|
|---|
| 149 | test_support.run_unittest(StropFunctionTestCase)
|
|---|
| 150 |
|
|---|
| 151 |
|
|---|
| 152 | if __name__ == "__main__":
|
|---|
| 153 | test_main()
|
|---|
