test_bigmem.py 38 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082
  1. from test import test_support
  2. from test.test_support import bigmemtest, _1G, _2G, _4G, precisionbigmemtest
  3. import unittest
  4. import operator
  5. import string
  6. import sys
  7. # Bigmem testing houserules:
  8. #
  9. # - Try not to allocate too many large objects. It's okay to rely on
  10. # refcounting semantics, but don't forget that 's = create_largestring()'
  11. # doesn't release the old 's' (if it exists) until well after its new
  12. # value has been created. Use 'del s' before the create_largestring call.
  13. #
  14. # - Do *not* compare large objects using assertEqual or similar. It's a
  15. # lengty operation and the errormessage will be utterly useless due to
  16. # its size. To make sure whether a result has the right contents, better
  17. # to use the strip or count methods, or compare meaningful slices.
  18. #
  19. # - Don't forget to test for large indices, offsets and results and such,
  20. # in addition to large sizes.
  21. #
  22. # - When repeating an object (say, a substring, or a small list) to create
  23. # a large object, make the subobject of a length that is not a power of
  24. # 2. That way, int-wrapping problems are more easily detected.
  25. #
  26. # - While the bigmemtest decorator speaks of 'minsize', all tests will
  27. # actually be called with a much smaller number too, in the normal
  28. # test run (5Kb currently.) This is so the tests themselves get frequent
  29. # testing. Consequently, always make all large allocations based on the
  30. # passed-in 'size', and don't rely on the size being very large. Also,
  31. # memuse-per-size should remain sane (less than a few thousand); if your
  32. # test uses more, adjust 'size' upward, instead.
  33. if test_support.have_unicode:
  34. character_size = 4 if sys.maxunicode > 0xFFFF else 2
  35. else:
  36. character_size = 1
  37. class StrTest(unittest.TestCase):
  38. @bigmemtest(minsize=_2G, memuse=2)
  39. def test_capitalize(self, size):
  40. SUBSTR = ' abc def ghi'
  41. s = '-' * size + SUBSTR
  42. caps = s.capitalize()
  43. self.assertEqual(caps[-len(SUBSTR):],
  44. SUBSTR.capitalize())
  45. self.assertEqual(caps.lstrip('-'), SUBSTR)
  46. @bigmemtest(minsize=_2G + 10, memuse=1)
  47. def test_center(self, size):
  48. SUBSTR = ' abc def ghi'
  49. s = SUBSTR.center(size)
  50. self.assertEqual(len(s), size)
  51. lpadsize = rpadsize = (len(s) - len(SUBSTR)) // 2
  52. if len(s) % 2:
  53. lpadsize += 1
  54. self.assertEqual(s[lpadsize:-rpadsize], SUBSTR)
  55. self.assertEqual(s.strip(), SUBSTR.strip())
  56. @test_support.requires_unicode
  57. @precisionbigmemtest(size=_2G - 1, memuse=character_size)
  58. def test_center_unicode(self, size):
  59. SUBSTR = u' abc def ghi'
  60. try:
  61. s = SUBSTR.center(size)
  62. except OverflowError:
  63. pass # acceptable on 32-bit
  64. else:
  65. self.assertEqual(len(s), size)
  66. lpadsize = rpadsize = (len(s) - len(SUBSTR)) // 2
  67. if len(s) % 2:
  68. lpadsize += 1
  69. self.assertEqual(s[lpadsize:-rpadsize], SUBSTR)
  70. self.assertEqual(s.strip(), SUBSTR.strip())
  71. del s
  72. @bigmemtest(minsize=_2G, memuse=2)
  73. def test_count(self, size):
  74. SUBSTR = ' abc def ghi'
  75. s = '.' * size + SUBSTR
  76. self.assertEqual(s.count('.'), size)
  77. s += '.'
  78. self.assertEqual(s.count('.'), size + 1)
  79. self.assertEqual(s.count(' '), 3)
  80. self.assertEqual(s.count('i'), 1)
  81. self.assertEqual(s.count('j'), 0)
  82. @test_support.requires_unicode
  83. @bigmemtest(minsize=_2G + 2, memuse=1 + character_size)
  84. def test_decode(self, size):
  85. s = '.' * size
  86. self.assertEqual(len(s.decode('utf-8')), size)
  87. def basic_encode_test(self, size, enc, c=u'.', expectedsize=None):
  88. if expectedsize is None:
  89. expectedsize = size
  90. s = c * size
  91. self.assertEqual(len(s.encode(enc)), expectedsize)
  92. @test_support.requires_unicode
  93. @bigmemtest(minsize=_2G + 2, memuse=character_size + 4)
  94. def test_encode(self, size):
  95. self.basic_encode_test(size, 'utf-8')
  96. @test_support.requires_unicode
  97. @precisionbigmemtest(size=_4G // 6 + 2, memuse=character_size + 6)
  98. def test_encode_raw_unicode_escape(self, size):
  99. self.basic_encode_test(size, 'raw_unicode_escape')
  100. @test_support.requires_unicode
  101. @precisionbigmemtest(size=_4G // 5 + 70, memuse=character_size + 8)
  102. def test_encode_utf7(self, size):
  103. self.basic_encode_test(size, 'utf7')
  104. @test_support.requires_unicode
  105. @precisionbigmemtest(size=_4G // 4 + 5, memuse=character_size + 4)
  106. def test_encode_utf32(self, size):
  107. self.basic_encode_test(size, 'utf32', expectedsize=4*size+4)
  108. @test_support.requires_unicode
  109. @precisionbigmemtest(size=_2G-1, memuse=4)
  110. def test_decodeascii(self, size):
  111. self.basic_encode_test(size, 'ascii', c='A')
  112. @bigmemtest(minsize=_2G, memuse=2)
  113. def test_endswith(self, size):
  114. SUBSTR = ' abc def ghi'
  115. s = '-' * size + SUBSTR
  116. self.assertTrue(s.endswith(SUBSTR))
  117. self.assertTrue(s.endswith(s))
  118. s2 = '...' + s
  119. self.assertTrue(s2.endswith(s))
  120. self.assertFalse(s.endswith('a' + SUBSTR))
  121. self.assertFalse(SUBSTR.endswith(s))
  122. @bigmemtest(minsize=_2G + 10, memuse=2)
  123. def test_expandtabs(self, size):
  124. s = '-' * size
  125. tabsize = 8
  126. self.assertEqual(s.expandtabs(), s)
  127. del s
  128. slen, remainder = divmod(size, tabsize)
  129. s = ' \t' * slen
  130. s = s.expandtabs(tabsize)
  131. self.assertEqual(len(s), size - remainder)
  132. self.assertEqual(len(s.strip(' ')), 0)
  133. @bigmemtest(minsize=_2G, memuse=2)
  134. def test_find(self, size):
  135. SUBSTR = ' abc def ghi'
  136. sublen = len(SUBSTR)
  137. s = ''.join([SUBSTR, '-' * size, SUBSTR])
  138. self.assertEqual(s.find(' '), 0)
  139. self.assertEqual(s.find(SUBSTR), 0)
  140. self.assertEqual(s.find(' ', sublen), sublen + size)
  141. self.assertEqual(s.find(SUBSTR, len(SUBSTR)), sublen + size)
  142. self.assertEqual(s.find('i'), SUBSTR.find('i'))
  143. self.assertEqual(s.find('i', sublen),
  144. sublen + size + SUBSTR.find('i'))
  145. self.assertEqual(s.find('i', size),
  146. sublen + size + SUBSTR.find('i'))
  147. self.assertEqual(s.find('j'), -1)
  148. @bigmemtest(minsize=_2G, memuse=2)
  149. def test_index(self, size):
  150. SUBSTR = ' abc def ghi'
  151. sublen = len(SUBSTR)
  152. s = ''.join([SUBSTR, '-' * size, SUBSTR])
  153. self.assertEqual(s.index(' '), 0)
  154. self.assertEqual(s.index(SUBSTR), 0)
  155. self.assertEqual(s.index(' ', sublen), sublen + size)
  156. self.assertEqual(s.index(SUBSTR, sublen), sublen + size)
  157. self.assertEqual(s.index('i'), SUBSTR.index('i'))
  158. self.assertEqual(s.index('i', sublen),
  159. sublen + size + SUBSTR.index('i'))
  160. self.assertEqual(s.index('i', size),
  161. sublen + size + SUBSTR.index('i'))
  162. self.assertRaises(ValueError, s.index, 'j')
  163. @bigmemtest(minsize=_2G, memuse=2)
  164. def test_isalnum(self, size):
  165. SUBSTR = '123456'
  166. s = 'a' * size + SUBSTR
  167. self.assertTrue(s.isalnum())
  168. s += '.'
  169. self.assertFalse(s.isalnum())
  170. @bigmemtest(minsize=_2G, memuse=2)
  171. def test_isalpha(self, size):
  172. SUBSTR = 'zzzzzzz'
  173. s = 'a' * size + SUBSTR
  174. self.assertTrue(s.isalpha())
  175. s += '.'
  176. self.assertFalse(s.isalpha())
  177. @bigmemtest(minsize=_2G, memuse=2)
  178. def test_isdigit(self, size):
  179. SUBSTR = '123456'
  180. s = '9' * size + SUBSTR
  181. self.assertTrue(s.isdigit())
  182. s += 'z'
  183. self.assertFalse(s.isdigit())
  184. @bigmemtest(minsize=_2G, memuse=2)
  185. def test_islower(self, size):
  186. chars = ''.join([ chr(c) for c in range(255) if not chr(c).isupper() ])
  187. repeats = size // len(chars) + 2
  188. s = chars * repeats
  189. self.assertTrue(s.islower())
  190. s += 'A'
  191. self.assertFalse(s.islower())
  192. @bigmemtest(minsize=_2G, memuse=2)
  193. def test_isspace(self, size):
  194. whitespace = ' \f\n\r\t\v'
  195. repeats = size // len(whitespace) + 2
  196. s = whitespace * repeats
  197. self.assertTrue(s.isspace())
  198. s += 'j'
  199. self.assertFalse(s.isspace())
  200. @bigmemtest(minsize=_2G, memuse=2)
  201. def test_istitle(self, size):
  202. SUBSTR = '123456'
  203. s = ''.join(['A', 'a' * size, SUBSTR])
  204. self.assertTrue(s.istitle())
  205. s += 'A'
  206. self.assertTrue(s.istitle())
  207. s += 'aA'
  208. self.assertFalse(s.istitle())
  209. @bigmemtest(minsize=_2G, memuse=2)
  210. def test_isupper(self, size):
  211. chars = ''.join([ chr(c) for c in range(255) if not chr(c).islower() ])
  212. repeats = size // len(chars) + 2
  213. s = chars * repeats
  214. self.assertTrue(s.isupper())
  215. s += 'a'
  216. self.assertFalse(s.isupper())
  217. @bigmemtest(minsize=_2G, memuse=2)
  218. def test_join(self, size):
  219. s = 'A' * size
  220. x = s.join(['aaaaa', 'bbbbb'])
  221. self.assertEqual(x.count('a'), 5)
  222. self.assertEqual(x.count('b'), 5)
  223. self.assertTrue(x.startswith('aaaaaA'))
  224. self.assertTrue(x.endswith('Abbbbb'))
  225. @bigmemtest(minsize=_2G + 10, memuse=1)
  226. def test_ljust(self, size):
  227. SUBSTR = ' abc def ghi'
  228. s = SUBSTR.ljust(size)
  229. self.assertTrue(s.startswith(SUBSTR + ' '))
  230. self.assertEqual(len(s), size)
  231. self.assertEqual(s.strip(), SUBSTR.strip())
  232. @bigmemtest(minsize=_2G + 10, memuse=2)
  233. def test_lower(self, size):
  234. s = 'A' * size
  235. s = s.lower()
  236. self.assertEqual(len(s), size)
  237. self.assertEqual(s.count('a'), size)
  238. @bigmemtest(minsize=_2G + 10, memuse=1)
  239. def test_lstrip(self, size):
  240. SUBSTR = 'abc def ghi'
  241. s = SUBSTR.rjust(size)
  242. self.assertEqual(len(s), size)
  243. self.assertEqual(s.lstrip(), SUBSTR.lstrip())
  244. del s
  245. s = SUBSTR.ljust(size)
  246. self.assertEqual(len(s), size)
  247. stripped = s.lstrip()
  248. self.assertTrue(stripped is s)
  249. @bigmemtest(minsize=_2G + 10, memuse=2)
  250. def test_replace(self, size):
  251. replacement = 'a'
  252. s = ' ' * size
  253. s = s.replace(' ', replacement)
  254. self.assertEqual(len(s), size)
  255. self.assertEqual(s.count(replacement), size)
  256. s = s.replace(replacement, ' ', size - 4)
  257. self.assertEqual(len(s), size)
  258. self.assertEqual(s.count(replacement), 4)
  259. self.assertEqual(s[-10:], ' aaaa')
  260. @bigmemtest(minsize=_2G, memuse=2)
  261. def test_rfind(self, size):
  262. SUBSTR = ' abc def ghi'
  263. sublen = len(SUBSTR)
  264. s = ''.join([SUBSTR, '-' * size, SUBSTR])
  265. self.assertEqual(s.rfind(' '), sublen + size + SUBSTR.rfind(' '))
  266. self.assertEqual(s.rfind(SUBSTR), sublen + size)
  267. self.assertEqual(s.rfind(' ', 0, size), SUBSTR.rfind(' '))
  268. self.assertEqual(s.rfind(SUBSTR, 0, sublen + size), 0)
  269. self.assertEqual(s.rfind('i'), sublen + size + SUBSTR.rfind('i'))
  270. self.assertEqual(s.rfind('i', 0, sublen), SUBSTR.rfind('i'))
  271. self.assertEqual(s.rfind('i', 0, sublen + size),
  272. SUBSTR.rfind('i'))
  273. self.assertEqual(s.rfind('j'), -1)
  274. @bigmemtest(minsize=_2G, memuse=2)
  275. def test_rindex(self, size):
  276. SUBSTR = ' abc def ghi'
  277. sublen = len(SUBSTR)
  278. s = ''.join([SUBSTR, '-' * size, SUBSTR])
  279. self.assertEqual(s.rindex(' '),
  280. sublen + size + SUBSTR.rindex(' '))
  281. self.assertEqual(s.rindex(SUBSTR), sublen + size)
  282. self.assertEqual(s.rindex(' ', 0, sublen + size - 1),
  283. SUBSTR.rindex(' '))
  284. self.assertEqual(s.rindex(SUBSTR, 0, sublen + size), 0)
  285. self.assertEqual(s.rindex('i'),
  286. sublen + size + SUBSTR.rindex('i'))
  287. self.assertEqual(s.rindex('i', 0, sublen), SUBSTR.rindex('i'))
  288. self.assertEqual(s.rindex('i', 0, sublen + size),
  289. SUBSTR.rindex('i'))
  290. self.assertRaises(ValueError, s.rindex, 'j')
  291. @bigmemtest(minsize=_2G + 10, memuse=1)
  292. def test_rjust(self, size):
  293. SUBSTR = ' abc def ghi'
  294. s = SUBSTR.ljust(size)
  295. self.assertTrue(s.startswith(SUBSTR + ' '))
  296. self.assertEqual(len(s), size)
  297. self.assertEqual(s.strip(), SUBSTR.strip())
  298. @bigmemtest(minsize=_2G + 10, memuse=1)
  299. def test_rstrip(self, size):
  300. SUBSTR = ' abc def ghi'
  301. s = SUBSTR.ljust(size)
  302. self.assertEqual(len(s), size)
  303. self.assertEqual(s.rstrip(), SUBSTR.rstrip())
  304. del s
  305. s = SUBSTR.rjust(size)
  306. self.assertEqual(len(s), size)
  307. stripped = s.rstrip()
  308. self.assertTrue(stripped is s)
  309. # The test takes about size bytes to build a string, and then about
  310. # sqrt(size) substrings of sqrt(size) in size and a list to
  311. # hold sqrt(size) items. It's close but just over 2x size.
  312. @bigmemtest(minsize=_2G, memuse=2.1)
  313. def test_split_small(self, size):
  314. # Crudely calculate an estimate so that the result of s.split won't
  315. # take up an inordinate amount of memory
  316. chunksize = int(size ** 0.5 + 2)
  317. SUBSTR = 'a' + ' ' * chunksize
  318. s = SUBSTR * chunksize
  319. l = s.split()
  320. self.assertEqual(len(l), chunksize)
  321. self.assertEqual(set(l), set(['a']))
  322. del l
  323. l = s.split('a')
  324. self.assertEqual(len(l), chunksize + 1)
  325. self.assertEqual(set(l), set(['', ' ' * chunksize]))
  326. # Allocates a string of twice size (and briefly two) and a list of
  327. # size. Because of internal affairs, the s.split() call produces a
  328. # list of size times the same one-character string, so we only
  329. # suffer for the list size. (Otherwise, it'd cost another 48 times
  330. # size in bytes!) Nevertheless, a list of size takes
  331. # 8*size bytes.
  332. @bigmemtest(minsize=_2G + 5, memuse=10)
  333. def test_split_large(self, size):
  334. s = ' a' * size + ' '
  335. l = s.split()
  336. self.assertEqual(len(l), size)
  337. self.assertEqual(set(l), set(['a']))
  338. del l
  339. l = s.split('a')
  340. self.assertEqual(len(l), size + 1)
  341. self.assertEqual(set(l), set([' ']))
  342. @bigmemtest(minsize=_2G, memuse=2.1)
  343. def test_splitlines(self, size):
  344. # Crudely calculate an estimate so that the result of s.split won't
  345. # take up an inordinate amount of memory
  346. chunksize = int(size ** 0.5 + 2) // 2
  347. SUBSTR = ' ' * chunksize + '\n' + ' ' * chunksize + '\r\n'
  348. s = SUBSTR * chunksize
  349. l = s.splitlines()
  350. self.assertEqual(len(l), chunksize * 2)
  351. self.assertEqual(set(l), set([' ' * chunksize]))
  352. @bigmemtest(minsize=_2G, memuse=2)
  353. def test_startswith(self, size):
  354. SUBSTR = ' abc def ghi'
  355. s = '-' * size + SUBSTR
  356. self.assertTrue(s.startswith(s))
  357. self.assertTrue(s.startswith('-' * size))
  358. self.assertFalse(s.startswith(SUBSTR))
  359. @bigmemtest(minsize=_2G, memuse=1)
  360. def test_strip(self, size):
  361. SUBSTR = ' abc def ghi '
  362. s = SUBSTR.rjust(size)
  363. self.assertEqual(len(s), size)
  364. self.assertEqual(s.strip(), SUBSTR.strip())
  365. del s
  366. s = SUBSTR.ljust(size)
  367. self.assertEqual(len(s), size)
  368. self.assertEqual(s.strip(), SUBSTR.strip())
  369. @bigmemtest(minsize=_2G, memuse=2)
  370. def test_swapcase(self, size):
  371. SUBSTR = "aBcDeFG12.'\xa9\x00"
  372. sublen = len(SUBSTR)
  373. repeats = size // sublen + 2
  374. s = SUBSTR * repeats
  375. s = s.swapcase()
  376. self.assertEqual(len(s), sublen * repeats)
  377. self.assertEqual(s[:sublen * 3], SUBSTR.swapcase() * 3)
  378. self.assertEqual(s[-sublen * 3:], SUBSTR.swapcase() * 3)
  379. @bigmemtest(minsize=_2G, memuse=2)
  380. def test_title(self, size):
  381. SUBSTR = 'SpaaHAaaAaham'
  382. s = SUBSTR * (size // len(SUBSTR) + 2)
  383. s = s.title()
  384. self.assertTrue(s.startswith((SUBSTR * 3).title()))
  385. self.assertTrue(s.endswith(SUBSTR.lower() * 3))
  386. @bigmemtest(minsize=_2G, memuse=2)
  387. def test_translate(self, size):
  388. trans = string.maketrans('.aZ', '-!$')
  389. SUBSTR = 'aZz.z.Aaz.'
  390. sublen = len(SUBSTR)
  391. repeats = size // sublen + 2
  392. s = SUBSTR * repeats
  393. s = s.translate(trans)
  394. self.assertEqual(len(s), repeats * sublen)
  395. self.assertEqual(s[:sublen], SUBSTR.translate(trans))
  396. self.assertEqual(s[-sublen:], SUBSTR.translate(trans))
  397. self.assertEqual(s.count('.'), 0)
  398. self.assertEqual(s.count('!'), repeats * 2)
  399. self.assertEqual(s.count('z'), repeats * 3)
  400. @bigmemtest(minsize=_2G + 5, memuse=2)
  401. def test_upper(self, size):
  402. s = 'a' * size
  403. s = s.upper()
  404. self.assertEqual(len(s), size)
  405. self.assertEqual(s.count('A'), size)
  406. @bigmemtest(minsize=_2G + 20, memuse=1)
  407. def test_zfill(self, size):
  408. SUBSTR = '-568324723598234'
  409. s = SUBSTR.zfill(size)
  410. self.assertTrue(s.endswith('0' + SUBSTR[1:]))
  411. self.assertTrue(s.startswith('-0'))
  412. self.assertEqual(len(s), size)
  413. self.assertEqual(s.count('0'), size - len(SUBSTR))
  414. @bigmemtest(minsize=_2G + 10, memuse=2)
  415. def test_format(self, size):
  416. s = '-' * size
  417. sf = '%s' % (s,)
  418. self.assertTrue(s == sf)
  419. del sf
  420. sf = '..%s..' % (s,)
  421. self.assertEqual(len(sf), len(s) + 4)
  422. self.assertTrue(sf.startswith('..-'))
  423. self.assertTrue(sf.endswith('-..'))
  424. del s, sf
  425. size //= 2
  426. edge = '-' * size
  427. s = ''.join([edge, '%s', edge])
  428. del edge
  429. s = s % '...'
  430. self.assertEqual(len(s), size * 2 + 3)
  431. self.assertEqual(s.count('.'), 3)
  432. self.assertEqual(s.count('-'), size * 2)
  433. @bigmemtest(minsize=_2G + 10, memuse=5)
  434. def test_repr_small(self, size):
  435. s = '-' * size
  436. s = repr(s)
  437. self.assertEqual(len(s), size + 2)
  438. self.assertEqual(s[0], "'")
  439. self.assertEqual(s[-1], "'")
  440. self.assertEqual(s.count('-'), size)
  441. del s
  442. # repr() will create a string four times as large as this 'binary
  443. # string', but we don't want to allocate much more than twice
  444. # size in total. (We do extra testing in test_repr_large())
  445. s = '\x00' * size
  446. s = repr(s)
  447. self.assertEqual(len(s), size * 4 + 2)
  448. self.assertEqual(s[0], "'")
  449. self.assertEqual(s[-1], "'")
  450. self.assertEqual(s.count('\\'), size)
  451. self.assertEqual(s.count('0'), size * 2)
  452. @bigmemtest(minsize=_2G + 10, memuse=5)
  453. def test_repr_large(self, size):
  454. s = '\x00' * size
  455. s = repr(s)
  456. self.assertEqual(len(s), size * 4 + 2)
  457. self.assertEqual(s[0], "'")
  458. self.assertEqual(s[-1], "'")
  459. self.assertEqual(s.count('\\'), size)
  460. self.assertEqual(s.count('0'), size * 2)
  461. @test_support.requires_unicode
  462. @bigmemtest(minsize=2**32 // 6, memuse=character_size + 6)
  463. def test_unicode_repr(self, size):
  464. s = unichr(0xABCD) * size
  465. try:
  466. r = repr(s)
  467. self.assertEqual(len(r), 3 + 6 * size)
  468. self.assertTrue(r.endswith(r"\uabcd'"), r[-10:])
  469. finally:
  470. s = r = None
  471. @test_support.requires_unicode
  472. @precisionbigmemtest(size=_4G // 6 + 1, memuse=character_size + 6)
  473. def test_unicode_repr_oflw(self, size):
  474. s = unichr(0xABCD) * size
  475. try:
  476. r = repr(s)
  477. self.assertEqual(len(r), 3 + 6 * size)
  478. self.assertTrue(r.endswith(r"\uabcd'"), r[-10:])
  479. finally:
  480. r = s = None
  481. # This test is meaningful even with size < 2G, as long as the
  482. # doubled string is > 2G (but it tests more if both are > 2G :)
  483. @bigmemtest(minsize=_1G + 2, memuse=3)
  484. def test_concat(self, size):
  485. s = '.' * size
  486. self.assertEqual(len(s), size)
  487. s = s + s
  488. self.assertEqual(len(s), size * 2)
  489. self.assertEqual(s.count('.'), size * 2)
  490. # This test is meaningful even with size < 2G, as long as the
  491. # repeated string is > 2G (but it tests more if both are > 2G :)
  492. @bigmemtest(minsize=_1G + 2, memuse=3)
  493. def test_repeat(self, size):
  494. s = '.' * size
  495. self.assertEqual(len(s), size)
  496. s = s * 2
  497. self.assertEqual(len(s), size * 2)
  498. self.assertEqual(s.count('.'), size * 2)
  499. @bigmemtest(minsize=_2G + 20, memuse=2)
  500. def test_slice_and_getitem(self, size):
  501. SUBSTR = '0123456789'
  502. sublen = len(SUBSTR)
  503. s = SUBSTR * (size // sublen)
  504. stepsize = len(s) // 100
  505. stepsize = stepsize - (stepsize % sublen)
  506. for i in range(0, len(s) - stepsize, stepsize):
  507. self.assertEqual(s[i], SUBSTR[0])
  508. self.assertEqual(s[i:i + sublen], SUBSTR)
  509. self.assertEqual(s[i:i + sublen:2], SUBSTR[::2])
  510. if i > 0:
  511. self.assertEqual(s[i + sublen - 1:i - 1:-3],
  512. SUBSTR[sublen::-3])
  513. # Make sure we do some slicing and indexing near the end of the
  514. # string, too.
  515. self.assertEqual(s[len(s) - 1], SUBSTR[-1])
  516. self.assertEqual(s[-1], SUBSTR[-1])
  517. self.assertEqual(s[len(s) - 10], SUBSTR[0])
  518. self.assertEqual(s[-sublen], SUBSTR[0])
  519. self.assertEqual(s[len(s):], '')
  520. self.assertEqual(s[len(s) - 1:], SUBSTR[-1])
  521. self.assertEqual(s[-1:], SUBSTR[-1])
  522. self.assertEqual(s[len(s) - sublen:], SUBSTR)
  523. self.assertEqual(s[-sublen:], SUBSTR)
  524. self.assertEqual(len(s[:]), len(s))
  525. self.assertEqual(len(s[:len(s) - 5]), len(s) - 5)
  526. self.assertEqual(len(s[5:-5]), len(s) - 10)
  527. self.assertRaises(IndexError, operator.getitem, s, len(s))
  528. self.assertRaises(IndexError, operator.getitem, s, len(s) + 1)
  529. self.assertRaises(IndexError, operator.getitem, s, len(s) + 1<<31)
  530. @bigmemtest(minsize=_2G, memuse=2)
  531. def test_contains(self, size):
  532. SUBSTR = '0123456789'
  533. edge = '-' * (size // 2)
  534. s = ''.join([edge, SUBSTR, edge])
  535. del edge
  536. self.assertIn(SUBSTR, s)
  537. self.assertNotIn(SUBSTR * 2, s)
  538. self.assertIn('-', s)
  539. self.assertNotIn('a', s)
  540. s += 'a'
  541. self.assertIn('a', s)
  542. @bigmemtest(minsize=_2G + 10, memuse=2)
  543. def test_compare(self, size):
  544. s1 = '-' * size
  545. s2 = '-' * size
  546. self.assertTrue(s1 == s2)
  547. del s2
  548. s2 = s1 + 'a'
  549. self.assertFalse(s1 == s2)
  550. del s2
  551. s2 = '.' * size
  552. self.assertFalse(s1 == s2)
  553. @bigmemtest(minsize=_2G + 10, memuse=1)
  554. def test_hash(self, size):
  555. # Not sure if we can do any meaningful tests here... Even if we
  556. # start relying on the exact algorithm used, the result will be
  557. # different depending on the size of the C 'long int'. Even this
  558. # test is dodgy (there's no *guarantee* that the two things should
  559. # have a different hash, even if they, in the current
  560. # implementation, almost always do.)
  561. s = '\x00' * size
  562. h1 = hash(s)
  563. del s
  564. s = '\x00' * (size + 1)
  565. self.assertFalse(h1 == hash(s))
  566. class TupleTest(unittest.TestCase):
  567. # Tuples have a small, fixed-sized head and an array of pointers to
  568. # data. Since we're testing 64-bit addressing, we can assume that the
  569. # pointers are 8 bytes, and that thus that the tuples take up 8 bytes
  570. # per size.
  571. # As a side-effect of testing long tuples, these tests happen to test
  572. # having more than 2<<31 references to any given object. Hence the
  573. # use of different types of objects as contents in different tests.
  574. @bigmemtest(minsize=_2G + 2, memuse=16)
  575. def test_compare(self, size):
  576. t1 = (u'',) * size
  577. t2 = (u'',) * size
  578. self.assertTrue(t1 == t2)
  579. del t2
  580. t2 = (u'',) * (size + 1)
  581. self.assertFalse(t1 == t2)
  582. del t2
  583. t2 = (1,) * size
  584. self.assertFalse(t1 == t2)
  585. # Test concatenating into a single tuple of more than 2G in length,
  586. # and concatenating a tuple of more than 2G in length separately, so
  587. # the smaller test still gets run even if there isn't memory for the
  588. # larger test (but we still let the tester know the larger test is
  589. # skipped, in verbose mode.)
  590. def basic_concat_test(self, size):
  591. t = ((),) * size
  592. self.assertEqual(len(t), size)
  593. t = t + t
  594. self.assertEqual(len(t), size * 2)
  595. @bigmemtest(minsize=_2G // 2 + 2, memuse=24)
  596. def test_concat_small(self, size):
  597. return self.basic_concat_test(size)
  598. @bigmemtest(minsize=_2G + 2, memuse=24)
  599. def test_concat_large(self, size):
  600. return self.basic_concat_test(size)
  601. @bigmemtest(minsize=_2G // 5 + 10, memuse=8 * 5)
  602. def test_contains(self, size):
  603. t = (1, 2, 3, 4, 5) * size
  604. self.assertEqual(len(t), size * 5)
  605. self.assertIn(5, t)
  606. self.assertNotIn((1, 2, 3, 4, 5), t)
  607. self.assertNotIn(0, t)
  608. @bigmemtest(minsize=_2G + 10, memuse=8)
  609. def test_hash(self, size):
  610. t1 = (0,) * size
  611. h1 = hash(t1)
  612. del t1
  613. t2 = (0,) * (size + 1)
  614. self.assertFalse(h1 == hash(t2))
  615. @bigmemtest(minsize=_2G + 10, memuse=8)
  616. def test_index_and_slice(self, size):
  617. t = (None,) * size
  618. self.assertEqual(len(t), size)
  619. self.assertEqual(t[-1], None)
  620. self.assertEqual(t[5], None)
  621. self.assertEqual(t[size - 1], None)
  622. self.assertRaises(IndexError, operator.getitem, t, size)
  623. self.assertEqual(t[:5], (None,) * 5)
  624. self.assertEqual(t[-5:], (None,) * 5)
  625. self.assertEqual(t[20:25], (None,) * 5)
  626. self.assertEqual(t[-25:-20], (None,) * 5)
  627. self.assertEqual(t[size - 5:], (None,) * 5)
  628. self.assertEqual(t[size - 5:size], (None,) * 5)
  629. self.assertEqual(t[size - 6:size - 2], (None,) * 4)
  630. self.assertEqual(t[size:size], ())
  631. self.assertEqual(t[size:size+5], ())
  632. # Like test_concat, split in two.
  633. def basic_test_repeat(self, size):
  634. t = ('',) * size
  635. self.assertEqual(len(t), size)
  636. t = t * 2
  637. self.assertEqual(len(t), size * 2)
  638. @bigmemtest(minsize=_2G // 2 + 2, memuse=24)
  639. def test_repeat_small(self, size):
  640. return self.basic_test_repeat(size)
  641. @bigmemtest(minsize=_2G + 2, memuse=24)
  642. def test_repeat_large(self, size):
  643. return self.basic_test_repeat(size)
  644. @bigmemtest(minsize=_1G - 1, memuse=12)
  645. def test_repeat_large_2(self, size):
  646. return self.basic_test_repeat(size)
  647. @precisionbigmemtest(size=_1G - 1, memuse=9)
  648. def test_from_2G_generator(self, size):
  649. try:
  650. t = tuple(xrange(size))
  651. except MemoryError:
  652. pass # acceptable on 32-bit
  653. else:
  654. count = 0
  655. for item in t:
  656. self.assertEqual(item, count)
  657. count += 1
  658. self.assertEqual(count, size)
  659. @precisionbigmemtest(size=_1G - 25, memuse=9)
  660. def test_from_almost_2G_generator(self, size):
  661. try:
  662. t = tuple(xrange(size))
  663. count = 0
  664. for item in t:
  665. self.assertEqual(item, count)
  666. count += 1
  667. self.assertEqual(count, size)
  668. except MemoryError:
  669. pass # acceptable, expected on 32-bit
  670. # Like test_concat, split in two.
  671. def basic_test_repr(self, size):
  672. t = (0,) * size
  673. s = repr(t)
  674. # The repr of a tuple of 0's is exactly three times the tuple length.
  675. self.assertEqual(len(s), size * 3)
  676. self.assertEqual(s[:5], '(0, 0')
  677. self.assertEqual(s[-5:], '0, 0)')
  678. self.assertEqual(s.count('0'), size)
  679. @bigmemtest(minsize=_2G // 3 + 2, memuse=8 + 3)
  680. def test_repr_small(self, size):
  681. return self.basic_test_repr(size)
  682. @bigmemtest(minsize=_2G + 2, memuse=8 + 3)
  683. def test_repr_large(self, size):
  684. return self.basic_test_repr(size)
  685. class ListTest(unittest.TestCase):
  686. # Like tuples, lists have a small, fixed-sized head and an array of
  687. # pointers to data, so 8 bytes per size. Also like tuples, we make the
  688. # lists hold references to various objects to test their refcount
  689. # limits.
  690. @bigmemtest(minsize=_2G + 2, memuse=16)
  691. def test_compare(self, size):
  692. l1 = [u''] * size
  693. l2 = [u''] * size
  694. self.assertTrue(l1 == l2)
  695. del l2
  696. l2 = [u''] * (size + 1)
  697. self.assertFalse(l1 == l2)
  698. del l2
  699. l2 = [2] * size
  700. self.assertFalse(l1 == l2)
  701. # Test concatenating into a single list of more than 2G in length,
  702. # and concatenating a list of more than 2G in length separately, so
  703. # the smaller test still gets run even if there isn't memory for the
  704. # larger test (but we still let the tester know the larger test is
  705. # skipped, in verbose mode.)
  706. def basic_test_concat(self, size):
  707. l = [[]] * size
  708. self.assertEqual(len(l), size)
  709. l = l + l
  710. self.assertEqual(len(l), size * 2)
  711. @bigmemtest(minsize=_2G // 2 + 2, memuse=24)
  712. def test_concat_small(self, size):
  713. return self.basic_test_concat(size)
  714. @bigmemtest(minsize=_2G + 2, memuse=24)
  715. def test_concat_large(self, size):
  716. return self.basic_test_concat(size)
  717. def basic_test_inplace_concat(self, size):
  718. l = [sys.stdout] * size
  719. l += l
  720. self.assertEqual(len(l), size * 2)
  721. self.assertTrue(l[0] is l[-1])
  722. self.assertTrue(l[size - 1] is l[size + 1])
  723. @bigmemtest(minsize=_2G // 2 + 2, memuse=24)
  724. def test_inplace_concat_small(self, size):
  725. return self.basic_test_inplace_concat(size)
  726. @bigmemtest(minsize=_2G + 2, memuse=24)
  727. def test_inplace_concat_large(self, size):
  728. return self.basic_test_inplace_concat(size)
  729. @bigmemtest(minsize=_2G // 5 + 10, memuse=8 * 5)
  730. def test_contains(self, size):
  731. l = [1, 2, 3, 4, 5] * size
  732. self.assertEqual(len(l), size * 5)
  733. self.assertIn(5, l)
  734. self.assertNotIn([1, 2, 3, 4, 5], l)
  735. self.assertNotIn(0, l)
  736. @bigmemtest(minsize=_2G + 10, memuse=8)
  737. def test_hash(self, size):
  738. l = [0] * size
  739. self.assertRaises(TypeError, hash, l)
  740. @bigmemtest(minsize=_2G + 10, memuse=8)
  741. def test_index_and_slice(self, size):
  742. l = [None] * size
  743. self.assertEqual(len(l), size)
  744. self.assertEqual(l[-1], None)
  745. self.assertEqual(l[5], None)
  746. self.assertEqual(l[size - 1], None)
  747. self.assertRaises(IndexError, operator.getitem, l, size)
  748. self.assertEqual(l[:5], [None] * 5)
  749. self.assertEqual(l[-5:], [None] * 5)
  750. self.assertEqual(l[20:25], [None] * 5)
  751. self.assertEqual(l[-25:-20], [None] * 5)
  752. self.assertEqual(l[size - 5:], [None] * 5)
  753. self.assertEqual(l[size - 5:size], [None] * 5)
  754. self.assertEqual(l[size - 6:size - 2], [None] * 4)
  755. self.assertEqual(l[size:size], [])
  756. self.assertEqual(l[size:size+5], [])
  757. l[size - 2] = 5
  758. self.assertEqual(len(l), size)
  759. self.assertEqual(l[-3:], [None, 5, None])
  760. self.assertEqual(l.count(5), 1)
  761. self.assertRaises(IndexError, operator.setitem, l, size, 6)
  762. self.assertEqual(len(l), size)
  763. l[size - 7:] = [1, 2, 3, 4, 5]
  764. size -= 2
  765. self.assertEqual(len(l), size)
  766. self.assertEqual(l[-7:], [None, None, 1, 2, 3, 4, 5])
  767. l[:7] = [1, 2, 3, 4, 5]
  768. size -= 2
  769. self.assertEqual(len(l), size)
  770. self.assertEqual(l[:7], [1, 2, 3, 4, 5, None, None])
  771. del l[size - 1]
  772. size -= 1
  773. self.assertEqual(len(l), size)
  774. self.assertEqual(l[-1], 4)
  775. del l[-2:]
  776. size -= 2
  777. self.assertEqual(len(l), size)
  778. self.assertEqual(l[-1], 2)
  779. del l[0]
  780. size -= 1
  781. self.assertEqual(len(l), size)
  782. self.assertEqual(l[0], 2)
  783. del l[:2]
  784. size -= 2
  785. self.assertEqual(len(l), size)
  786. self.assertEqual(l[0], 4)
  787. # Like test_concat, split in two.
  788. def basic_test_repeat(self, size):
  789. l = [] * size
  790. self.assertFalse(l)
  791. l = [''] * size
  792. self.assertEqual(len(l), size)
  793. l = l * 2
  794. self.assertEqual(len(l), size * 2)
  795. @bigmemtest(minsize=_2G // 2 + 2, memuse=24)
  796. def test_repeat_small(self, size):
  797. return self.basic_test_repeat(size)
  798. @bigmemtest(minsize=_2G + 2, memuse=24)
  799. def test_repeat_large(self, size):
  800. return self.basic_test_repeat(size)
  801. def basic_test_inplace_repeat(self, size):
  802. l = ['']
  803. l *= size
  804. self.assertEqual(len(l), size)
  805. self.assertTrue(l[0] is l[-1])
  806. del l
  807. l = [''] * size
  808. l *= 2
  809. self.assertEqual(len(l), size * 2)
  810. self.assertTrue(l[size - 1] is l[-1])
  811. @bigmemtest(minsize=_2G // 2 + 2, memuse=16)
  812. def test_inplace_repeat_small(self, size):
  813. return self.basic_test_inplace_repeat(size)
  814. @bigmemtest(minsize=_2G + 2, memuse=16)
  815. def test_inplace_repeat_large(self, size):
  816. return self.basic_test_inplace_repeat(size)
  817. def basic_test_repr(self, size):
  818. l = [0] * size
  819. s = repr(l)
  820. # The repr of a list of 0's is exactly three times the list length.
  821. self.assertEqual(len(s), size * 3)
  822. self.assertEqual(s[:5], '[0, 0')
  823. self.assertEqual(s[-5:], '0, 0]')
  824. self.assertEqual(s.count('0'), size)
  825. @bigmemtest(minsize=_2G // 3 + 2, memuse=8 + 3)
  826. def test_repr_small(self, size):
  827. return self.basic_test_repr(size)
  828. @bigmemtest(minsize=_2G + 2, memuse=8 + 3)
  829. def test_repr_large(self, size):
  830. return self.basic_test_repr(size)
  831. # list overallocates ~1/8th of the total size (on first expansion) so
  832. # the single list.append call puts memuse at 9 bytes per size.
  833. @bigmemtest(minsize=_2G, memuse=9)
  834. def test_append(self, size):
  835. l = [object()] * size
  836. l.append(object())
  837. self.assertEqual(len(l), size+1)
  838. self.assertTrue(l[-3] is l[-2])
  839. self.assertFalse(l[-2] is l[-1])
  840. @bigmemtest(minsize=_2G // 5 + 2, memuse=8 * 5)
  841. def test_count(self, size):
  842. l = [1, 2, 3, 4, 5] * size
  843. self.assertEqual(l.count(1), size)
  844. self.assertEqual(l.count("1"), 0)
  845. def basic_test_extend(self, size):
  846. l = [file] * size
  847. l.extend(l)
  848. self.assertEqual(len(l), size * 2)
  849. self.assertTrue(l[0] is l[-1])
  850. self.assertTrue(l[size - 1] is l[size + 1])
  851. @bigmemtest(minsize=_2G // 2 + 2, memuse=16)
  852. def test_extend_small(self, size):
  853. return self.basic_test_extend(size)
  854. @bigmemtest(minsize=_2G + 2, memuse=16)
  855. def test_extend_large(self, size):
  856. return self.basic_test_extend(size)
  857. @bigmemtest(minsize=_2G // 5 + 2, memuse=8 * 5)
  858. def test_index(self, size):
  859. l = [1L, 2L, 3L, 4L, 5L] * size
  860. size *= 5
  861. self.assertEqual(l.index(1), 0)
  862. self.assertEqual(l.index(5, size - 5), size - 1)
  863. self.assertEqual(l.index(5, size - 5, size), size - 1)
  864. self.assertRaises(ValueError, l.index, 1, size - 4, size)
  865. self.assertRaises(ValueError, l.index, 6L)
  866. # This tests suffers from overallocation, just like test_append.
  867. @bigmemtest(minsize=_2G + 10, memuse=9)
  868. def test_insert(self, size):
  869. l = [1.0] * size
  870. l.insert(size - 1, "A")
  871. size += 1
  872. self.assertEqual(len(l), size)
  873. self.assertEqual(l[-3:], [1.0, "A", 1.0])
  874. l.insert(size + 1, "B")
  875. size += 1
  876. self.assertEqual(len(l), size)
  877. self.assertEqual(l[-3:], ["A", 1.0, "B"])
  878. l.insert(1, "C")
  879. size += 1
  880. self.assertEqual(len(l), size)
  881. self.assertEqual(l[:3], [1.0, "C", 1.0])
  882. self.assertEqual(l[size - 3:], ["A", 1.0, "B"])
  883. @bigmemtest(minsize=_2G // 5 + 4, memuse=8 * 5)
  884. def test_pop(self, size):
  885. l = [u"a", u"b", u"c", u"d", u"e"] * size
  886. size *= 5
  887. self.assertEqual(len(l), size)
  888. item = l.pop()
  889. size -= 1
  890. self.assertEqual(len(l), size)
  891. self.assertEqual(item, u"e")
  892. self.assertEqual(l[-2:], [u"c", u"d"])
  893. item = l.pop(0)
  894. size -= 1
  895. self.assertEqual(len(l), size)
  896. self.assertEqual(item, u"a")
  897. self.assertEqual(l[:2], [u"b", u"c"])
  898. item = l.pop(size - 2)
  899. size -= 1
  900. self.assertEqual(len(l), size)
  901. self.assertEqual(item, u"c")
  902. self.assertEqual(l[-2:], [u"b", u"d"])
  903. @bigmemtest(minsize=_2G + 10, memuse=8)
  904. def test_remove(self, size):
  905. l = [10] * size
  906. self.assertEqual(len(l), size)
  907. l.remove(10)
  908. size -= 1
  909. self.assertEqual(len(l), size)
  910. # Because of the earlier l.remove(), this append doesn't trigger
  911. # a resize.
  912. l.append(5)
  913. size += 1
  914. self.assertEqual(len(l), size)
  915. self.assertEqual(l[-2:], [10, 5])
  916. l.remove(5)
  917. size -= 1
  918. self.assertEqual(len(l), size)
  919. self.assertEqual(l[-2:], [10, 10])
  920. @bigmemtest(minsize=_2G // 5 + 2, memuse=8 * 5)
  921. def test_reverse(self, size):
  922. l = [1, 2, 3, 4, 5] * size
  923. l.reverse()
  924. self.assertEqual(len(l), size * 5)
  925. self.assertEqual(l[-5:], [5, 4, 3, 2, 1])
  926. self.assertEqual(l[:5], [5, 4, 3, 2, 1])
  927. @bigmemtest(minsize=_2G // 5 + 2, memuse=8 * 5)
  928. def test_sort(self, size):
  929. l = [1, 2, 3, 4, 5] * size
  930. l.sort()
  931. self.assertEqual(len(l), size * 5)
  932. self.assertEqual(l.count(1), size)
  933. self.assertEqual(l[:10], [1] * 10)
  934. self.assertEqual(l[-10:], [5] * 10)
  935. class BufferTest(unittest.TestCase):
  936. @precisionbigmemtest(size=_1G, memuse=4)
  937. def test_repeat(self, size):
  938. try:
  939. with test_support.check_py3k_warnings():
  940. b = buffer("AAAA")*size
  941. except MemoryError:
  942. pass # acceptable on 32-bit
  943. else:
  944. count = 0
  945. for c in b:
  946. self.assertEqual(c, 'A')
  947. count += 1
  948. self.assertEqual(count, size*4)
  949. def test_main():
  950. test_support.run_unittest(StrTest, TupleTest, ListTest, BufferTest)
  951. if __name__ == '__main__':
  952. if len(sys.argv) > 1:
  953. test_support.set_memlimit(sys.argv[1])
  954. test_main()