test_deque.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809
  1. from collections import deque
  2. import unittest
  3. from test import test_support, seq_tests
  4. import gc
  5. import weakref
  6. import copy
  7. import cPickle as pickle
  8. import random
  9. import struct
  10. BIG = 100000
  11. def fail():
  12. raise SyntaxError
  13. yield 1
  14. class BadCmp:
  15. def __eq__(self, other):
  16. raise RuntimeError
  17. class MutateCmp:
  18. def __init__(self, deque, result):
  19. self.deque = deque
  20. self.result = result
  21. def __eq__(self, other):
  22. self.deque.clear()
  23. return self.result
  24. class TestBasic(unittest.TestCase):
  25. def test_basics(self):
  26. d = deque(xrange(-5125, -5000))
  27. d.__init__(xrange(200))
  28. for i in xrange(200, 400):
  29. d.append(i)
  30. for i in reversed(xrange(-200, 0)):
  31. d.appendleft(i)
  32. self.assertEqual(list(d), range(-200, 400))
  33. self.assertEqual(len(d), 600)
  34. left = [d.popleft() for i in xrange(250)]
  35. self.assertEqual(left, range(-200, 50))
  36. self.assertEqual(list(d), range(50, 400))
  37. right = [d.pop() for i in xrange(250)]
  38. right.reverse()
  39. self.assertEqual(right, range(150, 400))
  40. self.assertEqual(list(d), range(50, 150))
  41. def test_maxlen(self):
  42. self.assertRaises(ValueError, deque, 'abc', -1)
  43. self.assertRaises(ValueError, deque, 'abc', -2)
  44. it = iter(range(10))
  45. d = deque(it, maxlen=3)
  46. self.assertEqual(list(it), [])
  47. self.assertEqual(repr(d), 'deque([7, 8, 9], maxlen=3)')
  48. self.assertEqual(list(d), range(7, 10))
  49. self.assertEqual(d, deque(range(10), 3))
  50. d.append(10)
  51. self.assertEqual(list(d), range(8, 11))
  52. d.appendleft(7)
  53. self.assertEqual(list(d), range(7, 10))
  54. d.extend([10, 11])
  55. self.assertEqual(list(d), range(9, 12))
  56. d.extendleft([8, 7])
  57. self.assertEqual(list(d), range(7, 10))
  58. d = deque(xrange(200), maxlen=10)
  59. d.append(d)
  60. test_support.unlink(test_support.TESTFN)
  61. fo = open(test_support.TESTFN, "wb")
  62. try:
  63. print >> fo, d,
  64. fo.close()
  65. fo = open(test_support.TESTFN, "rb")
  66. self.assertEqual(fo.read(), repr(d))
  67. finally:
  68. fo.close()
  69. test_support.unlink(test_support.TESTFN)
  70. d = deque(range(10), maxlen=None)
  71. self.assertEqual(repr(d), 'deque([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])')
  72. fo = open(test_support.TESTFN, "wb")
  73. try:
  74. print >> fo, d,
  75. fo.close()
  76. fo = open(test_support.TESTFN, "rb")
  77. self.assertEqual(fo.read(), repr(d))
  78. finally:
  79. fo.close()
  80. test_support.unlink(test_support.TESTFN)
  81. def test_maxlen_zero(self):
  82. it = iter(range(100))
  83. deque(it, maxlen=0)
  84. self.assertEqual(list(it), [])
  85. it = iter(range(100))
  86. d = deque(maxlen=0)
  87. d.extend(it)
  88. self.assertEqual(list(it), [])
  89. it = iter(range(100))
  90. d = deque(maxlen=0)
  91. d.extendleft(it)
  92. self.assertEqual(list(it), [])
  93. def test_maxlen_attribute(self):
  94. self.assertEqual(deque().maxlen, None)
  95. self.assertEqual(deque('abc').maxlen, None)
  96. self.assertEqual(deque('abc', maxlen=4).maxlen, 4)
  97. self.assertEqual(deque('abc', maxlen=2).maxlen, 2)
  98. self.assertEqual(deque('abc', maxlen=0).maxlen, 0)
  99. with self.assertRaises(AttributeError):
  100. d = deque('abc')
  101. d.maxlen = 10
  102. def test_count(self):
  103. for s in ('', 'abracadabra', 'simsalabim'*500+'abc'):
  104. s = list(s)
  105. d = deque(s)
  106. for letter in 'abcdefghijklmnopqrstuvwxyz':
  107. self.assertEqual(s.count(letter), d.count(letter), (s, d, letter))
  108. self.assertRaises(TypeError, d.count) # too few args
  109. self.assertRaises(TypeError, d.count, 1, 2) # too many args
  110. class BadCompare:
  111. def __eq__(self, other):
  112. raise ArithmeticError
  113. d = deque([1, 2, BadCompare(), 3])
  114. self.assertRaises(ArithmeticError, d.count, 2)
  115. d = deque([1, 2, 3])
  116. self.assertRaises(ArithmeticError, d.count, BadCompare())
  117. class MutatingCompare:
  118. def __eq__(self, other):
  119. self.d.pop()
  120. return True
  121. m = MutatingCompare()
  122. d = deque([1, 2, 3, m, 4, 5])
  123. m.d = d
  124. self.assertRaises(RuntimeError, d.count, 3)
  125. # test issue11004
  126. # block advance failed after rotation aligned elements on right side of block
  127. d = deque([None]*16)
  128. for i in range(len(d)):
  129. d.rotate(-1)
  130. d.rotate(1)
  131. self.assertEqual(d.count(1), 0)
  132. self.assertEqual(d.count(None), 16)
  133. def test_comparisons(self):
  134. d = deque('xabc'); d.popleft()
  135. for e in [d, deque('abc'), deque('ab'), deque(), list(d)]:
  136. self.assertEqual(d==e, type(d)==type(e) and list(d)==list(e))
  137. self.assertEqual(d!=e, not(type(d)==type(e) and list(d)==list(e)))
  138. args = map(deque, ('', 'a', 'b', 'ab', 'ba', 'abc', 'xba', 'xabc', 'cba'))
  139. for x in args:
  140. for y in args:
  141. self.assertEqual(x == y, list(x) == list(y), (x,y))
  142. self.assertEqual(x != y, list(x) != list(y), (x,y))
  143. self.assertEqual(x < y, list(x) < list(y), (x,y))
  144. self.assertEqual(x <= y, list(x) <= list(y), (x,y))
  145. self.assertEqual(x > y, list(x) > list(y), (x,y))
  146. self.assertEqual(x >= y, list(x) >= list(y), (x,y))
  147. self.assertEqual(cmp(x,y), cmp(list(x),list(y)), (x,y))
  148. def test_extend(self):
  149. d = deque('a')
  150. self.assertRaises(TypeError, d.extend, 1)
  151. d.extend('bcd')
  152. self.assertEqual(list(d), list('abcd'))
  153. d.extend(d)
  154. self.assertEqual(list(d), list('abcdabcd'))
  155. def test_iadd(self):
  156. d = deque('a')
  157. d += 'bcd'
  158. self.assertEqual(list(d), list('abcd'))
  159. d += d
  160. self.assertEqual(list(d), list('abcdabcd'))
  161. def test_extendleft(self):
  162. d = deque('a')
  163. self.assertRaises(TypeError, d.extendleft, 1)
  164. d.extendleft('bcd')
  165. self.assertEqual(list(d), list(reversed('abcd')))
  166. d.extendleft(d)
  167. self.assertEqual(list(d), list('abcddcba'))
  168. d = deque()
  169. d.extendleft(range(1000))
  170. self.assertEqual(list(d), list(reversed(range(1000))))
  171. self.assertRaises(SyntaxError, d.extendleft, fail())
  172. def test_getitem(self):
  173. n = 200
  174. d = deque(xrange(n))
  175. l = range(n)
  176. for i in xrange(n):
  177. d.popleft()
  178. l.pop(0)
  179. if random.random() < 0.5:
  180. d.append(i)
  181. l.append(i)
  182. for j in xrange(1-len(l), len(l)):
  183. assert d[j] == l[j]
  184. d = deque('superman')
  185. self.assertEqual(d[0], 's')
  186. self.assertEqual(d[-1], 'n')
  187. d = deque()
  188. self.assertRaises(IndexError, d.__getitem__, 0)
  189. self.assertRaises(IndexError, d.__getitem__, -1)
  190. def test_setitem(self):
  191. n = 200
  192. d = deque(xrange(n))
  193. for i in xrange(n):
  194. d[i] = 10 * i
  195. self.assertEqual(list(d), [10*i for i in xrange(n)])
  196. l = list(d)
  197. for i in xrange(1-n, 0, -1):
  198. d[i] = 7*i
  199. l[i] = 7*i
  200. self.assertEqual(list(d), l)
  201. def test_delitem(self):
  202. n = 500 # O(n**2) test, don't make this too big
  203. d = deque(xrange(n))
  204. self.assertRaises(IndexError, d.__delitem__, -n-1)
  205. self.assertRaises(IndexError, d.__delitem__, n)
  206. for i in xrange(n):
  207. self.assertEqual(len(d), n-i)
  208. j = random.randrange(-len(d), len(d))
  209. val = d[j]
  210. self.assertIn(val, d)
  211. del d[j]
  212. self.assertNotIn(val, d)
  213. self.assertEqual(len(d), 0)
  214. def test_reverse(self):
  215. n = 500 # O(n**2) test, don't make this too big
  216. data = [random.random() for i in range(n)]
  217. for i in range(n):
  218. d = deque(data[:i])
  219. r = d.reverse()
  220. self.assertEqual(list(d), list(reversed(data[:i])))
  221. self.assertIs(r, None)
  222. d.reverse()
  223. self.assertEqual(list(d), data[:i])
  224. self.assertRaises(TypeError, d.reverse, 1) # Arity is zero
  225. def test_rotate(self):
  226. s = tuple('abcde')
  227. n = len(s)
  228. d = deque(s)
  229. d.rotate(1) # verify rot(1)
  230. self.assertEqual(''.join(d), 'eabcd')
  231. d = deque(s)
  232. d.rotate(-1) # verify rot(-1)
  233. self.assertEqual(''.join(d), 'bcdea')
  234. d.rotate() # check default to 1
  235. self.assertEqual(tuple(d), s)
  236. for i in xrange(n*3):
  237. d = deque(s)
  238. e = deque(d)
  239. d.rotate(i) # check vs. rot(1) n times
  240. for j in xrange(i):
  241. e.rotate(1)
  242. self.assertEqual(tuple(d), tuple(e))
  243. d.rotate(-i) # check that it works in reverse
  244. self.assertEqual(tuple(d), s)
  245. e.rotate(n-i) # check that it wraps forward
  246. self.assertEqual(tuple(e), s)
  247. for i in xrange(n*3):
  248. d = deque(s)
  249. e = deque(d)
  250. d.rotate(-i)
  251. for j in xrange(i):
  252. e.rotate(-1) # check vs. rot(-1) n times
  253. self.assertEqual(tuple(d), tuple(e))
  254. d.rotate(i) # check that it works in reverse
  255. self.assertEqual(tuple(d), s)
  256. e.rotate(i-n) # check that it wraps backaround
  257. self.assertEqual(tuple(e), s)
  258. d = deque(s)
  259. e = deque(s)
  260. e.rotate(BIG+17) # verify on long series of rotates
  261. dr = d.rotate
  262. for i in xrange(BIG+17):
  263. dr()
  264. self.assertEqual(tuple(d), tuple(e))
  265. self.assertRaises(TypeError, d.rotate, 'x') # Wrong arg type
  266. self.assertRaises(TypeError, d.rotate, 1, 10) # Too many args
  267. d = deque()
  268. d.rotate() # rotate an empty deque
  269. self.assertEqual(d, deque())
  270. def test_len(self):
  271. d = deque('ab')
  272. self.assertEqual(len(d), 2)
  273. d.popleft()
  274. self.assertEqual(len(d), 1)
  275. d.pop()
  276. self.assertEqual(len(d), 0)
  277. self.assertRaises(IndexError, d.pop)
  278. self.assertEqual(len(d), 0)
  279. d.append('c')
  280. self.assertEqual(len(d), 1)
  281. d.appendleft('d')
  282. self.assertEqual(len(d), 2)
  283. d.clear()
  284. self.assertEqual(len(d), 0)
  285. def test_underflow(self):
  286. d = deque()
  287. self.assertRaises(IndexError, d.pop)
  288. self.assertRaises(IndexError, d.popleft)
  289. def test_clear(self):
  290. d = deque(xrange(100))
  291. self.assertEqual(len(d), 100)
  292. d.clear()
  293. self.assertEqual(len(d), 0)
  294. self.assertEqual(list(d), [])
  295. d.clear() # clear an emtpy deque
  296. self.assertEqual(list(d), [])
  297. def test_remove(self):
  298. d = deque('abcdefghcij')
  299. d.remove('c')
  300. self.assertEqual(d, deque('abdefghcij'))
  301. d.remove('c')
  302. self.assertEqual(d, deque('abdefghij'))
  303. self.assertRaises(ValueError, d.remove, 'c')
  304. self.assertEqual(d, deque('abdefghij'))
  305. # Handle comparison errors
  306. d = deque(['a', 'b', BadCmp(), 'c'])
  307. e = deque(d)
  308. self.assertRaises(RuntimeError, d.remove, 'c')
  309. for x, y in zip(d, e):
  310. # verify that original order and values are retained.
  311. self.assertTrue(x is y)
  312. # Handle evil mutator
  313. for match in (True, False):
  314. d = deque(['ab'])
  315. d.extend([MutateCmp(d, match), 'c'])
  316. self.assertRaises(IndexError, d.remove, 'c')
  317. self.assertEqual(d, deque())
  318. def test_repr(self):
  319. d = deque(xrange(200))
  320. e = eval(repr(d))
  321. self.assertEqual(list(d), list(e))
  322. d.append(d)
  323. self.assertIn('...', repr(d))
  324. def test_print(self):
  325. d = deque(xrange(200))
  326. d.append(d)
  327. test_support.unlink(test_support.TESTFN)
  328. fo = open(test_support.TESTFN, "wb")
  329. try:
  330. print >> fo, d,
  331. fo.close()
  332. fo = open(test_support.TESTFN, "rb")
  333. self.assertEqual(fo.read(), repr(d))
  334. finally:
  335. fo.close()
  336. test_support.unlink(test_support.TESTFN)
  337. def test_init(self):
  338. self.assertRaises(TypeError, deque, 'abc', 2, 3);
  339. self.assertRaises(TypeError, deque, 1);
  340. def test_hash(self):
  341. self.assertRaises(TypeError, hash, deque('abc'))
  342. def test_long_steadystate_queue_popleft(self):
  343. for size in (0, 1, 2, 100, 1000):
  344. d = deque(xrange(size))
  345. append, pop = d.append, d.popleft
  346. for i in xrange(size, BIG):
  347. append(i)
  348. x = pop()
  349. if x != i - size:
  350. self.assertEqual(x, i-size)
  351. self.assertEqual(list(d), range(BIG-size, BIG))
  352. def test_long_steadystate_queue_popright(self):
  353. for size in (0, 1, 2, 100, 1000):
  354. d = deque(reversed(xrange(size)))
  355. append, pop = d.appendleft, d.pop
  356. for i in xrange(size, BIG):
  357. append(i)
  358. x = pop()
  359. if x != i - size:
  360. self.assertEqual(x, i-size)
  361. self.assertEqual(list(reversed(list(d))), range(BIG-size, BIG))
  362. def test_big_queue_popleft(self):
  363. pass
  364. d = deque()
  365. append, pop = d.append, d.popleft
  366. for i in xrange(BIG):
  367. append(i)
  368. for i in xrange(BIG):
  369. x = pop()
  370. if x != i:
  371. self.assertEqual(x, i)
  372. def test_big_queue_popright(self):
  373. d = deque()
  374. append, pop = d.appendleft, d.pop
  375. for i in xrange(BIG):
  376. append(i)
  377. for i in xrange(BIG):
  378. x = pop()
  379. if x != i:
  380. self.assertEqual(x, i)
  381. def test_big_stack_right(self):
  382. d = deque()
  383. append, pop = d.append, d.pop
  384. for i in xrange(BIG):
  385. append(i)
  386. for i in reversed(xrange(BIG)):
  387. x = pop()
  388. if x != i:
  389. self.assertEqual(x, i)
  390. self.assertEqual(len(d), 0)
  391. def test_big_stack_left(self):
  392. d = deque()
  393. append, pop = d.appendleft, d.popleft
  394. for i in xrange(BIG):
  395. append(i)
  396. for i in reversed(xrange(BIG)):
  397. x = pop()
  398. if x != i:
  399. self.assertEqual(x, i)
  400. self.assertEqual(len(d), 0)
  401. def test_roundtrip_iter_init(self):
  402. d = deque(xrange(200))
  403. e = deque(d)
  404. self.assertNotEqual(id(d), id(e))
  405. self.assertEqual(list(d), list(e))
  406. def test_pickle(self):
  407. d = deque(xrange(200))
  408. for i in range(pickle.HIGHEST_PROTOCOL + 1):
  409. s = pickle.dumps(d, i)
  410. e = pickle.loads(s)
  411. self.assertNotEqual(id(d), id(e))
  412. self.assertEqual(list(d), list(e))
  413. ## def test_pickle_recursive(self):
  414. ## d = deque('abc')
  415. ## d.append(d)
  416. ## for i in range(pickle.HIGHEST_PROTOCOL + 1):
  417. ## e = pickle.loads(pickle.dumps(d, i))
  418. ## self.assertNotEqual(id(d), id(e))
  419. ## self.assertEqual(id(e), id(e[-1]))
  420. def test_deepcopy(self):
  421. mut = [10]
  422. d = deque([mut])
  423. e = copy.deepcopy(d)
  424. self.assertEqual(list(d), list(e))
  425. mut[0] = 11
  426. self.assertNotEqual(id(d), id(e))
  427. self.assertNotEqual(list(d), list(e))
  428. def test_copy(self):
  429. mut = [10]
  430. d = deque([mut])
  431. e = copy.copy(d)
  432. self.assertEqual(list(d), list(e))
  433. mut[0] = 11
  434. self.assertNotEqual(id(d), id(e))
  435. self.assertEqual(list(d), list(e))
  436. def test_reversed(self):
  437. for s in ('abcd', xrange(2000)):
  438. self.assertEqual(list(reversed(deque(s))), list(reversed(s)))
  439. def test_gc_doesnt_blowup(self):
  440. import gc
  441. # This used to assert-fail in deque_traverse() under a debug
  442. # build, or run wild with a NULL pointer in a release build.
  443. d = deque()
  444. for i in xrange(100):
  445. d.append(1)
  446. gc.collect()
  447. def test_container_iterator(self):
  448. # Bug #3680: tp_traverse was not implemented for deque iterator objects
  449. class C(object):
  450. pass
  451. for i in range(2):
  452. obj = C()
  453. ref = weakref.ref(obj)
  454. if i == 0:
  455. container = deque([obj, 1])
  456. else:
  457. container = reversed(deque([obj, 1]))
  458. obj.x = iter(container)
  459. del obj, container
  460. gc.collect()
  461. self.assertTrue(ref() is None, "Cycle was not collected")
  462. check_sizeof = test_support.check_sizeof
  463. @test_support.cpython_only
  464. def test_sizeof(self):
  465. BLOCKLEN = 62
  466. basesize = test_support.calcobjsize('2P3PlPP')
  467. blocksize = struct.calcsize('%dP2P' % BLOCKLEN)
  468. self.assertEqual(object.__sizeof__(deque()), basesize)
  469. check = self.check_sizeof
  470. check(deque(), basesize + blocksize)
  471. check(deque('a'), basesize + blocksize)
  472. check(deque('a' * (BLOCKLEN // 2)), basesize + blocksize)
  473. check(deque('a' * (BLOCKLEN // 2 + 1)), basesize + 2 * blocksize)
  474. check(deque('a' * (42 * BLOCKLEN)), basesize + 43 * blocksize)
  475. class TestVariousIteratorArgs(unittest.TestCase):
  476. def test_constructor(self):
  477. for s in ("123", "", range(1000), ('do', 1.2), xrange(2000,2200,5)):
  478. for g in (seq_tests.Sequence, seq_tests.IterFunc,
  479. seq_tests.IterGen, seq_tests.IterFuncStop,
  480. seq_tests.itermulti, seq_tests.iterfunc):
  481. self.assertEqual(list(deque(g(s))), list(g(s)))
  482. self.assertRaises(TypeError, deque, seq_tests.IterNextOnly(s))
  483. self.assertRaises(TypeError, deque, seq_tests.IterNoNext(s))
  484. self.assertRaises(ZeroDivisionError, deque, seq_tests.IterGenExc(s))
  485. def test_iter_with_altered_data(self):
  486. d = deque('abcdefg')
  487. it = iter(d)
  488. d.pop()
  489. self.assertRaises(RuntimeError, it.next)
  490. def test_runtime_error_on_empty_deque(self):
  491. d = deque()
  492. it = iter(d)
  493. d.append(10)
  494. self.assertRaises(RuntimeError, it.next)
  495. class Deque(deque):
  496. pass
  497. class DequeWithBadIter(deque):
  498. def __iter__(self):
  499. raise TypeError
  500. class TestSubclass(unittest.TestCase):
  501. def test_basics(self):
  502. d = Deque(xrange(25))
  503. d.__init__(xrange(200))
  504. for i in xrange(200, 400):
  505. d.append(i)
  506. for i in reversed(xrange(-200, 0)):
  507. d.appendleft(i)
  508. self.assertEqual(list(d), range(-200, 400))
  509. self.assertEqual(len(d), 600)
  510. left = [d.popleft() for i in xrange(250)]
  511. self.assertEqual(left, range(-200, 50))
  512. self.assertEqual(list(d), range(50, 400))
  513. right = [d.pop() for i in xrange(250)]
  514. right.reverse()
  515. self.assertEqual(right, range(150, 400))
  516. self.assertEqual(list(d), range(50, 150))
  517. d.clear()
  518. self.assertEqual(len(d), 0)
  519. def test_copy_pickle(self):
  520. d = Deque('abc')
  521. e = d.__copy__()
  522. self.assertEqual(type(d), type(e))
  523. self.assertEqual(list(d), list(e))
  524. e = Deque(d)
  525. self.assertEqual(type(d), type(e))
  526. self.assertEqual(list(d), list(e))
  527. for proto in range(pickle.HIGHEST_PROTOCOL + 1):
  528. s = pickle.dumps(d, proto)
  529. e = pickle.loads(s)
  530. self.assertNotEqual(id(d), id(e))
  531. self.assertEqual(type(d), type(e))
  532. self.assertEqual(list(d), list(e))
  533. d = Deque('abcde', maxlen=4)
  534. e = d.__copy__()
  535. self.assertEqual(type(d), type(e))
  536. self.assertEqual(list(d), list(e))
  537. e = Deque(d)
  538. self.assertEqual(type(d), type(e))
  539. self.assertEqual(list(d), list(e))
  540. for proto in range(pickle.HIGHEST_PROTOCOL + 1):
  541. s = pickle.dumps(d, proto)
  542. e = pickle.loads(s)
  543. self.assertNotEqual(id(d), id(e))
  544. self.assertEqual(type(d), type(e))
  545. self.assertEqual(list(d), list(e))
  546. ## def test_pickle(self):
  547. ## d = Deque('abc')
  548. ## d.append(d)
  549. ##
  550. ## e = pickle.loads(pickle.dumps(d))
  551. ## self.assertNotEqual(id(d), id(e))
  552. ## self.assertEqual(type(d), type(e))
  553. ## dd = d.pop()
  554. ## ee = e.pop()
  555. ## self.assertEqual(id(e), id(ee))
  556. ## self.assertEqual(d, e)
  557. ##
  558. ## d.x = d
  559. ## e = pickle.loads(pickle.dumps(d))
  560. ## self.assertEqual(id(e), id(e.x))
  561. ##
  562. ## d = DequeWithBadIter('abc')
  563. ## self.assertRaises(TypeError, pickle.dumps, d)
  564. def test_weakref(self):
  565. d = deque('gallahad')
  566. p = weakref.proxy(d)
  567. self.assertEqual(str(p), str(d))
  568. d = None
  569. self.assertRaises(ReferenceError, str, p)
  570. def test_strange_subclass(self):
  571. class X(deque):
  572. def __iter__(self):
  573. return iter([])
  574. d1 = X([1,2,3])
  575. d2 = X([4,5,6])
  576. d1 == d2 # not clear if this is supposed to be True or False,
  577. # but it used to give a SystemError
  578. class SubclassWithKwargs(deque):
  579. def __init__(self, newarg=1):
  580. deque.__init__(self)
  581. class TestSubclassWithKwargs(unittest.TestCase):
  582. def test_subclass_with_kwargs(self):
  583. # SF bug #1486663 -- this used to erroneously raise a TypeError
  584. SubclassWithKwargs(newarg=1)
  585. def test_free_after_iterating(self):
  586. # For now, bypass tests that require slicing
  587. self.skipTest("Exhausted deque iterator doesn't free a deque")
  588. #==============================================================================
  589. libreftest = """
  590. Example from the Library Reference: Doc/lib/libcollections.tex
  591. >>> from collections import deque
  592. >>> d = deque('ghi') # make a new deque with three items
  593. >>> for elem in d: # iterate over the deque's elements
  594. ... print elem.upper()
  595. G
  596. H
  597. I
  598. >>> d.append('j') # add a new entry to the right side
  599. >>> d.appendleft('f') # add a new entry to the left side
  600. >>> d # show the representation of the deque
  601. deque(['f', 'g', 'h', 'i', 'j'])
  602. >>> d.pop() # return and remove the rightmost item
  603. 'j'
  604. >>> d.popleft() # return and remove the leftmost item
  605. 'f'
  606. >>> list(d) # list the contents of the deque
  607. ['g', 'h', 'i']
  608. >>> d[0] # peek at leftmost item
  609. 'g'
  610. >>> d[-1] # peek at rightmost item
  611. 'i'
  612. >>> list(reversed(d)) # list the contents of a deque in reverse
  613. ['i', 'h', 'g']
  614. >>> 'h' in d # search the deque
  615. True
  616. >>> d.extend('jkl') # add multiple elements at once
  617. >>> d
  618. deque(['g', 'h', 'i', 'j', 'k', 'l'])
  619. >>> d.rotate(1) # right rotation
  620. >>> d
  621. deque(['l', 'g', 'h', 'i', 'j', 'k'])
  622. >>> d.rotate(-1) # left rotation
  623. >>> d
  624. deque(['g', 'h', 'i', 'j', 'k', 'l'])
  625. >>> deque(reversed(d)) # make a new deque in reverse order
  626. deque(['l', 'k', 'j', 'i', 'h', 'g'])
  627. >>> d.clear() # empty the deque
  628. >>> d.pop() # cannot pop from an empty deque
  629. Traceback (most recent call last):
  630. File "<pyshell#6>", line 1, in -toplevel-
  631. d.pop()
  632. IndexError: pop from an empty deque
  633. >>> d.extendleft('abc') # extendleft() reverses the input order
  634. >>> d
  635. deque(['c', 'b', 'a'])
  636. >>> def delete_nth(d, n):
  637. ... d.rotate(-n)
  638. ... d.popleft()
  639. ... d.rotate(n)
  640. ...
  641. >>> d = deque('abcdef')
  642. >>> delete_nth(d, 2) # remove the entry at d[2]
  643. >>> d
  644. deque(['a', 'b', 'd', 'e', 'f'])
  645. >>> def roundrobin(*iterables):
  646. ... pending = deque(iter(i) for i in iterables)
  647. ... while pending:
  648. ... task = pending.popleft()
  649. ... try:
  650. ... yield task.next()
  651. ... except StopIteration:
  652. ... continue
  653. ... pending.append(task)
  654. ...
  655. >>> for value in roundrobin('abc', 'd', 'efgh'):
  656. ... print value
  657. ...
  658. a
  659. d
  660. e
  661. b
  662. f
  663. c
  664. g
  665. h
  666. >>> def maketree(iterable):
  667. ... d = deque(iterable)
  668. ... while len(d) > 1:
  669. ... pair = [d.popleft(), d.popleft()]
  670. ... d.append(pair)
  671. ... return list(d)
  672. ...
  673. >>> print maketree('abcdefgh')
  674. [[[['a', 'b'], ['c', 'd']], [['e', 'f'], ['g', 'h']]]]
  675. """
  676. #==============================================================================
  677. __test__ = {'libreftest' : libreftest}
  678. def test_main(verbose=None):
  679. import sys
  680. test_classes = (
  681. TestBasic,
  682. TestVariousIteratorArgs,
  683. TestSubclass,
  684. TestSubclassWithKwargs,
  685. )
  686. test_support.run_unittest(*test_classes)
  687. # verify reference counting
  688. if verbose and hasattr(sys, "gettotalrefcount"):
  689. import gc
  690. counts = [None] * 5
  691. for i in xrange(len(counts)):
  692. test_support.run_unittest(*test_classes)
  693. gc.collect()
  694. counts[i] = sys.gettotalrefcount()
  695. print counts
  696. # doctests
  697. from test import test_deque
  698. test_support.run_doctest(test_deque, verbose)
  699. if __name__ == "__main__":
  700. test_main(verbose=True)