test_strop.py 6.7 KB

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