Lib/test/test_strop.py

   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.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
 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()