/Lib/test/test_strop.py

  1import warnings
  2warnings.filterwarnings("ignore", "strop functions are obsolete;",
  3                        DeprecationWarning,
  4                        r'test.test_strop|unittest')
  5import strop
  6import unittest
  7from test import test_support
  8
  9
 10class StropFunctionTestCase(unittest.TestCase):
 11
 12    def test_atoi(self):
 13        self.assert_(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.assert_(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.assert_(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.assert_(strop.capitalize(" hello ") == " hello ")
 29        self.assert_(strop.capitalize("hello ") == "Hello ")
 30
 31    def test_find(self):
 32        self.assert_(strop.find("abcdefghiabc", "abc") == 0)
 33        self.assert_(strop.find("abcdefghiabc", "abc", 1) == 9)
 34        self.assert_(strop.find("abcdefghiabc", "def", 4) == -1)
 35
 36    def test_rfind(self):
 37        self.assert_(strop.rfind("abcdefghiabc", "abc") == 9)
 38
 39    def test_lower(self):
 40        self.assert_(strop.lower("HeLLo") == "hello")
 41
 42    def test_upper(self):
 43        self.assert_(strop.upper("HeLLo") == "HELLO")
 44
 45    def test_swapcase(self):
 46        self.assert_(strop.swapcase("HeLLo cOmpUteRs") == "hEllO CoMPuTErS")
 47
 48    def test_strip(self):
 49        self.assert_(strop.strip(" \t\n hello \t\n ") == "hello")
 50
 51    def test_lstrip(self):
 52        self.assert_(strop.lstrip(" \t\n hello \t\n ") == "hello \t\n ")
 53
 54    def test_rstrip(self):
 55        self.assert_(strop.rstrip(" \t\n hello \t\n ") == " \t\n hello")
 56
 57    def test_replace(self):
 58        replace = strop.replace
 59        self.assert_(replace("one!two!three!", '!', '@', 1)
 60                     == "one@two!three!")
 61        self.assert_(replace("one!two!three!", '!', '@', 2)
 62                     == "one@two@three!")
 63        self.assert_(replace("one!two!three!", '!', '@', 3)
 64                     == "one@two@three@")
 65        self.assert_(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.assert_(replace("one!two!three!", '!', '@', 0)
 73                     == "one@two@three@")
 74        self.assert_(replace("one!two!three!", '!', '@')
 75                     == "one@two@three@")
 76        self.assert_(replace("one!two!three!", 'x', '@')
 77                     == "one!two!three!")
 78        self.assert_(replace("one!two!three!", 'x', '@', 2)
 79                     == "one!two!three!")
 80
 81    def test_split(self):
 82        split = strop.split
 83        self.assert_(split("this is the split function")
 84                     == ['this', 'is', 'the', 'split', 'function'])
 85        self.assert_(split("a|b|c|d", '|') == ['a', 'b', 'c', 'd'])
 86        self.assert_(split("a|b|c|d", '|', 2) == ['a', 'b', 'c|d'])
 87        self.assert_(split("a b c d", None, 1) == ['a', 'b c d'])
 88        self.assert_(split("a b c d", None, 2) == ['a', 'b', 'c d'])
 89        self.assert_(split("a b c d", None, 3) == ['a', 'b', 'c', 'd'])
 90        self.assert_(split("a b c d", None, 4) == ['a', 'b', 'c', 'd'])
 91        self.assert_(split("a b c d", None, 0) == ['a', 'b', 'c', 'd'])
 92        self.assert_(split("a  b  c  d", None, 2) ==  ['a', 'b', 'c  d'])
 93
 94    def test_join(self):
 95        self.assert_(strop.join(['a', 'b', 'c', 'd']) == 'a b c d')
 96        self.assert_(strop.join(('a', 'b', 'c', 'd'), '') == 'abcd')
 97        self.assert_(strop.join(Sequence()) == 'w x y z')
 98
 99        # try a few long ones
100        self.assert_(strop.join(['x' * 100] * 100, ':')
101                     == (('x' * 100) + ":") * 99 + "x" * 100)
102        self.assert_(strop.join(('x' * 100,) * 100, ':')
103                     == (('x' * 100) + ":") * 99 + "x" * 100)
104
105    def test_maketrans(self):
106        self.assert_(strop.maketrans("abc", "xyz") == transtable)
107        self.assertRaises(ValueError, strop.maketrans, "abc", "xyzq")
108
109    def test_translate(self):
110        self.assert_(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.assertEquals(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.assertEquals(len(r), len(a) * 3)
137
138transtable = '\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.
142class 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
148def test_main():
149    test_support.run_unittest(StropFunctionTestCase)
150
151
152if __name__ == "__main__":
153    test_main()