blob: 87187161ab30d84bbc2febf7cfa31bec431bd339 [file] [log] [blame]
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001from collections import deque
2import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00003from test import support, seq_tests
Antoine Pitrou7ddda782009-01-01 15:35:33 +00004import gc
5import weakref
Raymond Hettinger756b3f32004-01-29 06:37:52 +00006import copy
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00007import pickle
Guido van Rossum34d19282007-08-09 01:03:29 +00008from io import StringIO
Raymond Hettinger0a4977c2004-03-01 23:16:22 +00009import random
Jesus Cea16e2fca2012-08-03 14:49:42 +020010import struct
Raymond Hettinger756b3f32004-01-29 06:37:52 +000011
12BIG = 100000
13
Raymond Hettingera435c532004-07-09 04:10:20 +000014def fail():
15 raise SyntaxError
16 yield 1
17
Raymond Hettinger4aec61e2005-03-18 21:20:23 +000018class BadCmp:
19 def __eq__(self, other):
20 raise RuntimeError
21
22class MutateCmp:
Raymond Hettingerd73202c2005-03-19 00:00:51 +000023 def __init__(self, deque, result):
Raymond Hettinger4aec61e2005-03-18 21:20:23 +000024 self.deque = deque
Raymond Hettingerd73202c2005-03-19 00:00:51 +000025 self.result = result
Raymond Hettinger4aec61e2005-03-18 21:20:23 +000026 def __eq__(self, other):
27 self.deque.clear()
Raymond Hettingerd73202c2005-03-19 00:00:51 +000028 return self.result
Raymond Hettinger4aec61e2005-03-18 21:20:23 +000029
Raymond Hettinger756b3f32004-01-29 06:37:52 +000030class TestBasic(unittest.TestCase):
31
32 def test_basics(self):
Christian Heimes38053212007-12-14 01:24:44 +000033 d = deque(range(-5125, -5000))
34 d.__init__(range(200))
Guido van Rossum805365e2007-05-07 22:24:25 +000035 for i in range(200, 400):
Raymond Hettinger756b3f32004-01-29 06:37:52 +000036 d.append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +000037 for i in reversed(range(-200, 0)):
Raymond Hettinger756b3f32004-01-29 06:37:52 +000038 d.appendleft(i)
Guido van Rossum805365e2007-05-07 22:24:25 +000039 self.assertEqual(list(d), list(range(-200, 400)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +000040 self.assertEqual(len(d), 600)
41
Guido van Rossum805365e2007-05-07 22:24:25 +000042 left = [d.popleft() for i in range(250)]
43 self.assertEqual(left, list(range(-200, 50)))
44 self.assertEqual(list(d), list(range(50, 400)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +000045
Guido van Rossum805365e2007-05-07 22:24:25 +000046 right = [d.pop() for i in range(250)]
Raymond Hettinger756b3f32004-01-29 06:37:52 +000047 right.reverse()
Guido van Rossum805365e2007-05-07 22:24:25 +000048 self.assertEqual(right, list(range(150, 400)))
49 self.assertEqual(list(d), list(range(50, 150)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +000050
Guido van Rossum8ce8a782007-11-01 19:42:39 +000051 def test_maxlen(self):
52 self.assertRaises(ValueError, deque, 'abc', -1)
53 self.assertRaises(ValueError, deque, 'abc', -2)
Raymond Hettinger060c7f62009-03-10 09:36:07 +000054 it = iter(range(10))
55 d = deque(it, maxlen=3)
56 self.assertEqual(list(it), [])
Guido van Rossum8ce8a782007-11-01 19:42:39 +000057 self.assertEqual(repr(d), 'deque([7, 8, 9], maxlen=3)')
58 self.assertEqual(list(d), [7, 8, 9])
59 self.assertEqual(d, deque(range(10), 3))
60 d.append(10)
61 self.assertEqual(list(d), [8, 9, 10])
62 d.appendleft(7)
63 self.assertEqual(list(d), [7, 8, 9])
64 d.extend([10, 11])
65 self.assertEqual(list(d), [9, 10, 11])
66 d.extendleft([8, 7])
67 self.assertEqual(list(d), [7, 8, 9])
68 d = deque(range(200), maxlen=10)
69 d.append(d)
Benjamin Petersonee8712c2008-05-20 21:35:26 +000070 support.unlink(support.TESTFN)
71 fo = open(support.TESTFN, "w")
Christian Heimescc47b052008-03-25 14:56:36 +000072 try:
73 fo.write(str(d))
74 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000075 fo = open(support.TESTFN, "r")
Christian Heimescc47b052008-03-25 14:56:36 +000076 self.assertEqual(fo.read(), repr(d))
77 finally:
78 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000079 support.unlink(support.TESTFN)
Christian Heimescc47b052008-03-25 14:56:36 +000080
Guido van Rossum8ce8a782007-11-01 19:42:39 +000081 d = deque(range(10), maxlen=None)
82 self.assertEqual(repr(d), 'deque([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])')
Benjamin Petersonee8712c2008-05-20 21:35:26 +000083 fo = open(support.TESTFN, "w")
Christian Heimescc47b052008-03-25 14:56:36 +000084 try:
85 fo.write(str(d))
86 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000087 fo = open(support.TESTFN, "r")
Christian Heimescc47b052008-03-25 14:56:36 +000088 self.assertEqual(fo.read(), repr(d))
89 finally:
90 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000091 support.unlink(support.TESTFN)
Guido van Rossum8ce8a782007-11-01 19:42:39 +000092
Raymond Hettinger060c7f62009-03-10 09:36:07 +000093 def test_maxlen_zero(self):
94 it = iter(range(100))
95 deque(it, maxlen=0)
96 self.assertEqual(list(it), [])
97
98 it = iter(range(100))
99 d = deque(maxlen=0)
100 d.extend(it)
101 self.assertEqual(list(it), [])
102
103 it = iter(range(100))
104 d = deque(maxlen=0)
105 d.extendleft(it)
106 self.assertEqual(list(it), [])
107
Raymond Hettinger5bb0f0e2009-03-10 12:56:32 +0000108 def test_maxlen_attribute(self):
109 self.assertEqual(deque().maxlen, None)
110 self.assertEqual(deque('abc').maxlen, None)
111 self.assertEqual(deque('abc', maxlen=4).maxlen, 4)
112 self.assertEqual(deque('abc', maxlen=2).maxlen, 2)
113 self.assertEqual(deque('abc', maxlen=0).maxlen, 0)
114 with self.assertRaises(AttributeError):
115 d = deque('abc')
116 d.maxlen = 10
117
Raymond Hettinger44459de2010-04-03 23:20:46 +0000118 def test_count(self):
119 for s in ('', 'abracadabra', 'simsalabim'*500+'abc'):
120 s = list(s)
121 d = deque(s)
122 for letter in 'abcdefghijklmnopqrstuvwxyz':
123 self.assertEqual(s.count(letter), d.count(letter), (s, d, letter))
124 self.assertRaises(TypeError, d.count) # too few args
125 self.assertRaises(TypeError, d.count, 1, 2) # too many args
126 class BadCompare:
127 def __eq__(self, other):
128 raise ArithmeticError
129 d = deque([1, 2, BadCompare(), 3])
130 self.assertRaises(ArithmeticError, d.count, 2)
131 d = deque([1, 2, 3])
132 self.assertRaises(ArithmeticError, d.count, BadCompare())
133 class MutatingCompare:
134 def __eq__(self, other):
135 self.d.pop()
136 return True
137 m = MutatingCompare()
138 d = deque([1, 2, 3, m, 4, 5])
139 m.d = d
140 self.assertRaises(RuntimeError, d.count, 3)
141
Raymond Hettinger512d2cc2011-01-25 21:32:39 +0000142 # test issue11004
143 # block advance failed after rotation aligned elements on right side of block
144 d = deque([None]*16)
145 for i in range(len(d)):
146 d.rotate(-1)
147 d.rotate(1)
148 self.assertEqual(d.count(1), 0)
149 self.assertEqual(d.count(None), 16)
150
Raymond Hettinger738ec902004-02-29 02:15:56 +0000151 def test_comparisons(self):
152 d = deque('xabc'); d.popleft()
153 for e in [d, deque('abc'), deque('ab'), deque(), list(d)]:
154 self.assertEqual(d==e, type(d)==type(e) and list(d)==list(e))
155 self.assertEqual(d!=e, not(type(d)==type(e) and list(d)==list(e)))
156
157 args = map(deque, ('', 'a', 'b', 'ab', 'ba', 'abc', 'xba', 'xabc', 'cba'))
158 for x in args:
159 for y in args:
160 self.assertEqual(x == y, list(x) == list(y), (x,y))
161 self.assertEqual(x != y, list(x) != list(y), (x,y))
162 self.assertEqual(x < y, list(x) < list(y), (x,y))
163 self.assertEqual(x <= y, list(x) <= list(y), (x,y))
164 self.assertEqual(x > y, list(x) > list(y), (x,y))
165 self.assertEqual(x >= y, list(x) >= list(y), (x,y))
Raymond Hettinger738ec902004-02-29 02:15:56 +0000166
Raymond Hettinger39dadf72015-03-20 16:38:56 -0700167 def test_contains(self):
168 n = 200
169
170 d = deque(range(n))
171 for i in range(n):
172 self.assertTrue(i in d)
173 self.assertTrue((n+1) not in d)
174
175 # Test detection of mutation during iteration
176 d = deque(range(n))
177 d[n//2] = MutateCmp(d, False)
178 with self.assertRaises(RuntimeError):
179 n in d
180
181 # Test detection of comparison exceptions
182 d = deque(range(n))
183 d[n//2] = BadCmp()
184 with self.assertRaises(RuntimeError):
185 n in d
186
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000187 def test_extend(self):
188 d = deque('a')
189 self.assertRaises(TypeError, d.extend, 1)
190 d.extend('bcd')
191 self.assertEqual(list(d), list('abcd'))
Raymond Hettinger3f9afd82009-12-10 03:03:02 +0000192 d.extend(d)
193 self.assertEqual(list(d), list('abcdabcd'))
194
Raymond Hettinger41290a62015-03-31 08:12:23 -0700195 def test_add(self):
196 d = deque()
197 e = deque('abc')
198 f = deque('def')
199 self.assertEqual(d + d, deque())
200 self.assertEqual(e + f, deque('abcdef'))
201 self.assertEqual(e + e, deque('abcabc'))
202 self.assertEqual(e + d, deque('abc'))
203 self.assertEqual(d + e, deque('abc'))
204 self.assertIsNot(d + d, deque())
205 self.assertIsNot(e + d, deque('abc'))
206 self.assertIsNot(d + e, deque('abc'))
207
208 g = deque('abcdef', maxlen=4)
209 h = deque('gh')
210 self.assertEqual(g + h, deque('efgh'))
211
212 with self.assertRaises(TypeError):
213 deque('abc') + 'def'
214
Raymond Hettinger3f9afd82009-12-10 03:03:02 +0000215 def test_iadd(self):
216 d = deque('a')
217 d += 'bcd'
218 self.assertEqual(list(d), list('abcd'))
219 d += d
220 self.assertEqual(list(d), list('abcdabcd'))
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000221
222 def test_extendleft(self):
223 d = deque('a')
224 self.assertRaises(TypeError, d.extendleft, 1)
225 d.extendleft('bcd')
226 self.assertEqual(list(d), list(reversed('abcd')))
Raymond Hettinger3f9afd82009-12-10 03:03:02 +0000227 d.extendleft(d)
228 self.assertEqual(list(d), list('abcddcba'))
Raymond Hettingera435c532004-07-09 04:10:20 +0000229 d = deque()
230 d.extendleft(range(1000))
231 self.assertEqual(list(d), list(reversed(range(1000))))
232 self.assertRaises(SyntaxError, d.extendleft, fail())
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000233
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000234 def test_getitem(self):
235 n = 200
Guido van Rossum805365e2007-05-07 22:24:25 +0000236 d = deque(range(n))
237 l = list(range(n))
238 for i in range(n):
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000239 d.popleft()
240 l.pop(0)
241 if random.random() < 0.5:
242 d.append(i)
243 l.append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000244 for j in range(1-len(l), len(l)):
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000245 assert d[j] == l[j]
246
Raymond Hettinger738ec902004-02-29 02:15:56 +0000247 d = deque('superman')
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000248 self.assertEqual(d[0], 's')
249 self.assertEqual(d[-1], 'n')
Raymond Hettinger738ec902004-02-29 02:15:56 +0000250 d = deque()
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000251 self.assertRaises(IndexError, d.__getitem__, 0)
252 self.assertRaises(IndexError, d.__getitem__, -1)
253
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700254 def test_index(self):
255 for n in 1, 2, 30, 40, 200:
256
257 d = deque(range(n))
258 for i in range(n):
259 self.assertEqual(d.index(i), i)
260
261 with self.assertRaises(ValueError):
262 d.index(n+1)
263
264 # Test detection of mutation during iteration
265 d = deque(range(n))
266 d[n//2] = MutateCmp(d, False)
267 with self.assertRaises(RuntimeError):
268 d.index(n)
269
270 # Test detection of comparison exceptions
271 d = deque(range(n))
272 d[n//2] = BadCmp()
273 with self.assertRaises(RuntimeError):
274 d.index(n)
275
276 # Test start and stop arguments behavior matches list.index()
277 elements = 'ABCDEFGHI'
278 nonelement = 'Z'
279 d = deque(elements * 2)
280 s = list(elements * 2)
281 for start in range(-5 - len(s)*2, 5 + len(s) * 2):
282 for stop in range(-5 - len(s)*2, 5 + len(s) * 2):
283 for element in elements + 'Z':
284 try:
285 target = s.index(element, start, stop)
286 except ValueError:
287 with self.assertRaises(ValueError):
288 d.index(element, start, stop)
289 else:
290 self.assertEqual(d.index(element, start, stop), target)
291
Brett Cannondf6b5442015-09-03 10:15:03 -0700292 def test_insert_bug_24913(self):
293 d = deque('A' * 3)
294 with self.assertRaises(ValueError):
295 i = d.index("Hello world", 0, 4)
296
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700297 def test_insert(self):
298 # Test to make sure insert behaves like lists
299 elements = 'ABCDEFGHI'
300 for i in range(-5 - len(elements)*2, 5 + len(elements) * 2):
301 d = deque('ABCDEFGHI')
302 s = list('ABCDEFGHI')
303 d.insert(i, 'Z')
304 s.insert(i, 'Z')
305 self.assertEqual(list(d), s)
306
Raymond Hettinger41290a62015-03-31 08:12:23 -0700307 def test_imul(self):
308 for n in (-10, -1, 0, 1, 2, 10, 1000):
309 d = deque()
310 d *= n
311 self.assertEqual(d, deque())
312 self.assertIsNone(d.maxlen)
313
314 for n in (-10, -1, 0, 1, 2, 10, 1000):
315 d = deque('a')
316 d *= n
317 self.assertEqual(d, deque('a' * n))
318 self.assertIsNone(d.maxlen)
319
320 for n in (-10, -1, 0, 1, 2, 10, 499, 500, 501, 1000):
321 d = deque('a', 500)
322 d *= n
323 self.assertEqual(d, deque('a' * min(n, 500)))
324 self.assertEqual(d.maxlen, 500)
325
326 for n in (-10, -1, 0, 1, 2, 10, 1000):
327 d = deque('abcdef')
328 d *= n
329 self.assertEqual(d, deque('abcdef' * n))
330 self.assertIsNone(d.maxlen)
331
332 for n in (-10, -1, 0, 1, 2, 10, 499, 500, 501, 1000):
333 d = deque('abcdef', 500)
334 d *= n
335 self.assertEqual(d, deque(('abcdef' * n)[-500:]))
336 self.assertEqual(d.maxlen, 500)
337
338 def test_mul(self):
339 d = deque('abc')
340 self.assertEqual(d * -5, deque())
341 self.assertEqual(d * 0, deque())
342 self.assertEqual(d * 1, deque('abc'))
343 self.assertEqual(d * 2, deque('abcabc'))
344 self.assertEqual(d * 3, deque('abcabcabc'))
345 self.assertIsNot(d * 1, d)
346
347 self.assertEqual(deque() * 0, deque())
348 self.assertEqual(deque() * 1, deque())
349 self.assertEqual(deque() * 5, deque())
350
351 self.assertEqual(-5 * d, deque())
352 self.assertEqual(0 * d, deque())
353 self.assertEqual(1 * d, deque('abc'))
354 self.assertEqual(2 * d, deque('abcabc'))
355 self.assertEqual(3 * d, deque('abcabcabc'))
356
357 d = deque('abc', maxlen=5)
358 self.assertEqual(d * -5, deque())
359 self.assertEqual(d * 0, deque())
360 self.assertEqual(d * 1, deque('abc'))
361 self.assertEqual(d * 2, deque('bcabc'))
362 self.assertEqual(d * 30, deque('bcabc'))
363
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000364 def test_setitem(self):
365 n = 200
Guido van Rossum805365e2007-05-07 22:24:25 +0000366 d = deque(range(n))
367 for i in range(n):
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000368 d[i] = 10 * i
Guido van Rossum805365e2007-05-07 22:24:25 +0000369 self.assertEqual(list(d), [10*i for i in range(n)])
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000370 l = list(d)
Guido van Rossum805365e2007-05-07 22:24:25 +0000371 for i in range(1-n, 0, -1):
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000372 d[i] = 7*i
373 l[i] = 7*i
374 self.assertEqual(list(d), l)
Raymond Hettinger738ec902004-02-29 02:15:56 +0000375
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000376 def test_delitem(self):
377 n = 500 # O(n**2) test, don't make this too big
Guido van Rossum805365e2007-05-07 22:24:25 +0000378 d = deque(range(n))
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000379 self.assertRaises(IndexError, d.__delitem__, -n-1)
380 self.assertRaises(IndexError, d.__delitem__, n)
Guido van Rossum805365e2007-05-07 22:24:25 +0000381 for i in range(n):
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000382 self.assertEqual(len(d), n-i)
383 j = random.randrange(-len(d), len(d))
384 val = d[j]
Benjamin Peterson577473f2010-01-19 00:09:57 +0000385 self.assertIn(val, d)
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000386 del d[j]
Benjamin Peterson577473f2010-01-19 00:09:57 +0000387 self.assertNotIn(val, d)
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000388 self.assertEqual(len(d), 0)
389
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000390 def test_reverse(self):
391 n = 500 # O(n**2) test, don't make this too big
392 data = [random.random() for i in range(n)]
393 for i in range(n):
394 d = deque(data[:i])
395 r = d.reverse()
396 self.assertEqual(list(d), list(reversed(data[:i])))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000397 self.assertIs(r, None)
Raymond Hettingere5fdedb2009-12-10 00:47:21 +0000398 d.reverse()
399 self.assertEqual(list(d), data[:i])
400 self.assertRaises(TypeError, d.reverse, 1) # Arity is zero
401
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000402 def test_rotate(self):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000403 s = tuple('abcde')
404 n = len(s)
405
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000406 d = deque(s)
Raymond Hettingeree33b272004-02-08 04:05:26 +0000407 d.rotate(1) # verify rot(1)
408 self.assertEqual(''.join(d), 'eabcd')
409
410 d = deque(s)
411 d.rotate(-1) # verify rot(-1)
412 self.assertEqual(''.join(d), 'bcdea')
413 d.rotate() # check default to 1
414 self.assertEqual(tuple(d), s)
415
Guido van Rossum805365e2007-05-07 22:24:25 +0000416 for i in range(n*3):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000417 d = deque(s)
418 e = deque(d)
419 d.rotate(i) # check vs. rot(1) n times
Guido van Rossum805365e2007-05-07 22:24:25 +0000420 for j in range(i):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000421 e.rotate(1)
422 self.assertEqual(tuple(d), tuple(e))
423 d.rotate(-i) # check that it works in reverse
424 self.assertEqual(tuple(d), s)
425 e.rotate(n-i) # check that it wraps forward
426 self.assertEqual(tuple(e), s)
427
Guido van Rossum805365e2007-05-07 22:24:25 +0000428 for i in range(n*3):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000429 d = deque(s)
430 e = deque(d)
431 d.rotate(-i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000432 for j in range(i):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000433 e.rotate(-1) # check vs. rot(-1) n times
434 self.assertEqual(tuple(d), tuple(e))
435 d.rotate(i) # check that it works in reverse
436 self.assertEqual(tuple(d), s)
437 e.rotate(i-n) # check that it wraps backaround
438 self.assertEqual(tuple(e), s)
439
440 d = deque(s)
441 e = deque(s)
442 e.rotate(BIG+17) # verify on long series of rotates
443 dr = d.rotate
Guido van Rossum805365e2007-05-07 22:24:25 +0000444 for i in range(BIG+17):
Raymond Hettingeree33b272004-02-08 04:05:26 +0000445 dr()
446 self.assertEqual(tuple(d), tuple(e))
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000447
Raymond Hettingera435c532004-07-09 04:10:20 +0000448 self.assertRaises(TypeError, d.rotate, 'x') # Wrong arg type
449 self.assertRaises(TypeError, d.rotate, 1, 10) # Too many args
450
451 d = deque()
452 d.rotate() # rotate an empty deque
453 self.assertEqual(d, deque())
454
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000455 def test_len(self):
456 d = deque('ab')
457 self.assertEqual(len(d), 2)
458 d.popleft()
459 self.assertEqual(len(d), 1)
460 d.pop()
461 self.assertEqual(len(d), 0)
Raymond Hettinger738ec902004-02-29 02:15:56 +0000462 self.assertRaises(IndexError, d.pop)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000463 self.assertEqual(len(d), 0)
464 d.append('c')
465 self.assertEqual(len(d), 1)
466 d.appendleft('d')
467 self.assertEqual(len(d), 2)
468 d.clear()
469 self.assertEqual(len(d), 0)
470
471 def test_underflow(self):
472 d = deque()
Raymond Hettinger738ec902004-02-29 02:15:56 +0000473 self.assertRaises(IndexError, d.pop)
474 self.assertRaises(IndexError, d.popleft)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000475
476 def test_clear(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000477 d = deque(range(100))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000478 self.assertEqual(len(d), 100)
479 d.clear()
480 self.assertEqual(len(d), 0)
481 self.assertEqual(list(d), [])
Raymond Hettingera435c532004-07-09 04:10:20 +0000482 d.clear() # clear an emtpy deque
483 self.assertEqual(list(d), [])
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000484
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000485 def test_remove(self):
486 d = deque('abcdefghcij')
487 d.remove('c')
488 self.assertEqual(d, deque('abdefghcij'))
489 d.remove('c')
490 self.assertEqual(d, deque('abdefghij'))
491 self.assertRaises(ValueError, d.remove, 'c')
492 self.assertEqual(d, deque('abdefghij'))
493
Walter Dörwaldc448a912005-03-22 11:22:38 +0000494 # Handle comparison errors
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000495 d = deque(['a', 'b', BadCmp(), 'c'])
496 e = deque(d)
497 self.assertRaises(RuntimeError, d.remove, 'c')
498 for x, y in zip(d, e):
499 # verify that original order and values are retained.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000500 self.assertTrue(x is y)
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000501
502 # Handle evil mutator
Raymond Hettingerd73202c2005-03-19 00:00:51 +0000503 for match in (True, False):
504 d = deque(['ab'])
505 d.extend([MutateCmp(d, match), 'c'])
506 self.assertRaises(IndexError, d.remove, 'c')
507 self.assertEqual(d, deque())
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000508
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000509 def test_repr(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000510 d = deque(range(200))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000511 e = eval(repr(d))
512 self.assertEqual(list(d), list(e))
513 d.append(d)
Benjamin Peterson577473f2010-01-19 00:09:57 +0000514 self.assertIn('...', repr(d))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000515
516 def test_print(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000517 d = deque(range(200))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000518 d.append(d)
Raymond Hettingera435c532004-07-09 04:10:20 +0000519 try:
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000520 support.unlink(support.TESTFN)
521 fo = open(support.TESTFN, "w")
Christian Heimescc47b052008-03-25 14:56:36 +0000522 print(d, file=fo, end='')
Raymond Hettingera435c532004-07-09 04:10:20 +0000523 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000524 fo = open(support.TESTFN, "r")
Raymond Hettingera435c532004-07-09 04:10:20 +0000525 self.assertEqual(fo.read(), repr(d))
526 finally:
527 fo.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000528 support.unlink(support.TESTFN)
Raymond Hettingera435c532004-07-09 04:10:20 +0000529
530 def test_init(self):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000531 self.assertRaises(TypeError, deque, 'abc', 2, 3);
Raymond Hettingera435c532004-07-09 04:10:20 +0000532 self.assertRaises(TypeError, deque, 1);
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000533
534 def test_hash(self):
535 self.assertRaises(TypeError, hash, deque('abc'))
536
537 def test_long_steadystate_queue_popleft(self):
538 for size in (0, 1, 2, 100, 1000):
Guido van Rossum805365e2007-05-07 22:24:25 +0000539 d = deque(range(size))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000540 append, pop = d.append, d.popleft
Guido van Rossum805365e2007-05-07 22:24:25 +0000541 for i in range(size, BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000542 append(i)
543 x = pop()
544 if x != i - size:
545 self.assertEqual(x, i-size)
Guido van Rossum805365e2007-05-07 22:24:25 +0000546 self.assertEqual(list(d), list(range(BIG-size, BIG)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000547
548 def test_long_steadystate_queue_popright(self):
549 for size in (0, 1, 2, 100, 1000):
Guido van Rossum805365e2007-05-07 22:24:25 +0000550 d = deque(reversed(range(size)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000551 append, pop = d.appendleft, d.pop
Guido van Rossum805365e2007-05-07 22:24:25 +0000552 for i in range(size, BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000553 append(i)
554 x = pop()
555 if x != i - size:
556 self.assertEqual(x, i-size)
Guido van Rossum805365e2007-05-07 22:24:25 +0000557 self.assertEqual(list(reversed(list(d))),
558 list(range(BIG-size, BIG)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000559
560 def test_big_queue_popleft(self):
561 pass
562 d = deque()
563 append, pop = d.append, d.popleft
Guido van Rossum805365e2007-05-07 22:24:25 +0000564 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000565 append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000566 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000567 x = pop()
568 if x != i:
569 self.assertEqual(x, i)
570
571 def test_big_queue_popright(self):
572 d = deque()
573 append, pop = d.appendleft, d.pop
Guido van Rossum805365e2007-05-07 22:24:25 +0000574 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000575 append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000576 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000577 x = pop()
578 if x != i:
579 self.assertEqual(x, i)
580
581 def test_big_stack_right(self):
582 d = deque()
583 append, pop = d.append, d.pop
Guido van Rossum805365e2007-05-07 22:24:25 +0000584 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000585 append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000586 for i in reversed(range(BIG)):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000587 x = pop()
588 if x != i:
589 self.assertEqual(x, i)
590 self.assertEqual(len(d), 0)
591
592 def test_big_stack_left(self):
593 d = deque()
594 append, pop = d.appendleft, d.popleft
Guido van Rossum805365e2007-05-07 22:24:25 +0000595 for i in range(BIG):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000596 append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000597 for i in reversed(range(BIG)):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000598 x = pop()
599 if x != i:
600 self.assertEqual(x, i)
601 self.assertEqual(len(d), 0)
602
603 def test_roundtrip_iter_init(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000604 d = deque(range(200))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000605 e = deque(d)
606 self.assertNotEqual(id(d), id(e))
607 self.assertEqual(list(d), list(e))
608
609 def test_pickle(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000610 d = deque(range(200))
Hirokazu Yamamoto801f9d32008-12-27 04:21:44 +0000611 for i in range(pickle.HIGHEST_PROTOCOL + 1):
Raymond Hettinger952f8802004-11-09 07:27:35 +0000612 s = pickle.dumps(d, i)
613 e = pickle.loads(s)
614 self.assertNotEqual(id(d), id(e))
615 self.assertEqual(list(d), list(e))
616
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000617## def test_pickle_recursive(self):
618## d = deque('abc')
619## d.append(d)
Hirokazu Yamamoto801f9d32008-12-27 04:21:44 +0000620## for i in range(pickle.HIGHEST_PROTOCOL + 1):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000621## e = pickle.loads(pickle.dumps(d, i))
622## self.assertNotEqual(id(d), id(e))
623## self.assertEqual(id(e), id(e[-1]))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000624
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000625 def test_iterator_pickle(self):
626 data = deque(range(200))
Serhiy Storchakabad12572014-12-15 14:03:42 +0200627 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
628 it = itorg = iter(data)
629 d = pickle.dumps(it, proto)
630 it = pickle.loads(d)
631 self.assertEqual(type(itorg), type(it))
632 self.assertEqual(list(it), list(data))
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000633
Serhiy Storchakabad12572014-12-15 14:03:42 +0200634 it = pickle.loads(d)
635 next(it)
636 d = pickle.dumps(it, proto)
637 self.assertEqual(list(it), list(data)[1:])
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000638
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000639 def test_deepcopy(self):
640 mut = [10]
641 d = deque([mut])
642 e = copy.deepcopy(d)
643 self.assertEqual(list(d), list(e))
644 mut[0] = 11
645 self.assertNotEqual(id(d), id(e))
646 self.assertNotEqual(list(d), list(e))
647
648 def test_copy(self):
649 mut = [10]
650 d = deque([mut])
651 e = copy.copy(d)
652 self.assertEqual(list(d), list(e))
653 mut[0] = 11
654 self.assertNotEqual(id(d), id(e))
655 self.assertEqual(list(d), list(e))
656
Raymond Hettinger32ea1652015-03-21 01:37:37 -0700657 def test_copy_method(self):
658 mut = [10]
659 d = deque([mut])
660 e = d.copy()
661 self.assertEqual(list(d), list(e))
662 mut[0] = 11
663 self.assertNotEqual(id(d), id(e))
664 self.assertEqual(list(d), list(e))
665
Raymond Hettingerc058fd12004-02-07 02:45:22 +0000666 def test_reversed(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000667 for s in ('abcd', range(2000)):
Raymond Hettingerc058fd12004-02-07 02:45:22 +0000668 self.assertEqual(list(reversed(deque(s))), list(reversed(s)))
669
Raymond Hettingerbadf5d82014-06-14 20:41:22 -0700670 def test_reversed_new(self):
671 klass = type(reversed(deque()))
672 for s in ('abcd', range(2000)):
673 self.assertEqual(list(klass(deque(s))), list(reversed(s)))
674
Tim Peters10c7e862004-10-01 02:01:04 +0000675 def test_gc_doesnt_blowup(self):
676 import gc
677 # This used to assert-fail in deque_traverse() under a debug
678 # build, or run wild with a NULL pointer in a release build.
679 d = deque()
Guido van Rossum805365e2007-05-07 22:24:25 +0000680 for i in range(100):
Tim Peters10c7e862004-10-01 02:01:04 +0000681 d.append(1)
682 gc.collect()
683
Antoine Pitrou7ddda782009-01-01 15:35:33 +0000684 def test_container_iterator(self):
685 # Bug #3680: tp_traverse was not implemented for deque iterator objects
686 class C(object):
687 pass
688 for i in range(2):
689 obj = C()
690 ref = weakref.ref(obj)
691 if i == 0:
692 container = deque([obj, 1])
693 else:
694 container = reversed(deque([obj, 1]))
695 obj.x = iter(container)
696 del obj, container
697 gc.collect()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000698 self.assertTrue(ref() is None, "Cycle was not collected")
Antoine Pitrou7ddda782009-01-01 15:35:33 +0000699
Jesus Cea16e2fca2012-08-03 14:49:42 +0200700 check_sizeof = support.check_sizeof
701
702 @support.cpython_only
703 def test_sizeof(self):
Raymond Hettingerdaf57f22015-02-26 23:21:29 -0800704 BLOCKLEN = 64
Jesus Cea1659b752012-08-03 14:52:12 +0200705 basesize = support.calcobjsize('2P4nlP')
Jesus Cea16e2fca2012-08-03 14:49:42 +0200706 blocksize = struct.calcsize('2P%dP' % BLOCKLEN)
707 self.assertEqual(object.__sizeof__(deque()), basesize)
708 check = self.check_sizeof
709 check(deque(), basesize + blocksize)
710 check(deque('a'), basesize + blocksize)
Raymond Hettingerd9c116c2013-07-09 00:13:21 -0700711 check(deque('a' * (BLOCKLEN - 1)), basesize + blocksize)
712 check(deque('a' * BLOCKLEN), basesize + 2 * blocksize)
Jesus Cea16e2fca2012-08-03 14:49:42 +0200713 check(deque('a' * (42 * BLOCKLEN)), basesize + 43 * blocksize)
714
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000715class TestVariousIteratorArgs(unittest.TestCase):
716
717 def test_constructor(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000718 for s in ("123", "", range(1000), ('do', 1.2), range(2000,2200,5)):
Walter Dörwald09a3f2c2005-03-22 22:43:28 +0000719 for g in (seq_tests.Sequence, seq_tests.IterFunc,
720 seq_tests.IterGen, seq_tests.IterFuncStop,
721 seq_tests.itermulti, seq_tests.iterfunc):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000722 self.assertEqual(list(deque(g(s))), list(g(s)))
Walter Dörwald09a3f2c2005-03-22 22:43:28 +0000723 self.assertRaises(TypeError, deque, seq_tests.IterNextOnly(s))
724 self.assertRaises(TypeError, deque, seq_tests.IterNoNext(s))
725 self.assertRaises(ZeroDivisionError, deque, seq_tests.IterGenExc(s))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000726
727 def test_iter_with_altered_data(self):
728 d = deque('abcdefg')
729 it = iter(d)
730 d.pop()
Georg Brandla18af4e2007-04-21 15:47:16 +0000731 self.assertRaises(RuntimeError, next, it)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000732
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000733 def test_runtime_error_on_empty_deque(self):
734 d = deque()
735 it = iter(d)
736 d.append(10)
Georg Brandla18af4e2007-04-21 15:47:16 +0000737 self.assertRaises(RuntimeError, next, it)
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000738
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000739class Deque(deque):
740 pass
741
Raymond Hettinger952f8802004-11-09 07:27:35 +0000742class DequeWithBadIter(deque):
743 def __iter__(self):
744 raise TypeError
745
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000746class TestSubclass(unittest.TestCase):
747
748 def test_basics(self):
Christian Heimes38053212007-12-14 01:24:44 +0000749 d = Deque(range(25))
750 d.__init__(range(200))
Guido van Rossum805365e2007-05-07 22:24:25 +0000751 for i in range(200, 400):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000752 d.append(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000753 for i in reversed(range(-200, 0)):
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000754 d.appendleft(i)
Guido van Rossum805365e2007-05-07 22:24:25 +0000755 self.assertEqual(list(d), list(range(-200, 400)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000756 self.assertEqual(len(d), 600)
757
Guido van Rossum805365e2007-05-07 22:24:25 +0000758 left = [d.popleft() for i in range(250)]
759 self.assertEqual(left, list(range(-200, 50)))
760 self.assertEqual(list(d), list(range(50, 400)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000761
Guido van Rossum805365e2007-05-07 22:24:25 +0000762 right = [d.pop() for i in range(250)]
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000763 right.reverse()
Guido van Rossum805365e2007-05-07 22:24:25 +0000764 self.assertEqual(right, list(range(150, 400)))
765 self.assertEqual(list(d), list(range(50, 150)))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000766
767 d.clear()
768 self.assertEqual(len(d), 0)
769
770 def test_copy_pickle(self):
771
772 d = Deque('abc')
773
774 e = d.__copy__()
775 self.assertEqual(type(d), type(e))
776 self.assertEqual(list(d), list(e))
777
778 e = Deque(d)
779 self.assertEqual(type(d), type(e))
780 self.assertEqual(list(d), list(e))
781
Serhiy Storchakabad12572014-12-15 14:03:42 +0200782 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
783 s = pickle.dumps(d, proto)
784 e = pickle.loads(s)
785 self.assertNotEqual(id(d), id(e))
786 self.assertEqual(type(d), type(e))
787 self.assertEqual(list(d), list(e))
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000788
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000789 d = Deque('abcde', maxlen=4)
Raymond Hettinger952f8802004-11-09 07:27:35 +0000790
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000791 e = d.__copy__()
792 self.assertEqual(type(d), type(e))
793 self.assertEqual(list(d), list(e))
794
795 e = Deque(d)
796 self.assertEqual(type(d), type(e))
797 self.assertEqual(list(d), list(e))
798
Serhiy Storchakabad12572014-12-15 14:03:42 +0200799 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
800 s = pickle.dumps(d, proto)
801 e = pickle.loads(s)
802 self.assertNotEqual(id(d), id(e))
803 self.assertEqual(type(d), type(e))
804 self.assertEqual(list(d), list(e))
Raymond Hettinger952f8802004-11-09 07:27:35 +0000805
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000806## def test_pickle(self):
807## d = Deque('abc')
808## d.append(d)
809##
810## e = pickle.loads(pickle.dumps(d))
811## self.assertNotEqual(id(d), id(e))
812## self.assertEqual(type(d), type(e))
813## dd = d.pop()
814## ee = e.pop()
815## self.assertEqual(id(e), id(ee))
816## self.assertEqual(d, e)
817##
818## d.x = d
819## e = pickle.loads(pickle.dumps(d))
820## self.assertEqual(id(e), id(e.x))
821##
822## d = DequeWithBadIter('abc')
823## self.assertRaises(TypeError, pickle.dumps, d)
Raymond Hettinger952f8802004-11-09 07:27:35 +0000824
Raymond Hettinger691d8052004-05-30 07:26:47 +0000825 def test_weakref(self):
826 d = deque('gallahad')
Antoine Pitrou7ddda782009-01-01 15:35:33 +0000827 p = weakref.proxy(d)
Raymond Hettinger691d8052004-05-30 07:26:47 +0000828 self.assertEqual(str(p), str(d))
829 d = None
830 self.assertRaises(ReferenceError, str, p)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000831
Armin Rigo974d7572004-10-02 13:59:34 +0000832 def test_strange_subclass(self):
833 class X(deque):
834 def __iter__(self):
835 return iter([])
836 d1 = X([1,2,3])
837 d2 = X([4,5,6])
838 d1 == d2 # not clear if this is supposed to be True or False,
839 # but it used to give a SystemError
840
Thomas Woutersb2137042007-02-01 18:02:27 +0000841
842class SubclassWithKwargs(deque):
843 def __init__(self, newarg=1):
844 deque.__init__(self)
845
846class TestSubclassWithKwargs(unittest.TestCase):
847 def test_subclass_with_kwargs(self):
848 # SF bug #1486663 -- this used to erroneously raise a TypeError
849 SubclassWithKwargs(newarg=1)
850
Raymond Hettinger067bbba2015-04-01 08:11:09 -0700851class TestSequence(seq_tests.CommonTest):
852 type2test = deque
853
854 def test_getitem(self):
855 # For now, bypass tests that require slicing
856 pass
857
858 def test_getslice(self):
859 # For now, bypass tests that require slicing
860 pass
861
862 def test_subscript(self):
863 # For now, bypass tests that require slicing
864 pass
865
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000866#==============================================================================
867
Raymond Hettinger738ec902004-02-29 02:15:56 +0000868libreftest = """
869Example from the Library Reference: Doc/lib/libcollections.tex
870
871>>> from collections import deque
872>>> d = deque('ghi') # make a new deque with three items
873>>> for elem in d: # iterate over the deque's elements
Guido van Rossum7131f842007-02-09 20:13:25 +0000874... print(elem.upper())
Raymond Hettinger738ec902004-02-29 02:15:56 +0000875G
876H
877I
878>>> d.append('j') # add a new entry to the right side
879>>> d.appendleft('f') # add a new entry to the left side
880>>> d # show the representation of the deque
881deque(['f', 'g', 'h', 'i', 'j'])
882>>> d.pop() # return and remove the rightmost item
883'j'
884>>> d.popleft() # return and remove the leftmost item
885'f'
886>>> list(d) # list the contents of the deque
887['g', 'h', 'i']
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000888>>> d[0] # peek at leftmost item
Raymond Hettinger738ec902004-02-29 02:15:56 +0000889'g'
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000890>>> d[-1] # peek at rightmost item
Raymond Hettinger738ec902004-02-29 02:15:56 +0000891'i'
892>>> list(reversed(d)) # list the contents of a deque in reverse
893['i', 'h', 'g']
894>>> 'h' in d # search the deque
895True
896>>> d.extend('jkl') # add multiple elements at once
897>>> d
898deque(['g', 'h', 'i', 'j', 'k', 'l'])
899>>> d.rotate(1) # right rotation
900>>> d
901deque(['l', 'g', 'h', 'i', 'j', 'k'])
902>>> d.rotate(-1) # left rotation
903>>> d
904deque(['g', 'h', 'i', 'j', 'k', 'l'])
905>>> deque(reversed(d)) # make a new deque in reverse order
906deque(['l', 'k', 'j', 'i', 'h', 'g'])
907>>> d.clear() # empty the deque
908>>> d.pop() # cannot pop from an empty deque
909Traceback (most recent call last):
910 File "<pyshell#6>", line 1, in -toplevel-
911 d.pop()
912IndexError: pop from an empty deque
913
914>>> d.extendleft('abc') # extendleft() reverses the input order
915>>> d
916deque(['c', 'b', 'a'])
917
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000918
919
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000920>>> def delete_nth(d, n):
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000921... d.rotate(-n)
922... d.popleft()
923... d.rotate(n)
924...
925>>> d = deque('abcdef')
926>>> delete_nth(d, 2) # remove the entry at d[2]
927>>> d
928deque(['a', 'b', 'd', 'e', 'f'])
929
930
931
932>>> def roundrobin(*iterables):
933... pending = deque(iter(i) for i in iterables)
934... while pending:
935... task = pending.popleft()
936... try:
Georg Brandla18af4e2007-04-21 15:47:16 +0000937... yield next(task)
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000938... except StopIteration:
939... continue
940... pending.append(task)
941...
942
943>>> for value in roundrobin('abc', 'd', 'efgh'):
Guido van Rossum7131f842007-02-09 20:13:25 +0000944... print(value)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000945...
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000946a
947d
948e
949b
950f
951c
952g
953h
954
955
956>>> def maketree(iterable):
957... d = deque(iterable)
958... while len(d) > 1:
959... pair = [d.popleft(), d.popleft()]
960... d.append(pair)
961... return list(d)
962...
Guido van Rossum7131f842007-02-09 20:13:25 +0000963>>> print(maketree('abcdefgh'))
Raymond Hettingere7169eb2004-05-09 01:15:01 +0000964[[[['a', 'b'], ['c', 'd']], [['e', 'f'], ['g', 'h']]]]
965
Raymond Hettinger738ec902004-02-29 02:15:56 +0000966"""
967
968
969#==============================================================================
970
971__test__ = {'libreftest' : libreftest}
972
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000973def test_main(verbose=None):
974 import sys
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000975 test_classes = (
976 TestBasic,
977 TestVariousIteratorArgs,
978 TestSubclass,
Thomas Woutersb2137042007-02-01 18:02:27 +0000979 TestSubclassWithKwargs,
Raymond Hettinger067bbba2015-04-01 08:11:09 -0700980 TestSequence,
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000981 )
982
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000983 support.run_unittest(*test_classes)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000984
985 # verify reference counting
986 if verbose and hasattr(sys, "gettotalrefcount"):
987 import gc
988 counts = [None] * 5
Guido van Rossum805365e2007-05-07 22:24:25 +0000989 for i in range(len(counts)):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000990 support.run_unittest(*test_classes)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000991 gc.collect()
992 counts[i] = sys.gettotalrefcount()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000993 print(counts)
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000994
Raymond Hettinger738ec902004-02-29 02:15:56 +0000995 # doctests
996 from test import test_deque
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000997 support.run_doctest(test_deque, verbose)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000998
999if __name__ == "__main__":
1000 test_main(verbose=True)