blob: a15651fd7d2b54fbabb95c98236ea2357266fd45 [file] [log] [blame]
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001"""Unit tests for collections.py."""
2
Benjamin Peterson4ad6bd52010-05-21 20:55:22 +00003import unittest, doctest, operator
Raymond Hettingerf6d3e8e2011-03-23 20:33:30 -07004from test.support import TESTFN, forget, unlink
Raymond Hettinger2d32f632009-03-02 21:24:57 +00005import inspect
Benjamin Petersonee8712c2008-05-20 21:35:26 +00006from test import support
Raymond Hettinger426e0522011-01-03 02:12:02 +00007from collections import namedtuple, Counter, OrderedDict, _count_elements
Raymond Hettinger2d32f632009-03-02 21:24:57 +00008from test import mapping_tests
Georg Brandlc28e1fa2008-06-10 19:20:26 +00009import pickle, copy
Raymond Hettinger2d32f632009-03-02 21:24:57 +000010from random import randrange, shuffle
Raymond Hettinger499b2ee2009-05-27 01:53:46 +000011import keyword
12import re
R. David Murray378c0cf2010-02-24 01:46:21 +000013import sys
Yury Selivanov75445082015-05-11 22:57:16 -040014import types
Raymond Hettinger573b44c2015-05-22 16:56:32 -070015from collections import UserDict, UserString, UserList
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000016from collections import ChainMap
Raymond Hettinger32ea1652015-03-21 01:37:37 -070017from collections import deque
Yury Selivanove0104ae2015-05-14 12:19:16 -040018from collections.abc import Awaitable, Coroutine, AsyncIterator, AsyncIterable
Raymond Hettingerbd60e8d2015-05-09 01:07:23 -040019from collections.abc import Hashable, Iterable, Iterator, Generator
Raymond Hettinger57d1a882011-02-23 00:46:28 +000020from collections.abc import Sized, Container, Callable
21from collections.abc import Set, MutableSet
22from collections.abc import Mapping, MutableMapping, KeysView, ItemsView
23from collections.abc import Sequence, MutableSequence
24from collections.abc import ByteString
Guido van Rossumcd16bf62007-06-13 18:07:49 +000025
Raymond Hettinger499e1932011-02-23 07:56:53 +000026
Raymond Hettinger573b44c2015-05-22 16:56:32 -070027class TestUserObjects(unittest.TestCase):
28 def _superset_test(self, a, b):
29 self.assertGreaterEqual(
30 set(dir(a)),
31 set(dir(b)),
32 '{a} should have all the methods of {b}'.format(
33 a=a.__name__,
34 b=b.__name__,
35 ),
36 )
37 def test_str_protocol(self):
38 self._superset_test(UserString, str)
39
40 def test_list_protocol(self):
41 self._superset_test(UserList, list)
42
43 def test_dict_protocol(self):
44 self._superset_test(UserDict, dict)
45
46
Raymond Hettinger499e1932011-02-23 07:56:53 +000047################################################################################
Raymond Hettinger9fe1ccf2011-02-26 01:02:51 +000048### ChainMap (helper class for configparser and the string module)
Raymond Hettinger499e1932011-02-23 07:56:53 +000049################################################################################
50
51class TestChainMap(unittest.TestCase):
52
53 def test_basics(self):
54 c = ChainMap()
55 c['a'] = 1
56 c['b'] = 2
57 d = c.new_child()
58 d['b'] = 20
59 d['c'] = 30
60 self.assertEqual(d.maps, [{'b':20, 'c':30}, {'a':1, 'b':2}]) # check internal state
61 self.assertEqual(d.items(), dict(a=1, b=20, c=30).items()) # check items/iter/getitem
62 self.assertEqual(len(d), 3) # check len
63 for key in 'abc': # check contains
64 self.assertIn(key, d)
65 for k, v in dict(a=1, b=20, c=30, z=100).items(): # check get
66 self.assertEqual(d.get(k, 100), v)
67
68 del d['b'] # unmask a value
69 self.assertEqual(d.maps, [{'c':30}, {'a':1, 'b':2}]) # check internal state
70 self.assertEqual(d.items(), dict(a=1, b=2, c=30).items()) # check items/iter/getitem
71 self.assertEqual(len(d), 3) # check len
72 for key in 'abc': # check contains
73 self.assertIn(key, d)
74 for k, v in dict(a=1, b=2, c=30, z=100).items(): # check get
75 self.assertEqual(d.get(k, 100), v)
76 self.assertIn(repr(d), [ # check repr
77 type(d).__name__ + "({'c': 30}, {'a': 1, 'b': 2})",
78 type(d).__name__ + "({'c': 30}, {'b': 2, 'a': 1})"
79 ])
80
81 for e in d.copy(), copy.copy(d): # check shallow copies
82 self.assertEqual(d, e)
83 self.assertEqual(d.maps, e.maps)
84 self.assertIsNot(d, e)
85 self.assertIsNot(d.maps[0], e.maps[0])
86 for m1, m2 in zip(d.maps[1:], e.maps[1:]):
87 self.assertIs(m1, m2)
88
Serhiy Storchakabad12572014-12-15 14:03:42 +020089 # check deep copies
90 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
91 e = pickle.loads(pickle.dumps(d, proto))
92 self.assertEqual(d, e)
93 self.assertEqual(d.maps, e.maps)
94 self.assertIsNot(d, e)
95 for m1, m2 in zip(d.maps, e.maps):
96 self.assertIsNot(m1, m2, e)
97 for e in [copy.deepcopy(d),
Raymond Hettinger499e1932011-02-23 07:56:53 +000098 eval(repr(d))
Serhiy Storchakabad12572014-12-15 14:03:42 +020099 ]:
Raymond Hettinger499e1932011-02-23 07:56:53 +0000100 self.assertEqual(d, e)
101 self.assertEqual(d.maps, e.maps)
102 self.assertIsNot(d, e)
103 for m1, m2 in zip(d.maps, e.maps):
104 self.assertIsNot(m1, m2, e)
105
Raymond Hettingerd0321312011-02-26 06:53:58 +0000106 f = d.new_child()
107 f['b'] = 5
108 self.assertEqual(f.maps, [{'b': 5}, {'c':30}, {'a':1, 'b':2}])
109 self.assertEqual(f.parents.maps, [{'c':30}, {'a':1, 'b':2}]) # check parents
110 self.assertEqual(f['b'], 5) # find first in chain
111 self.assertEqual(f.parents['b'], 2) # look beyond maps[0]
Raymond Hettinger499e1932011-02-23 07:56:53 +0000112
113 def test_contructor(self):
Raymond Hettingerd0321312011-02-26 06:53:58 +0000114 self.assertEqual(ChainMap().maps, [{}]) # no-args --> one new dict
Raymond Hettinger499e1932011-02-23 07:56:53 +0000115 self.assertEqual(ChainMap({1:2}).maps, [{1:2}]) # 1 arg --> list
116
Raymond Hettingerd0321312011-02-26 06:53:58 +0000117 def test_bool(self):
118 self.assertFalse(ChainMap())
119 self.assertFalse(ChainMap({}, {}))
120 self.assertTrue(ChainMap({1:2}, {}))
121 self.assertTrue(ChainMap({}, {1:2}))
122
Raymond Hettinger499e1932011-02-23 07:56:53 +0000123 def test_missing(self):
124 class DefaultChainMap(ChainMap):
125 def __missing__(self, key):
126 return 999
127 d = DefaultChainMap(dict(a=1, b=2), dict(b=20, c=30))
128 for k, v in dict(a=1, b=2, c=30, d=999).items():
129 self.assertEqual(d[k], v) # check __getitem__ w/missing
130 for k, v in dict(a=1, b=2, c=30, d=77).items():
131 self.assertEqual(d.get(k, 77), v) # check get() w/ missing
132 for k, v in dict(a=True, b=True, c=True, d=False).items():
133 self.assertEqual(k in d, v) # check __contains__ w/missing
134 self.assertEqual(d.pop('a', 1001), 1, d)
135 self.assertEqual(d.pop('a', 1002), 1002) # check pop() w/missing
136 self.assertEqual(d.popitem(), ('b', 2)) # check popitem() w/missing
137 with self.assertRaises(KeyError):
138 d.popitem()
139
140 def test_dict_coercion(self):
141 d = ChainMap(dict(a=1, b=2), dict(b=20, c=30))
142 self.assertEqual(dict(d), dict(a=1, b=2, c=30))
143 self.assertEqual(dict(d.items()), dict(a=1, b=2, c=30))
144
Vinay Sajip1ba81ee2013-01-11 23:39:53 +0000145 def test_new_child(self):
146 'Tests for changes for issue #16613.'
147 c = ChainMap()
148 c['a'] = 1
149 c['b'] = 2
150 m = {'b':20, 'c': 30}
151 d = c.new_child(m)
152 self.assertEqual(d.maps, [{'b':20, 'c':30}, {'a':1, 'b':2}]) # check internal state
153 self.assertIs(m, d.maps[0])
154
155 # Use a different map than a dict
156 class lowerdict(dict):
157 def __getitem__(self, key):
158 if isinstance(key, str):
159 key = key.lower()
160 return dict.__getitem__(self, key)
161 def __contains__(self, key):
162 if isinstance(key, str):
163 key = key.lower()
164 return dict.__contains__(self, key)
165
166 c = ChainMap()
167 c['a'] = 1
168 c['b'] = 2
169 m = lowerdict(b=20, c=30)
170 d = c.new_child(m)
171 self.assertIs(m, d.maps[0])
172 for key in 'abc': # check contains
173 self.assertIn(key, d)
174 for k, v in dict(a=1, B=20, C=30, z=100).items(): # check get
175 self.assertEqual(d.get(k, 100), v)
176
Raymond Hettinger499e1932011-02-23 07:56:53 +0000177
178################################################################################
179### Named Tuples
180################################################################################
181
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000182TestNT = namedtuple('TestNT', 'x y z') # type used for pickle tests
Guido van Rossumd8faa362007-04-27 19:54:29 +0000183
184class TestNamedTuple(unittest.TestCase):
185
186 def test_factory(self):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000187 Point = namedtuple('Point', 'x y')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000188 self.assertEqual(Point.__name__, 'Point')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000189 self.assertEqual(Point.__slots__, ())
190 self.assertEqual(Point.__module__, __name__)
191 self.assertEqual(Point.__getitem__, tuple.__getitem__)
Christian Heimesfaf2f632008-01-06 16:59:19 +0000192 self.assertEqual(Point._fields, ('x', 'y'))
Raymond Hettinger2ebea412011-03-23 12:52:23 -0700193 self.assertIn('class Point(tuple)', Point._source)
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000194
195 self.assertRaises(ValueError, namedtuple, 'abc%', 'efg ghi') # type has non-alpha char
196 self.assertRaises(ValueError, namedtuple, 'class', 'efg ghi') # type has keyword
197 self.assertRaises(ValueError, namedtuple, '9abc', 'efg ghi') # type starts with digit
198
199 self.assertRaises(ValueError, namedtuple, 'abc', 'efg g%hi') # field with non-alpha char
200 self.assertRaises(ValueError, namedtuple, 'abc', 'abc class') # field has keyword
201 self.assertRaises(ValueError, namedtuple, 'abc', '8efg 9ghi') # field starts with digit
Christian Heimes0449f632007-12-15 01:27:15 +0000202 self.assertRaises(ValueError, namedtuple, 'abc', '_efg ghi') # field with leading underscore
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000203 self.assertRaises(ValueError, namedtuple, 'abc', 'efg efg ghi') # duplicate field
204
205 namedtuple('Point0', 'x1 y2') # Verify that numbers are allowed in names
Christian Heimes0449f632007-12-15 01:27:15 +0000206 namedtuple('_', 'a b c') # Test leading underscores in a typename
Guido van Rossumd8faa362007-04-27 19:54:29 +0000207
Benjamin Petersone9bbc8b2008-09-28 02:06:32 +0000208 nt = namedtuple('nt', 'the quick brown fox') # check unicode input
Benjamin Peterson577473f2010-01-19 00:09:57 +0000209 self.assertNotIn("u'", repr(nt._fields))
Benjamin Petersone9bbc8b2008-09-28 02:06:32 +0000210 nt = namedtuple('nt', ('the', 'quick')) # check unicode input
Benjamin Peterson577473f2010-01-19 00:09:57 +0000211 self.assertNotIn("u'", repr(nt._fields))
Benjamin Petersone9bbc8b2008-09-28 02:06:32 +0000212
Christian Heimesfaf2f632008-01-06 16:59:19 +0000213 self.assertRaises(TypeError, Point._make, [11]) # catch too few args
214 self.assertRaises(TypeError, Point._make, [11, 22, 33]) # catch too many args
215
R. David Murray378c0cf2010-02-24 01:46:21 +0000216 @unittest.skipIf(sys.flags.optimize >= 2,
217 "Docstrings are omitted with -O2 and above")
218 def test_factory_doc_attr(self):
219 Point = namedtuple('Point', 'x y')
220 self.assertEqual(Point.__doc__, 'Point(x, y)')
221
Raymond Hettingereac503a2015-05-13 01:09:59 -0700222 @unittest.skipIf(sys.flags.optimize >= 2,
223 "Docstrings are omitted with -O2 and above")
224 def test_doc_writable(self):
225 Point = namedtuple('Point', 'x y')
226 self.assertEqual(Point.x.__doc__, 'Alias for field number 0')
227 Point.x.__doc__ = 'docstring for Point.x'
228 self.assertEqual(Point.x.__doc__, 'docstring for Point.x')
229
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000230 def test_name_fixer(self):
231 for spec, renamed in [
Raymond Hettinger56145242009-04-02 22:31:59 +0000232 [('efg', 'g%hi'), ('efg', '_1')], # field with non-alpha char
233 [('abc', 'class'), ('abc', '_1')], # field has keyword
234 [('8efg', '9ghi'), ('_0', '_1')], # field starts with digit
235 [('abc', '_efg'), ('abc', '_1')], # field with leading underscore
236 [('abc', 'efg', 'efg', 'ghi'), ('abc', 'efg', '_2', 'ghi')], # duplicate field
237 [('abc', '', 'x'), ('abc', '_1', 'x')], # fieldname is a space
Benjamin Petersona86f2c02009-02-10 02:41:10 +0000238 ]:
239 self.assertEqual(namedtuple('NT', spec, rename=True)._fields, renamed)
240
Guido van Rossumd8faa362007-04-27 19:54:29 +0000241 def test_instance(self):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000242 Point = namedtuple('Point', 'x y')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000243 p = Point(11, 22)
244 self.assertEqual(p, Point(x=11, y=22))
245 self.assertEqual(p, Point(11, y=22))
246 self.assertEqual(p, Point(y=22, x=11))
247 self.assertEqual(p, Point(*(11, 22)))
248 self.assertEqual(p, Point(**dict(x=11, y=22)))
249 self.assertRaises(TypeError, Point, 1) # too few args
250 self.assertRaises(TypeError, Point, 1, 2, 3) # too many args
251 self.assertRaises(TypeError, eval, 'Point(XXX=1, y=2)', locals()) # wrong keyword argument
252 self.assertRaises(TypeError, eval, 'Point(x=1)', locals()) # missing keyword argument
253 self.assertEqual(repr(p), 'Point(x=11, y=22)')
Benjamin Peterson577473f2010-01-19 00:09:57 +0000254 self.assertNotIn('__weakref__', dir(p))
Christian Heimesfaf2f632008-01-06 16:59:19 +0000255 self.assertEqual(p, Point._make([11, 22])) # test _make classmethod
Christian Heimes0449f632007-12-15 01:27:15 +0000256 self.assertEqual(p._fields, ('x', 'y')) # test _fields attribute
257 self.assertEqual(p._replace(x=1), (1, 22)) # test _replace method
Raymond Hettinger163e9822013-05-18 00:05:20 -0700258 self.assertEqual(p._asdict(), dict(x=11, y=22)) # test _asdict method
259 self.assertEqual(vars(p), p._asdict()) # verify that vars() works
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000260
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000261 try:
Christian Heimesfaf2f632008-01-06 16:59:19 +0000262 p._replace(x=1, error=2)
263 except ValueError:
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000264 pass
265 else:
Christian Heimesfaf2f632008-01-06 16:59:19 +0000266 self._fail('Did not detect an incorrect fieldname')
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000267
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000268 # verify that field string can have commas
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000269 Point = namedtuple('Point', 'x, y')
270 p = Point(x=11, y=22)
271 self.assertEqual(repr(p), 'Point(x=11, y=22)')
272
273 # verify that fieldspec can be a non-string sequence
274 Point = namedtuple('Point', ('x', 'y'))
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000275 p = Point(x=11, y=22)
276 self.assertEqual(repr(p), 'Point(x=11, y=22)')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000277
278 def test_tupleness(self):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000279 Point = namedtuple('Point', 'x y')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000280 p = Point(11, 22)
281
Ezio Melottie9615932010-01-24 19:26:24 +0000282 self.assertIsInstance(p, tuple)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000283 self.assertEqual(p, (11, 22)) # matches a real tuple
284 self.assertEqual(tuple(p), (11, 22)) # coercable to a real tuple
285 self.assertEqual(list(p), [11, 22]) # coercable to a list
286 self.assertEqual(max(p), 22) # iterable
287 self.assertEqual(max(*p), 22) # star-able
288 x, y = p
289 self.assertEqual(p, (x, y)) # unpacks like a tuple
290 self.assertEqual((p[0], p[1]), (11, 22)) # indexable like a tuple
291 self.assertRaises(IndexError, p.__getitem__, 3)
292
293 self.assertEqual(p.x, x)
294 self.assertEqual(p.y, y)
295 self.assertRaises(AttributeError, eval, 'p.z', locals())
296
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000297 def test_odd_sizes(self):
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000298 Zero = namedtuple('Zero', '')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000299 self.assertEqual(Zero(), ())
Christian Heimesfaf2f632008-01-06 16:59:19 +0000300 self.assertEqual(Zero._make([]), ())
Christian Heimes99170a52007-12-19 02:07:34 +0000301 self.assertEqual(repr(Zero()), 'Zero()')
302 self.assertEqual(Zero()._asdict(), {})
303 self.assertEqual(Zero()._fields, ())
304
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000305 Dot = namedtuple('Dot', 'd')
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000306 self.assertEqual(Dot(1), (1,))
Christian Heimesfaf2f632008-01-06 16:59:19 +0000307 self.assertEqual(Dot._make([1]), (1,))
Christian Heimes99170a52007-12-19 02:07:34 +0000308 self.assertEqual(Dot(1).d, 1)
309 self.assertEqual(repr(Dot(1)), 'Dot(d=1)')
310 self.assertEqual(Dot(1)._asdict(), {'d':1})
311 self.assertEqual(Dot(1)._replace(d=999), (999,))
312 self.assertEqual(Dot(1)._fields, ('d',))
Thomas Wouters1b7f8912007-09-19 03:06:30 +0000313
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000314 # n = 5000
Christian Heimes99170a52007-12-19 02:07:34 +0000315 n = 254 # SyntaxError: more than 255 arguments:
316 import string, random
Georg Brandlb533e262008-05-25 18:19:30 +0000317 names = list(set(''.join([random.choice(string.ascii_letters)
318 for j in range(10)]) for i in range(n)))
319 n = len(names)
Christian Heimes99170a52007-12-19 02:07:34 +0000320 Big = namedtuple('Big', names)
321 b = Big(*range(n))
322 self.assertEqual(b, tuple(range(n)))
Christian Heimesfaf2f632008-01-06 16:59:19 +0000323 self.assertEqual(Big._make(range(n)), tuple(range(n)))
Christian Heimes99170a52007-12-19 02:07:34 +0000324 for pos, name in enumerate(names):
325 self.assertEqual(getattr(b, name), pos)
326 repr(b) # make sure repr() doesn't blow-up
327 d = b._asdict()
328 d_expected = dict(zip(names, range(n)))
329 self.assertEqual(d, d_expected)
330 b2 = b._replace(**dict([(names[1], 999),(names[-5], 42)]))
331 b2_expected = list(range(n))
332 b2_expected[1] = 999
333 b2_expected[-5] = 42
334 self.assertEqual(b2, tuple(b2_expected))
335 self.assertEqual(b._fields, tuple(names))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000336
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000337 def test_pickle(self):
338 p = TestNT(x=10, y=20, z=30)
339 for module in (pickle,):
340 loads = getattr(module, 'loads')
341 dumps = getattr(module, 'dumps')
Eric V. Smith4d5d69d2014-02-05 10:33:14 -0500342 for protocol in range(-1, module.HIGHEST_PROTOCOL + 1):
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000343 q = loads(dumps(p, protocol))
344 self.assertEqual(p, q)
345 self.assertEqual(p._fields, q._fields)
Raymond Hettingerb98dcc12013-05-03 02:24:15 -0700346 self.assertNotIn(b'OrderedDict', dumps(p, protocol))
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000347
348 def test_copy(self):
349 p = TestNT(x=10, y=20, z=30)
350 for copier in copy.copy, copy.deepcopy:
351 q = copier(p)
352 self.assertEqual(p, q)
353 self.assertEqual(p._fields, q._fields)
354
Raymond Hettinger089ba7f2009-05-27 00:38:24 +0000355 def test_name_conflicts(self):
356 # Some names like "self", "cls", "tuple", "itemgetter", and "property"
357 # failed when used as field names. Test to make sure these now work.
358 T = namedtuple('T', 'itemgetter property self cls tuple')
359 t = T(1, 2, 3, 4, 5)
360 self.assertEqual(t, (1,2,3,4,5))
361 newt = t._replace(itemgetter=10, property=20, self=30, cls=40, tuple=50)
362 self.assertEqual(newt, (10,20,30,40,50))
363
Raymond Hettinger499b2ee2009-05-27 01:53:46 +0000364 # Broader test of all interesting names in a template
365 with support.captured_stdout() as template:
366 T = namedtuple('T', 'x', verbose=True)
367 words = set(re.findall('[A-Za-z]+', template.getvalue()))
368 words -= set(keyword.kwlist)
369 T = namedtuple('T', words)
370 # test __new__
371 values = tuple(range(len(words)))
372 t = T(*values)
373 self.assertEqual(t, values)
374 t = T(**dict(zip(T._fields, values)))
375 self.assertEqual(t, values)
376 # test _make
377 t = T._make(values)
378 self.assertEqual(t, values)
379 # exercise __repr__
380 repr(t)
381 # test _asdict
382 self.assertEqual(t._asdict(), dict(zip(T._fields, values)))
383 # test _replace
384 t = T._make(values)
385 newvalues = tuple(v*10 for v in values)
386 newt = t._replace(**dict(zip(T._fields, newvalues)))
387 self.assertEqual(newt, newvalues)
388 # test _fields
389 self.assertEqual(T._fields, tuple(words))
390 # test __getnewargs__
391 self.assertEqual(t.__getnewargs__(), values)
392
Raymond Hettingerd331ce92010-08-08 01:13:42 +0000393 def test_repr(self):
394 with support.captured_stdout() as template:
395 A = namedtuple('A', 'x', verbose=True)
396 self.assertEqual(repr(A(1)), 'A(x=1)')
397 # repr should show the name of the subclass
398 class B(A):
399 pass
400 self.assertEqual(repr(B(1)), 'B(x=1)')
401
Raymond Hettingerf6d3e8e2011-03-23 20:33:30 -0700402 def test_source(self):
403 # verify that _source can be run through exec()
Raymond Hettingerd4652fa2011-03-24 09:45:43 -0700404 tmp = namedtuple('NTColor', 'red green blue')
405 globals().pop('NTColor', None) # remove artifacts from other tests
Raymond Hettingerf6d3e8e2011-03-23 20:33:30 -0700406 exec(tmp._source, globals())
Raymond Hettingerd4652fa2011-03-24 09:45:43 -0700407 self.assertIn('NTColor', globals())
408 c = NTColor(10, 20, 30)
Raymond Hettingerf6d3e8e2011-03-23 20:33:30 -0700409 self.assertEqual((c.red, c.green, c.blue), (10, 20, 30))
Raymond Hettingerd4652fa2011-03-24 09:45:43 -0700410 self.assertEqual(NTColor._fields, ('red', 'green', 'blue'))
411 globals().pop('NTColor', None) # clean-up after this test
Raymond Hettingerf6d3e8e2011-03-23 20:33:30 -0700412
Raymond Hettingerd331ce92010-08-08 01:13:42 +0000413
Raymond Hettinger499e1932011-02-23 07:56:53 +0000414################################################################################
415### Abstract Base Classes
416################################################################################
417
Raymond Hettingerae650182009-01-28 23:33:59 +0000418class ABCTestCase(unittest.TestCase):
419
420 def validate_abstract_methods(self, abc, *names):
421 methodstubs = dict.fromkeys(names, lambda s, *args: 0)
422
423 # everything should work will all required methods are present
424 C = type('C', (abc,), methodstubs)
425 C()
426
427 # instantiation should fail if a required method is missing
428 for name in names:
429 stubs = methodstubs.copy()
430 del stubs[name]
431 C = type('C', (abc,), stubs)
432 self.assertRaises(TypeError, C, name)
433
Florent Xiclunace153f62010-03-08 15:34:35 +0000434 def validate_isinstance(self, abc, name):
435 stub = lambda s, *args: 0
436
437 C = type('C', (object,), {'__hash__': None})
438 setattr(C, name, stub)
439 self.assertIsInstance(C(), abc)
440 self.assertTrue(issubclass(C, abc))
441
442 C = type('C', (object,), {'__hash__': None})
443 self.assertNotIsInstance(C(), abc)
444 self.assertFalse(issubclass(C, abc))
Raymond Hettingerae650182009-01-28 23:33:59 +0000445
Benjamin Peterson4ad6bd52010-05-21 20:55:22 +0000446 def validate_comparison(self, instance):
447 ops = ['lt', 'gt', 'le', 'ge', 'ne', 'or', 'and', 'xor', 'sub']
448 operators = {}
449 for op in ops:
450 name = '__' + op + '__'
451 operators[name] = getattr(operator, name)
452
453 class Other:
454 def __init__(self):
455 self.right_side = False
456 def __eq__(self, other):
457 self.right_side = True
458 return True
459 __lt__ = __eq__
460 __gt__ = __eq__
461 __le__ = __eq__
462 __ge__ = __eq__
463 __ne__ = __eq__
464 __ror__ = __eq__
465 __rand__ = __eq__
466 __rxor__ = __eq__
467 __rsub__ = __eq__
468
469 for name, op in operators.items():
470 if not hasattr(instance, name):
471 continue
472 other = Other()
473 op(instance, other)
474 self.assertTrue(other.right_side,'Right side not called for %s.%s'
475 % (type(instance), name))
476
Raymond Hettingerae650182009-01-28 23:33:59 +0000477class TestOneTrickPonyABCs(ABCTestCase):
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000478
Yury Selivanov75445082015-05-11 22:57:16 -0400479 def test_Awaitable(self):
480 def gen():
481 yield
482
483 @types.coroutine
484 def coro():
485 yield
486
487 async def new_coro():
488 pass
489
490 class Bar:
491 def __await__(self):
492 yield
493
494 class MinimalCoro(Coroutine):
495 def send(self, value):
496 return value
497 def throw(self, typ, val=None, tb=None):
498 super().throw(typ, val, tb)
499
500 non_samples = [None, int(), gen(), object()]
501 for x in non_samples:
502 self.assertNotIsInstance(x, Awaitable)
503 self.assertFalse(issubclass(type(x), Awaitable), repr(type(x)))
504
505 samples = [Bar(), MinimalCoro()]
506 for x in samples:
507 self.assertIsInstance(x, Awaitable)
508 self.assertTrue(issubclass(type(x), Awaitable))
509
510 c = coro()
511 self.assertIsInstance(c, Awaitable)
512 c.close() # awoid RuntimeWarning that coro() was not awaited
513
514 c = new_coro()
515 self.assertIsInstance(c, Awaitable)
516 c.close() # awoid RuntimeWarning that coro() was not awaited
517
Yury Selivanovaded55c2015-05-13 23:41:55 -0400518 class CoroLike:
519 def send(self, value):
520 pass
521 def throw(self, typ, val=None, tb=None):
522 pass
523 def close(self):
524 pass
525 Coroutine.register(CoroLike)
Yury Selivanov08e53002015-05-13 23:57:59 -0400526 self.assertTrue(isinstance(CoroLike(), Awaitable))
527 self.assertTrue(issubclass(CoroLike, Awaitable))
528 CoroLike = None
529 support.gc_collect() # Kill CoroLike to clean-up ABCMeta cache
Yury Selivanovaded55c2015-05-13 23:41:55 -0400530
Yury Selivanov75445082015-05-11 22:57:16 -0400531 def test_Coroutine(self):
532 def gen():
533 yield
534
535 @types.coroutine
536 def coro():
537 yield
538
539 async def new_coro():
540 pass
541
542 class Bar:
543 def __await__(self):
544 yield
545
546 class MinimalCoro(Coroutine):
547 def send(self, value):
548 return value
549 def throw(self, typ, val=None, tb=None):
550 super().throw(typ, val, tb)
551
552 non_samples = [None, int(), gen(), object(), Bar()]
553 for x in non_samples:
554 self.assertNotIsInstance(x, Coroutine)
555 self.assertFalse(issubclass(type(x), Coroutine), repr(type(x)))
556
557 samples = [MinimalCoro()]
558 for x in samples:
559 self.assertIsInstance(x, Awaitable)
560 self.assertTrue(issubclass(type(x), Awaitable))
561
562 c = coro()
563 self.assertIsInstance(c, Coroutine)
564 c.close() # awoid RuntimeWarning that coro() was not awaited
565
566 c = new_coro()
567 self.assertIsInstance(c, Coroutine)
568 c.close() # awoid RuntimeWarning that coro() was not awaited
569
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000570 def test_Hashable(self):
571 # Check some non-hashables
Guido van Rossum254348e2007-11-21 19:29:53 +0000572 non_samples = [bytearray(), list(), set(), dict()]
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000573 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000574 self.assertNotIsInstance(x, Hashable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000575 self.assertFalse(issubclass(type(x), Hashable), repr(type(x)))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000576 # Check some hashables
577 samples = [None,
578 int(), float(), complex(),
Guido van Rossum07d4e782007-07-03 16:59:47 +0000579 str(),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000580 tuple(), frozenset(),
Guido van Rossum98297ee2007-11-06 21:34:58 +0000581 int, list, object, type, bytes()
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000582 ]
583 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000584 self.assertIsInstance(x, Hashable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000585 self.assertTrue(issubclass(type(x), Hashable), repr(type(x)))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000586 self.assertRaises(TypeError, Hashable)
587 # Check direct subclassing
588 class H(Hashable):
589 def __hash__(self):
590 return super().__hash__()
591 self.assertEqual(hash(H()), 0)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000592 self.assertFalse(issubclass(int, H))
Raymond Hettingerae650182009-01-28 23:33:59 +0000593 self.validate_abstract_methods(Hashable, '__hash__')
Florent Xiclunace153f62010-03-08 15:34:35 +0000594 self.validate_isinstance(Hashable, '__hash__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000595
Yury Selivanove0104ae2015-05-14 12:19:16 -0400596 def test_AsyncIterable(self):
597 class AI:
598 async def __aiter__(self):
599 return self
600 self.assertTrue(isinstance(AI(), AsyncIterable))
601 self.assertTrue(issubclass(AI, AsyncIterable))
602 # Check some non-iterables
603 non_samples = [None, object, []]
604 for x in non_samples:
605 self.assertNotIsInstance(x, AsyncIterable)
606 self.assertFalse(issubclass(type(x), AsyncIterable), repr(type(x)))
607 self.validate_abstract_methods(AsyncIterable, '__aiter__')
608 self.validate_isinstance(AsyncIterable, '__aiter__')
609
610 def test_AsyncIterator(self):
611 class AI:
612 async def __aiter__(self):
613 return self
614 async def __anext__(self):
615 raise StopAsyncIteration
616 self.assertTrue(isinstance(AI(), AsyncIterator))
617 self.assertTrue(issubclass(AI, AsyncIterator))
618 non_samples = [None, object, []]
619 # Check some non-iterables
620 for x in non_samples:
621 self.assertNotIsInstance(x, AsyncIterator)
622 self.assertFalse(issubclass(type(x), AsyncIterator), repr(type(x)))
623 # Similarly to regular iterators (see issue 10565)
624 class AnextOnly:
625 async def __anext__(self):
626 raise StopAsyncIteration
627 self.assertNotIsInstance(AnextOnly(), AsyncIterator)
628 self.validate_abstract_methods(AsyncIterator, '__anext__', '__aiter__')
629
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000630 def test_Iterable(self):
631 # Check some non-iterables
632 non_samples = [None, 42, 3.14, 1j]
633 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000634 self.assertNotIsInstance(x, Iterable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000635 self.assertFalse(issubclass(type(x), Iterable), repr(type(x)))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000636 # Check some iterables
Guido van Rossum07d4e782007-07-03 16:59:47 +0000637 samples = [bytes(), str(),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000638 tuple(), list(), set(), frozenset(), dict(),
639 dict().keys(), dict().items(), dict().values(),
640 (lambda: (yield))(),
641 (x for x in []),
642 ]
643 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000644 self.assertIsInstance(x, Iterable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000645 self.assertTrue(issubclass(type(x), Iterable), repr(type(x)))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000646 # Check direct subclassing
647 class I(Iterable):
648 def __iter__(self):
649 return super().__iter__()
650 self.assertEqual(list(I()), [])
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000651 self.assertFalse(issubclass(str, I))
Raymond Hettingerae650182009-01-28 23:33:59 +0000652 self.validate_abstract_methods(Iterable, '__iter__')
Florent Xiclunace153f62010-03-08 15:34:35 +0000653 self.validate_isinstance(Iterable, '__iter__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000654
655 def test_Iterator(self):
Guido van Rossum07d4e782007-07-03 16:59:47 +0000656 non_samples = [None, 42, 3.14, 1j, b"", "", (), [], {}, set()]
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000657 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000658 self.assertNotIsInstance(x, Iterator)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000659 self.assertFalse(issubclass(type(x), Iterator), repr(type(x)))
Guido van Rossum07d4e782007-07-03 16:59:47 +0000660 samples = [iter(bytes()), iter(str()),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000661 iter(tuple()), iter(list()), iter(dict()),
662 iter(set()), iter(frozenset()),
663 iter(dict().keys()), iter(dict().items()),
664 iter(dict().values()),
665 (lambda: (yield))(),
666 (x for x in []),
667 ]
668 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000669 self.assertIsInstance(x, Iterator)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000670 self.assertTrue(issubclass(type(x), Iterator), repr(type(x)))
Raymond Hettingeread22222010-11-29 03:56:12 +0000671 self.validate_abstract_methods(Iterator, '__next__', '__iter__')
672
673 # Issue 10565
674 class NextOnly:
675 def __next__(self):
676 yield 1
Raymond Hettingerbb6c0aa2014-11-22 22:14:41 -0800677 return
Raymond Hettingeread22222010-11-29 03:56:12 +0000678 self.assertNotIsInstance(NextOnly(), Iterator)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000679
Raymond Hettingerbd60e8d2015-05-09 01:07:23 -0400680 def test_Generator(self):
681 class NonGen1:
682 def __iter__(self): return self
683 def __next__(self): return None
684 def close(self): pass
685 def throw(self, typ, val=None, tb=None): pass
686
687 class NonGen2:
688 def __iter__(self): return self
689 def __next__(self): return None
690 def close(self): pass
691 def send(self, value): return value
692
693 class NonGen3:
694 def close(self): pass
695 def send(self, value): return value
696 def throw(self, typ, val=None, tb=None): pass
697
698 non_samples = [
699 None, 42, 3.14, 1j, b"", "", (), [], {}, set(),
700 iter(()), iter([]), NonGen1(), NonGen2(), NonGen3()]
701 for x in non_samples:
702 self.assertNotIsInstance(x, Generator)
703 self.assertFalse(issubclass(type(x), Generator), repr(type(x)))
704
705 class Gen:
706 def __iter__(self): return self
707 def __next__(self): return None
708 def close(self): pass
709 def send(self, value): return value
710 def throw(self, typ, val=None, tb=None): pass
711
712 class MinimalGen(Generator):
713 def send(self, value):
714 return value
715 def throw(self, typ, val=None, tb=None):
716 super().throw(typ, val, tb)
717
718 def gen():
719 yield 1
720
721 samples = [gen(), (lambda: (yield))(), Gen(), MinimalGen()]
722 for x in samples:
723 self.assertIsInstance(x, Iterator)
724 self.assertIsInstance(x, Generator)
725 self.assertTrue(issubclass(type(x), Generator), repr(type(x)))
726 self.validate_abstract_methods(Generator, 'send', 'throw')
727
728 # mixin tests
729 mgen = MinimalGen()
730 self.assertIs(mgen, iter(mgen))
731 self.assertIs(mgen.send(None), next(mgen))
732 self.assertEqual(2, mgen.send(2))
733 self.assertIsNone(mgen.close())
734 self.assertRaises(ValueError, mgen.throw, ValueError)
735 self.assertRaisesRegex(ValueError, "^huhu$",
736 mgen.throw, ValueError, ValueError("huhu"))
737 self.assertRaises(StopIteration, mgen.throw, StopIteration())
738
739 class FailOnClose(Generator):
740 def send(self, value): return value
741 def throw(self, *args): raise ValueError
742
743 self.assertRaises(ValueError, FailOnClose().close)
744
745 class IgnoreGeneratorExit(Generator):
746 def send(self, value): return value
747 def throw(self, *args): pass
748
749 self.assertRaises(RuntimeError, IgnoreGeneratorExit().close)
750
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000751 def test_Sized(self):
752 non_samples = [None, 42, 3.14, 1j,
753 (lambda: (yield))(),
754 (x for x in []),
755 ]
756 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000757 self.assertNotIsInstance(x, Sized)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000758 self.assertFalse(issubclass(type(x), Sized), repr(type(x)))
Guido van Rossum07d4e782007-07-03 16:59:47 +0000759 samples = [bytes(), str(),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000760 tuple(), list(), set(), frozenset(), dict(),
761 dict().keys(), dict().items(), dict().values(),
762 ]
763 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000764 self.assertIsInstance(x, Sized)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000765 self.assertTrue(issubclass(type(x), Sized), repr(type(x)))
Raymond Hettingerae650182009-01-28 23:33:59 +0000766 self.validate_abstract_methods(Sized, '__len__')
Florent Xiclunace153f62010-03-08 15:34:35 +0000767 self.validate_isinstance(Sized, '__len__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000768
769 def test_Container(self):
770 non_samples = [None, 42, 3.14, 1j,
771 (lambda: (yield))(),
772 (x for x in []),
773 ]
774 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000775 self.assertNotIsInstance(x, Container)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000776 self.assertFalse(issubclass(type(x), Container), repr(type(x)))
Guido van Rossum07d4e782007-07-03 16:59:47 +0000777 samples = [bytes(), str(),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000778 tuple(), list(), set(), frozenset(), dict(),
779 dict().keys(), dict().items(),
780 ]
781 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000782 self.assertIsInstance(x, Container)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000783 self.assertTrue(issubclass(type(x), Container), repr(type(x)))
Raymond Hettingerae650182009-01-28 23:33:59 +0000784 self.validate_abstract_methods(Container, '__contains__')
Florent Xiclunace153f62010-03-08 15:34:35 +0000785 self.validate_isinstance(Container, '__contains__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000786
787 def test_Callable(self):
788 non_samples = [None, 42, 3.14, 1j,
789 "", b"", (), [], {}, set(),
790 (lambda: (yield))(),
791 (x for x in []),
792 ]
793 for x in non_samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000794 self.assertNotIsInstance(x, Callable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000795 self.assertFalse(issubclass(type(x), Callable), repr(type(x)))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000796 samples = [lambda: None,
797 type, int, object,
798 len,
799 list.append, [].append,
800 ]
801 for x in samples:
Ezio Melottie9615932010-01-24 19:26:24 +0000802 self.assertIsInstance(x, Callable)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000803 self.assertTrue(issubclass(type(x), Callable), repr(type(x)))
Raymond Hettingerae650182009-01-28 23:33:59 +0000804 self.validate_abstract_methods(Callable, '__call__')
Florent Xiclunace153f62010-03-08 15:34:35 +0000805 self.validate_isinstance(Callable, '__call__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000806
807 def test_direct_subclassing(self):
808 for B in Hashable, Iterable, Iterator, Sized, Container, Callable:
809 class C(B):
810 pass
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000811 self.assertTrue(issubclass(C, B))
812 self.assertFalse(issubclass(int, C))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000813
814 def test_registration(self):
815 for B in Hashable, Iterable, Iterator, Sized, Container, Callable:
816 class C:
817 __hash__ = None # Make sure it isn't hashable by default
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000818 self.assertFalse(issubclass(C, B), B.__name__)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000819 B.register(C)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000820 self.assertTrue(issubclass(C, B))
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000821
Raymond Hettinger3f10a952009-04-01 19:05:50 +0000822class WithSet(MutableSet):
823
824 def __init__(self, it=()):
825 self.data = set(it)
826
827 def __len__(self):
828 return len(self.data)
829
830 def __iter__(self):
831 return iter(self.data)
832
833 def __contains__(self, item):
834 return item in self.data
835
836 def add(self, item):
837 self.data.add(item)
838
839 def discard(self, item):
840 self.data.discard(item)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000841
Raymond Hettingerae650182009-01-28 23:33:59 +0000842class TestCollectionABCs(ABCTestCase):
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000843
844 # XXX For now, we only test some virtual inheritance properties.
845 # We should also test the proper behavior of the collection ABCs
846 # as real base classes or mix-in classes.
847
848 def test_Set(self):
849 for sample in [set, frozenset]:
Ezio Melottie9615932010-01-24 19:26:24 +0000850 self.assertIsInstance(sample(), Set)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000851 self.assertTrue(issubclass(sample, Set))
Raymond Hettingerae650182009-01-28 23:33:59 +0000852 self.validate_abstract_methods(Set, '__contains__', '__iter__', '__len__')
Benjamin Peterson4ad6bd52010-05-21 20:55:22 +0000853 class MySet(Set):
854 def __contains__(self, x):
855 return False
856 def __len__(self):
857 return 0
858 def __iter__(self):
859 return iter([])
860 self.validate_comparison(MySet())
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000861
Benjamin Peterson41181742008-07-02 20:22:54 +0000862 def test_hash_Set(self):
863 class OneTwoThreeSet(Set):
864 def __init__(self):
865 self.contents = [1, 2, 3]
866 def __contains__(self, x):
867 return x in self.contents
868 def __len__(self):
869 return len(self.contents)
870 def __iter__(self):
871 return iter(self.contents)
872 def __hash__(self):
873 return self._hash()
874 a, b = OneTwoThreeSet(), OneTwoThreeSet()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000875 self.assertTrue(hash(a) == hash(b))
Benjamin Peterson41181742008-07-02 20:22:54 +0000876
Raymond Hettinger2d452ee2014-05-25 18:28:39 -0700877 def test_isdisjoint_Set(self):
878 class MySet(Set):
879 def __init__(self, itr):
880 self.contents = itr
881 def __contains__(self, x):
882 return x in self.contents
883 def __iter__(self):
884 return iter(self.contents)
885 def __len__(self):
886 return len([x for x in self.contents])
887 s1 = MySet((1, 2, 3))
888 s2 = MySet((4, 5, 6))
889 s3 = MySet((1, 5, 6))
890 self.assertTrue(s1.isdisjoint(s2))
891 self.assertFalse(s1.isdisjoint(s3))
892
893 def test_equality_Set(self):
894 class MySet(Set):
895 def __init__(self, itr):
896 self.contents = itr
897 def __contains__(self, x):
898 return x in self.contents
899 def __iter__(self):
900 return iter(self.contents)
901 def __len__(self):
902 return len([x for x in self.contents])
903 s1 = MySet((1,))
904 s2 = MySet((1, 2))
905 s3 = MySet((3, 4))
906 s4 = MySet((3, 4))
907 self.assertTrue(s2 > s1)
908 self.assertTrue(s1 < s2)
909 self.assertFalse(s2 <= s1)
910 self.assertFalse(s2 <= s3)
911 self.assertFalse(s1 >= s2)
912 self.assertEqual(s3, s4)
913 self.assertNotEqual(s2, s3)
914
915 def test_arithmetic_Set(self):
916 class MySet(Set):
917 def __init__(self, itr):
918 self.contents = itr
919 def __contains__(self, x):
920 return x in self.contents
921 def __iter__(self):
922 return iter(self.contents)
923 def __len__(self):
924 return len([x for x in self.contents])
925 s1 = MySet((1, 2, 3))
926 s2 = MySet((3, 4, 5))
927 s3 = s1 & s2
928 self.assertEqual(s3, MySet((3,)))
929
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000930 def test_MutableSet(self):
Ezio Melottie9615932010-01-24 19:26:24 +0000931 self.assertIsInstance(set(), MutableSet)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000932 self.assertTrue(issubclass(set, MutableSet))
Ezio Melottie9615932010-01-24 19:26:24 +0000933 self.assertNotIsInstance(frozenset(), MutableSet)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000934 self.assertFalse(issubclass(frozenset, MutableSet))
Raymond Hettingerae650182009-01-28 23:33:59 +0000935 self.validate_abstract_methods(MutableSet, '__contains__', '__iter__', '__len__',
936 'add', 'discard')
937
Raymond Hettinger3f10a952009-04-01 19:05:50 +0000938 def test_issue_5647(self):
939 # MutableSet.__iand__ mutated the set during iteration
940 s = WithSet('abcd')
941 s &= WithSet('cdef') # This used to fail
942 self.assertEqual(set(s), set('cd'))
943
Raymond Hettingerae650182009-01-28 23:33:59 +0000944 def test_issue_4920(self):
945 # MutableSet.pop() method did not work
Raymond Hettinger57d1a882011-02-23 00:46:28 +0000946 class MySet(MutableSet):
Raymond Hettingerae650182009-01-28 23:33:59 +0000947 __slots__=['__s']
948 def __init__(self,items=None):
949 if items is None:
950 items=[]
951 self.__s=set(items)
952 def __contains__(self,v):
953 return v in self.__s
954 def __iter__(self):
955 return iter(self.__s)
956 def __len__(self):
957 return len(self.__s)
958 def add(self,v):
959 result=v not in self.__s
960 self.__s.add(v)
961 return result
962 def discard(self,v):
963 result=v in self.__s
964 self.__s.discard(v)
965 return result
966 def __repr__(self):
967 return "MySet(%s)" % repr(list(self))
968 s = MySet([5,43,2,1])
969 self.assertEqual(s.pop(), 1)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000970
Daniel Stutzbach31da5b22010-08-24 20:49:57 +0000971 def test_issue8750(self):
972 empty = WithSet()
973 full = WithSet(range(10))
974 s = WithSet(full)
975 s -= s
976 self.assertEqual(s, empty)
977 s = WithSet(full)
978 s ^= s
979 self.assertEqual(s, empty)
980 s = WithSet(full)
981 s &= s
982 self.assertEqual(s, full)
983 s |= s
984 self.assertEqual(s, full)
985
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +0200986 def test_issue16373(self):
987 # Recursion error comparing comparable and noncomparable
988 # Set instances
989 class MyComparableSet(Set):
990 def __contains__(self, x):
991 return False
992 def __len__(self):
993 return 0
994 def __iter__(self):
995 return iter([])
996 class MyNonComparableSet(Set):
997 def __contains__(self, x):
998 return False
999 def __len__(self):
1000 return 0
1001 def __iter__(self):
1002 return iter([])
1003 def __le__(self, x):
1004 return NotImplemented
1005 def __lt__(self, x):
1006 return NotImplemented
1007
1008 cs = MyComparableSet()
1009 ncs = MyNonComparableSet()
Raymond Hettingerdd5e53a2014-05-26 00:09:04 -07001010 self.assertFalse(ncs < cs)
1011 self.assertTrue(ncs <= cs)
1012 self.assertFalse(ncs > cs)
1013 self.assertTrue(ncs >= cs)
1014
1015 def assertSameSet(self, s1, s2):
1016 # coerce both to a real set then check equality
1017 self.assertSetEqual(set(s1), set(s2))
1018
1019 def test_Set_interoperability_with_real_sets(self):
1020 # Issue: 8743
1021 class ListSet(Set):
1022 def __init__(self, elements=()):
1023 self.data = []
1024 for elem in elements:
1025 if elem not in self.data:
1026 self.data.append(elem)
1027 def __contains__(self, elem):
1028 return elem in self.data
1029 def __iter__(self):
1030 return iter(self.data)
1031 def __len__(self):
1032 return len(self.data)
1033 def __repr__(self):
1034 return 'Set({!r})'.format(self.data)
1035
1036 r1 = set('abc')
1037 r2 = set('bcd')
1038 r3 = set('abcde')
1039 f1 = ListSet('abc')
1040 f2 = ListSet('bcd')
1041 f3 = ListSet('abcde')
1042 l1 = list('abccba')
1043 l2 = list('bcddcb')
1044 l3 = list('abcdeedcba')
1045
1046 target = r1 & r2
1047 self.assertSameSet(f1 & f2, target)
1048 self.assertSameSet(f1 & r2, target)
1049 self.assertSameSet(r2 & f1, target)
1050 self.assertSameSet(f1 & l2, target)
1051
1052 target = r1 | r2
1053 self.assertSameSet(f1 | f2, target)
1054 self.assertSameSet(f1 | r2, target)
1055 self.assertSameSet(r2 | f1, target)
1056 self.assertSameSet(f1 | l2, target)
1057
1058 fwd_target = r1 - r2
1059 rev_target = r2 - r1
1060 self.assertSameSet(f1 - f2, fwd_target)
1061 self.assertSameSet(f2 - f1, rev_target)
1062 self.assertSameSet(f1 - r2, fwd_target)
1063 self.assertSameSet(f2 - r1, rev_target)
1064 self.assertSameSet(r1 - f2, fwd_target)
1065 self.assertSameSet(r2 - f1, rev_target)
1066 self.assertSameSet(f1 - l2, fwd_target)
1067 self.assertSameSet(f2 - l1, rev_target)
1068
1069 target = r1 ^ r2
1070 self.assertSameSet(f1 ^ f2, target)
1071 self.assertSameSet(f1 ^ r2, target)
1072 self.assertSameSet(r2 ^ f1, target)
1073 self.assertSameSet(f1 ^ l2, target)
1074
1075 # Don't change the following to use assertLess or other
1076 # "more specific" unittest assertions. The current
1077 # assertTrue/assertFalse style makes the pattern of test
1078 # case combinations clear and allows us to know for sure
1079 # the exact operator being invoked.
1080
1081 # proper subset
1082 self.assertTrue(f1 < f3)
1083 self.assertFalse(f1 < f1)
1084 self.assertFalse(f1 < f2)
1085 self.assertTrue(r1 < f3)
1086 self.assertFalse(r1 < f1)
1087 self.assertFalse(r1 < f2)
1088 self.assertTrue(r1 < r3)
1089 self.assertFalse(r1 < r1)
1090 self.assertFalse(r1 < r2)
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +02001091 with self.assertRaises(TypeError):
Raymond Hettingerdd5e53a2014-05-26 00:09:04 -07001092 f1 < l3
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +02001093 with self.assertRaises(TypeError):
Raymond Hettingerdd5e53a2014-05-26 00:09:04 -07001094 f1 < l1
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +02001095 with self.assertRaises(TypeError):
Raymond Hettingerdd5e53a2014-05-26 00:09:04 -07001096 f1 < l2
1097
1098 # any subset
1099 self.assertTrue(f1 <= f3)
1100 self.assertTrue(f1 <= f1)
1101 self.assertFalse(f1 <= f2)
1102 self.assertTrue(r1 <= f3)
1103 self.assertTrue(r1 <= f1)
1104 self.assertFalse(r1 <= f2)
1105 self.assertTrue(r1 <= r3)
1106 self.assertTrue(r1 <= r1)
1107 self.assertFalse(r1 <= r2)
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +02001108 with self.assertRaises(TypeError):
Raymond Hettingerdd5e53a2014-05-26 00:09:04 -07001109 f1 <= l3
1110 with self.assertRaises(TypeError):
1111 f1 <= l1
1112 with self.assertRaises(TypeError):
1113 f1 <= l2
1114
1115 # proper superset
1116 self.assertTrue(f3 > f1)
1117 self.assertFalse(f1 > f1)
1118 self.assertFalse(f2 > f1)
1119 self.assertTrue(r3 > r1)
1120 self.assertFalse(f1 > r1)
1121 self.assertFalse(f2 > r1)
1122 self.assertTrue(r3 > r1)
1123 self.assertFalse(r1 > r1)
1124 self.assertFalse(r2 > r1)
1125 with self.assertRaises(TypeError):
1126 f1 > l3
1127 with self.assertRaises(TypeError):
1128 f1 > l1
1129 with self.assertRaises(TypeError):
1130 f1 > l2
1131
1132 # any superset
1133 self.assertTrue(f3 >= f1)
1134 self.assertTrue(f1 >= f1)
1135 self.assertFalse(f2 >= f1)
1136 self.assertTrue(r3 >= r1)
1137 self.assertTrue(f1 >= r1)
1138 self.assertFalse(f2 >= r1)
1139 self.assertTrue(r3 >= r1)
1140 self.assertTrue(r1 >= r1)
1141 self.assertFalse(r2 >= r1)
1142 with self.assertRaises(TypeError):
1143 f1 >= l3
1144 with self.assertRaises(TypeError):
1145 f1 >=l1
1146 with self.assertRaises(TypeError):
1147 f1 >= l2
1148
1149 # equality
1150 self.assertTrue(f1 == f1)
1151 self.assertTrue(r1 == f1)
1152 self.assertTrue(f1 == r1)
1153 self.assertFalse(f1 == f3)
1154 self.assertFalse(r1 == f3)
1155 self.assertFalse(f1 == r3)
1156 self.assertFalse(f1 == l3)
1157 self.assertFalse(f1 == l1)
1158 self.assertFalse(f1 == l2)
1159
1160 # inequality
1161 self.assertFalse(f1 != f1)
1162 self.assertFalse(r1 != f1)
1163 self.assertFalse(f1 != r1)
1164 self.assertTrue(f1 != f3)
1165 self.assertTrue(r1 != f3)
1166 self.assertTrue(f1 != r3)
1167 self.assertTrue(f1 != l3)
1168 self.assertTrue(f1 != l1)
1169 self.assertTrue(f1 != l2)
Andrew Svetlovbcac6ad2012-11-01 13:28:54 +02001170
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001171 def test_Mapping(self):
1172 for sample in [dict]:
Ezio Melottie9615932010-01-24 19:26:24 +00001173 self.assertIsInstance(sample(), Mapping)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001174 self.assertTrue(issubclass(sample, Mapping))
Raymond Hettingerae650182009-01-28 23:33:59 +00001175 self.validate_abstract_methods(Mapping, '__contains__', '__iter__', '__len__',
1176 '__getitem__')
Raymond Hettinger57d1a882011-02-23 00:46:28 +00001177 class MyMapping(Mapping):
Benjamin Peterson4ad6bd52010-05-21 20:55:22 +00001178 def __len__(self):
1179 return 0
1180 def __getitem__(self, i):
1181 raise IndexError
1182 def __iter__(self):
1183 return iter(())
1184 self.validate_comparison(MyMapping())
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001185
1186 def test_MutableMapping(self):
1187 for sample in [dict]:
Ezio Melottie9615932010-01-24 19:26:24 +00001188 self.assertIsInstance(sample(), MutableMapping)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001189 self.assertTrue(issubclass(sample, MutableMapping))
Raymond Hettingerae650182009-01-28 23:33:59 +00001190 self.validate_abstract_methods(MutableMapping, '__contains__', '__iter__', '__len__',
1191 '__getitem__', '__setitem__', '__delitem__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001192
Raymond Hettinger9117c752010-08-22 07:44:24 +00001193 def test_MutableMapping_subclass(self):
1194 # Test issue 9214
1195 mymap = UserDict()
1196 mymap['red'] = 5
1197 self.assertIsInstance(mymap.keys(), Set)
1198 self.assertIsInstance(mymap.keys(), KeysView)
1199 self.assertIsInstance(mymap.items(), Set)
1200 self.assertIsInstance(mymap.items(), ItemsView)
1201
1202 mymap = UserDict()
1203 mymap['red'] = 5
1204 z = mymap.keys() | {'orange'}
1205 self.assertIsInstance(z, set)
1206 list(z)
1207 mymap['blue'] = 7 # Shouldn't affect 'z'
1208 self.assertEqual(sorted(z), ['orange', 'red'])
1209
1210 mymap = UserDict()
1211 mymap['red'] = 5
1212 z = mymap.items() | {('orange', 3)}
1213 self.assertIsInstance(z, set)
1214 list(z)
1215 mymap['blue'] = 7 # Shouldn't affect 'z'
1216 self.assertEqual(sorted(z), [('orange', 3), ('red', 5)])
1217
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001218 def test_Sequence(self):
1219 for sample in [tuple, list, bytes, str]:
Ezio Melottie9615932010-01-24 19:26:24 +00001220 self.assertIsInstance(sample(), Sequence)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001221 self.assertTrue(issubclass(sample, Sequence))
Ezio Melottie9615932010-01-24 19:26:24 +00001222 self.assertIsInstance(range(10), Sequence)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001223 self.assertTrue(issubclass(range, Sequence))
Nick Coghlan45163cc2013-10-02 22:31:47 +10001224 self.assertIsInstance(memoryview(b""), Sequence)
1225 self.assertTrue(issubclass(memoryview, Sequence))
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001226 self.assertTrue(issubclass(str, Sequence))
Raymond Hettingerae650182009-01-28 23:33:59 +00001227 self.validate_abstract_methods(Sequence, '__contains__', '__iter__', '__len__',
1228 '__getitem__')
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001229
Guido van Rossumd05eb002007-11-21 22:26:24 +00001230 def test_ByteString(self):
1231 for sample in [bytes, bytearray]:
Ezio Melottie9615932010-01-24 19:26:24 +00001232 self.assertIsInstance(sample(), ByteString)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001233 self.assertTrue(issubclass(sample, ByteString))
Guido van Rossumd05eb002007-11-21 22:26:24 +00001234 for sample in [str, list, tuple]:
Ezio Melottie9615932010-01-24 19:26:24 +00001235 self.assertNotIsInstance(sample(), ByteString)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001236 self.assertFalse(issubclass(sample, ByteString))
Ezio Melottie9615932010-01-24 19:26:24 +00001237 self.assertNotIsInstance(memoryview(b""), ByteString)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001238 self.assertFalse(issubclass(memoryview, ByteString))
Guido van Rossumd05eb002007-11-21 22:26:24 +00001239
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001240 def test_MutableSequence(self):
Guido van Rossumd05eb002007-11-21 22:26:24 +00001241 for sample in [tuple, str, bytes]:
Ezio Melottie9615932010-01-24 19:26:24 +00001242 self.assertNotIsInstance(sample(), MutableSequence)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001243 self.assertFalse(issubclass(sample, MutableSequence))
Raymond Hettinger32ea1652015-03-21 01:37:37 -07001244 for sample in [list, bytearray, deque]:
Ezio Melottie9615932010-01-24 19:26:24 +00001245 self.assertIsInstance(sample(), MutableSequence)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001246 self.assertTrue(issubclass(sample, MutableSequence))
1247 self.assertFalse(issubclass(str, MutableSequence))
Raymond Hettingerae650182009-01-28 23:33:59 +00001248 self.validate_abstract_methods(MutableSequence, '__contains__', '__iter__',
1249 '__len__', '__getitem__', '__setitem__', '__delitem__', 'insert')
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001250
Eli Bendersky0716a572011-03-04 10:38:14 +00001251 def test_MutableSequence_mixins(self):
1252 # Test the mixins of MutableSequence by creating a miminal concrete
1253 # class inherited from it.
1254 class MutableSequenceSubclass(MutableSequence):
1255 def __init__(self):
1256 self.lst = []
1257
1258 def __setitem__(self, index, value):
1259 self.lst[index] = value
1260
1261 def __getitem__(self, index):
1262 return self.lst[index]
1263
1264 def __len__(self):
1265 return len(self.lst)
1266
1267 def __delitem__(self, index):
1268 del self.lst[index]
1269
1270 def insert(self, index, value):
1271 self.lst.insert(index, value)
1272
1273 mss = MutableSequenceSubclass()
1274 mss.append(0)
1275 mss.extend((1, 2, 3, 4))
1276 self.assertEqual(len(mss), 5)
1277 self.assertEqual(mss[3], 3)
1278 mss.reverse()
1279 self.assertEqual(mss[3], 1)
1280 mss.pop()
1281 self.assertEqual(len(mss), 4)
1282 mss.remove(3)
1283 self.assertEqual(len(mss), 3)
1284 mss += (10, 20, 30)
1285 self.assertEqual(len(mss), 6)
1286 self.assertEqual(mss[-1], 30)
1287 mss.clear()
1288 self.assertEqual(len(mss), 0)
Raymond Hettinger499e1932011-02-23 07:56:53 +00001289
1290################################################################################
1291### Counter
1292################################################################################
1293
Raymond Hettingercb1d96f2013-10-04 16:51:02 -07001294class CounterSubclassWithSetItem(Counter):
1295 # Test a counter subclass that overrides __setitem__
1296 def __init__(self, *args, **kwds):
1297 self.called = False
1298 Counter.__init__(self, *args, **kwds)
1299 def __setitem__(self, key, value):
1300 self.called = True
1301 Counter.__setitem__(self, key, value)
1302
1303class CounterSubclassWithGet(Counter):
1304 # Test a counter subclass that overrides get()
1305 def __init__(self, *args, **kwds):
1306 self.called = False
1307 Counter.__init__(self, *args, **kwds)
1308 def get(self, key, default):
1309 self.called = True
1310 return Counter.get(self, key, default)
1311
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001312class TestCounter(unittest.TestCase):
1313
1314 def test_basics(self):
1315 c = Counter('abcaba')
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001316 self.assertEqual(c, Counter({'a':3 , 'b': 2, 'c': 1}))
1317 self.assertEqual(c, Counter(a=3, b=2, c=1))
Ezio Melottie9615932010-01-24 19:26:24 +00001318 self.assertIsInstance(c, dict)
1319 self.assertIsInstance(c, Mapping)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001320 self.assertTrue(issubclass(Counter, dict))
1321 self.assertTrue(issubclass(Counter, Mapping))
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001322 self.assertEqual(len(c), 3)
1323 self.assertEqual(sum(c.values()), 6)
1324 self.assertEqual(sorted(c.values()), [1, 2, 3])
1325 self.assertEqual(sorted(c.keys()), ['a', 'b', 'c'])
1326 self.assertEqual(sorted(c), ['a', 'b', 'c'])
1327 self.assertEqual(sorted(c.items()),
1328 [('a', 3), ('b', 2), ('c', 1)])
1329 self.assertEqual(c['b'], 2)
1330 self.assertEqual(c['z'], 0)
1331 self.assertEqual(c.__contains__('c'), True)
1332 self.assertEqual(c.__contains__('z'), False)
1333 self.assertEqual(c.get('b', 10), 2)
1334 self.assertEqual(c.get('z', 10), 10)
1335 self.assertEqual(c, dict(a=3, b=2, c=1))
1336 self.assertEqual(repr(c), "Counter({'a': 3, 'b': 2, 'c': 1})")
1337 self.assertEqual(c.most_common(), [('a', 3), ('b', 2), ('c', 1)])
1338 for i in range(5):
1339 self.assertEqual(c.most_common(i),
1340 [('a', 3), ('b', 2), ('c', 1)][:i])
1341 self.assertEqual(''.join(sorted(c.elements())), 'aaabbc')
1342 c['a'] += 1 # increment an existing value
1343 c['b'] -= 2 # sub existing value to zero
1344 del c['c'] # remove an entry
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001345 del c['c'] # make sure that del doesn't raise KeyError
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001346 c['d'] -= 2 # sub from a missing value
1347 c['e'] = -5 # directly assign a missing value
1348 c['f'] += 4 # add to a missing value
1349 self.assertEqual(c, dict(a=4, b=0, d=-2, e=-5, f=4))
1350 self.assertEqual(''.join(sorted(c.elements())), 'aaaaffff')
1351 self.assertEqual(c.pop('f'), 4)
Ezio Melottib58e0bd2010-01-23 15:40:09 +00001352 self.assertNotIn('f', c)
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001353 for i in range(3):
1354 elem, cnt = c.popitem()
Ezio Melottib58e0bd2010-01-23 15:40:09 +00001355 self.assertNotIn(elem, c)
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001356 c.clear()
1357 self.assertEqual(c, {})
1358 self.assertEqual(repr(c), 'Counter()')
1359 self.assertRaises(NotImplementedError, Counter.fromkeys, 'abc')
1360 self.assertRaises(TypeError, hash, c)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001361 c.update(dict(a=5, b=3))
1362 c.update(c=1)
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001363 c.update(Counter('a' * 50 + 'b' * 30))
1364 c.update() # test case with no args
1365 c.__init__('a' * 500 + 'b' * 300)
1366 c.__init__('cdc')
1367 c.__init__()
1368 self.assertEqual(c, dict(a=555, b=333, c=3, d=1))
1369 self.assertEqual(c.setdefault('d', 5), 1)
1370 self.assertEqual(c['d'], 1)
1371 self.assertEqual(c.setdefault('e', 5), 5)
1372 self.assertEqual(c['e'], 5)
1373
Serhiy Storchakaae5cb212014-11-27 16:25:51 +02001374 def test_init(self):
1375 self.assertEqual(list(Counter(self=42).items()), [('self', 42)])
1376 self.assertEqual(list(Counter(iterable=42).items()), [('iterable', 42)])
1377 self.assertEqual(list(Counter(iterable=None).items()), [('iterable', None)])
1378 self.assertRaises(TypeError, Counter, 42)
1379 self.assertRaises(TypeError, Counter, (), ())
1380 self.assertRaises(TypeError, Counter.__init__)
1381
1382 def test_update(self):
1383 c = Counter()
1384 c.update(self=42)
1385 self.assertEqual(list(c.items()), [('self', 42)])
1386 c = Counter()
1387 c.update(iterable=42)
1388 self.assertEqual(list(c.items()), [('iterable', 42)])
1389 c = Counter()
1390 c.update(iterable=None)
1391 self.assertEqual(list(c.items()), [('iterable', None)])
1392 self.assertRaises(TypeError, Counter().update, 42)
1393 self.assertRaises(TypeError, Counter().update, {}, {})
1394 self.assertRaises(TypeError, Counter.update)
1395
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001396 def test_copying(self):
1397 # Check that counters are copyable, deepcopyable, picklable, and
1398 #have a repr/eval round-trip
1399 words = Counter('which witch had which witches wrist watch'.split())
Serhiy Storchakabad12572014-12-15 14:03:42 +02001400 def check(dup):
1401 msg = "\ncopy: %s\nwords: %s" % (dup, words)
1402 self.assertIsNot(dup, words, msg)
1403 self.assertEqual(dup, words)
1404 check(words.copy())
1405 check(copy.copy(words))
1406 check(copy.deepcopy(words))
1407 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
1408 with self.subTest(proto=proto):
1409 check(pickle.loads(pickle.dumps(words, proto)))
1410 check(eval(repr(words)))
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001411 update_test = Counter()
1412 update_test.update(words)
Serhiy Storchakabad12572014-12-15 14:03:42 +02001413 check(update_test)
1414 check(Counter(words))
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001415
Raymond Hettinger1c746c22011-04-15 13:16:46 -07001416 def test_copy_subclass(self):
1417 class MyCounter(Counter):
1418 pass
1419 c = MyCounter('slartibartfast')
1420 d = c.copy()
1421 self.assertEqual(d, c)
1422 self.assertEqual(len(d), len(c))
1423 self.assertEqual(type(d), type(c))
1424
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001425 def test_conversions(self):
1426 # Convert to: set, list, dict
1427 s = 'she sells sea shells by the sea shore'
1428 self.assertEqual(sorted(Counter(s).elements()), sorted(s))
1429 self.assertEqual(sorted(Counter(s)), sorted(set(s)))
1430 self.assertEqual(dict(Counter(s)), dict(Counter(s).items()))
1431 self.assertEqual(set(Counter(s)), set(s))
1432
Raymond Hettinger670eaec2009-01-21 23:14:07 +00001433 def test_invariant_for_the_in_operator(self):
1434 c = Counter(a=10, b=-2, c=0)
1435 for elem in c:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001436 self.assertTrue(elem in c)
Benjamin Peterson577473f2010-01-19 00:09:57 +00001437 self.assertIn(elem, c)
Raymond Hettinger670eaec2009-01-21 23:14:07 +00001438
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001439 def test_multiset_operations(self):
1440 # Verify that adding a zero counter will strip zeros and negatives
1441 c = Counter(a=10, b=-2, c=0) + Counter()
1442 self.assertEqual(dict(c), dict(a=10))
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001443
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001444 elements = 'abcd'
1445 for i in range(1000):
1446 # test random pairs of multisets
1447 p = Counter(dict((elem, randrange(-2,4)) for elem in elements))
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +00001448 p.update(e=1, f=-1, g=0)
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001449 q = Counter(dict((elem, randrange(-2,4)) for elem in elements))
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +00001450 q.update(h=1, i=-1, j=0)
1451 for counterop, numberop in [
1452 (Counter.__add__, lambda x, y: max(0, x+y)),
1453 (Counter.__sub__, lambda x, y: max(0, x-y)),
1454 (Counter.__or__, lambda x, y: max(0,x,y)),
1455 (Counter.__and__, lambda x, y: max(0, min(x,y))),
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001456 ]:
1457 result = counterop(p, q)
1458 for x in elements:
Raymond Hettingere0d1b9f2009-01-21 20:36:27 +00001459 self.assertEqual(numberop(p[x], q[x]), result[x],
1460 (counterop, x, p, q))
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001461 # verify that results exclude non-positive counts
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001462 self.assertTrue(x>0 for x in result.values())
Raymond Hettinger4d2073a2009-01-20 03:41:22 +00001463
1464 elements = 'abcdef'
1465 for i in range(100):
1466 # verify that random multisets with no repeats are exactly like sets
1467 p = Counter(dict((elem, randrange(0, 2)) for elem in elements))
1468 q = Counter(dict((elem, randrange(0, 2)) for elem in elements))
1469 for counterop, setop in [
1470 (Counter.__sub__, set.__sub__),
1471 (Counter.__or__, set.__or__),
1472 (Counter.__and__, set.__and__),
1473 ]:
1474 counter_result = counterop(p, q)
1475 set_result = setop(set(p.elements()), set(q.elements()))
1476 self.assertEqual(counter_result, dict.fromkeys(set_result, 1))
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001477
Raymond Hettingerbecd5682011-10-19 13:40:37 -07001478 def test_inplace_operations(self):
1479 elements = 'abcd'
1480 for i in range(1000):
1481 # test random pairs of multisets
1482 p = Counter(dict((elem, randrange(-2,4)) for elem in elements))
1483 p.update(e=1, f=-1, g=0)
1484 q = Counter(dict((elem, randrange(-2,4)) for elem in elements))
1485 q.update(h=1, i=-1, j=0)
1486 for inplace_op, regular_op in [
1487 (Counter.__iadd__, Counter.__add__),
1488 (Counter.__isub__, Counter.__sub__),
1489 (Counter.__ior__, Counter.__or__),
1490 (Counter.__iand__, Counter.__and__),
1491 ]:
1492 c = p.copy()
1493 c_id = id(c)
1494 regular_result = regular_op(c, q)
1495 inplace_result = inplace_op(c, q)
1496 self.assertEqual(inplace_result, regular_result)
1497 self.assertEqual(id(inplace_result), c_id)
1498
Raymond Hettinger9c01e442010-04-03 10:32:58 +00001499 def test_subtract(self):
1500 c = Counter(a=-5, b=0, c=5, d=10, e=15,g=40)
1501 c.subtract(a=1, b=2, c=-3, d=10, e=20, f=30, h=-50)
1502 self.assertEqual(c, Counter(a=-6, b=-2, c=8, d=0, e=-5, f=-30, g=40, h=50))
1503 c = Counter(a=-5, b=0, c=5, d=10, e=15,g=40)
1504 c.subtract(Counter(a=1, b=2, c=-3, d=10, e=20, f=30, h=-50))
1505 self.assertEqual(c, Counter(a=-6, b=-2, c=8, d=0, e=-5, f=-30, g=40, h=50))
1506 c = Counter('aaabbcd')
1507 c.subtract('aaaabbcce')
1508 self.assertEqual(c, Counter(a=-1, b=0, c=-1, d=1, e=-1))
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001509
Serhiy Storchakaae5cb212014-11-27 16:25:51 +02001510 c = Counter()
1511 c.subtract(self=42)
1512 self.assertEqual(list(c.items()), [('self', -42)])
1513 c = Counter()
1514 c.subtract(iterable=42)
1515 self.assertEqual(list(c.items()), [('iterable', -42)])
1516 self.assertRaises(TypeError, Counter().subtract, 42)
1517 self.assertRaises(TypeError, Counter().subtract, {}, {})
1518 self.assertRaises(TypeError, Counter.subtract)
1519
Raymond Hettingerfcb393c2011-08-09 13:00:40 -07001520 def test_unary(self):
1521 c = Counter(a=-5, b=0, c=5, d=10, e=15,g=40)
1522 self.assertEqual(dict(+c), dict(c=5, d=10, e=15, g=40))
1523 self.assertEqual(dict(-c), dict(a=5))
1524
Raymond Hettinger4e6bf412011-11-05 13:35:26 -07001525 def test_repr_nonsortable(self):
1526 c = Counter(a=2, b=None)
1527 r = repr(c)
1528 self.assertIn("'a': 2", r)
1529 self.assertIn("'b': None", r)
Raymond Hettinger68fb89f2011-11-05 13:43:01 -07001530
Raymond Hettinger426e0522011-01-03 02:12:02 +00001531 def test_helper_function(self):
1532 # two paths, one for real dicts and one for other mappings
1533 elems = list('abracadabra')
1534
1535 d = dict()
1536 _count_elements(d, elems)
1537 self.assertEqual(d, {'a': 5, 'r': 2, 'b': 2, 'c': 1, 'd': 1})
1538
1539 m = OrderedDict()
1540 _count_elements(m, elems)
1541 self.assertEqual(m,
1542 OrderedDict([('a', 5), ('b', 2), ('r', 2), ('c', 1), ('d', 1)]))
1543
Raymond Hettingercb1d96f2013-10-04 16:51:02 -07001544 # test fidelity to the pure python version
1545 c = CounterSubclassWithSetItem('abracadabra')
1546 self.assertTrue(c.called)
Raymond Hettingerfacd0a32013-10-05 17:14:51 -07001547 self.assertEqual(dict(c), {'a': 5, 'b': 2, 'c': 1, 'd': 1, 'r':2 })
Raymond Hettingercb1d96f2013-10-04 16:51:02 -07001548 c = CounterSubclassWithGet('abracadabra')
1549 self.assertTrue(c.called)
Raymond Hettingerfacd0a32013-10-05 17:14:51 -07001550 self.assertEqual(dict(c), {'a': 5, 'b': 2, 'c': 1, 'd': 1, 'r':2 })
Raymond Hettingercb1d96f2013-10-04 16:51:02 -07001551
Raymond Hettinger499e1932011-02-23 07:56:53 +00001552
1553################################################################################
1554### OrderedDict
1555################################################################################
1556
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001557class TestOrderedDict(unittest.TestCase):
1558
1559 def test_init(self):
1560 with self.assertRaises(TypeError):
1561 OrderedDict([('a', 1), ('b', 2)], None) # too many args
1562 pairs = [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', 5)]
1563 self.assertEqual(sorted(OrderedDict(dict(pairs)).items()), pairs) # dict input
1564 self.assertEqual(sorted(OrderedDict(**dict(pairs)).items()), pairs) # kwds input
1565 self.assertEqual(list(OrderedDict(pairs).items()), pairs) # pairs input
1566 self.assertEqual(list(OrderedDict([('a', 1), ('b', 2), ('c', 9), ('d', 4)],
1567 c=3, e=5).items()), pairs) # mixed input
1568
1569 # make sure no positional args conflict with possible kwdargs
Serhiy Storchakaae5cb212014-11-27 16:25:51 +02001570 self.assertEqual(list(OrderedDict(self=42).items()), [('self', 42)])
1571 self.assertEqual(list(OrderedDict(other=42).items()), [('other', 42)])
1572 self.assertRaises(TypeError, OrderedDict, 42)
1573 self.assertRaises(TypeError, OrderedDict, (), ())
1574 self.assertRaises(TypeError, OrderedDict.__init__)
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001575
1576 # Make sure that direct calls to __init__ do not clear previous contents
1577 d = OrderedDict([('a', 1), ('b', 2), ('c', 3), ('d', 44), ('e', 55)])
1578 d.__init__([('e', 5), ('f', 6)], g=7, d=4)
1579 self.assertEqual(list(d.items()),
1580 [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', 5), ('f', 6), ('g', 7)])
1581
1582 def test_update(self):
1583 with self.assertRaises(TypeError):
1584 OrderedDict().update([('a', 1), ('b', 2)], None) # too many args
1585 pairs = [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', 5)]
1586 od = OrderedDict()
1587 od.update(dict(pairs))
1588 self.assertEqual(sorted(od.items()), pairs) # dict input
1589 od = OrderedDict()
1590 od.update(**dict(pairs))
1591 self.assertEqual(sorted(od.items()), pairs) # kwds input
1592 od = OrderedDict()
1593 od.update(pairs)
1594 self.assertEqual(list(od.items()), pairs) # pairs input
1595 od = OrderedDict()
1596 od.update([('a', 1), ('b', 2), ('c', 9), ('d', 4)], c=3, e=5)
1597 self.assertEqual(list(od.items()), pairs) # mixed input
1598
Mark Dickinsonb214e902010-07-11 18:53:06 +00001599 # Issue 9137: Named argument called 'other' or 'self'
1600 # shouldn't be treated specially.
1601 od = OrderedDict()
1602 od.update(self=23)
1603 self.assertEqual(list(od.items()), [('self', 23)])
1604 od = OrderedDict()
1605 od.update(other={})
1606 self.assertEqual(list(od.items()), [('other', {})])
1607 od = OrderedDict()
1608 od.update(red=5, blue=6, other=7, self=8)
1609 self.assertEqual(sorted(list(od.items())),
1610 [('blue', 6), ('other', 7), ('red', 5), ('self', 8)])
1611
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001612 # Make sure that direct calls to update do not clear previous contents
1613 # add that updates items are not moved to the end
1614 d = OrderedDict([('a', 1), ('b', 2), ('c', 3), ('d', 44), ('e', 55)])
1615 d.update([('e', 5), ('f', 6)], g=7, d=4)
1616 self.assertEqual(list(d.items()),
1617 [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', 5), ('f', 6), ('g', 7)])
1618
Serhiy Storchakaae5cb212014-11-27 16:25:51 +02001619 self.assertRaises(TypeError, OrderedDict().update, 42)
1620 self.assertRaises(TypeError, OrderedDict().update, (), ())
1621 self.assertRaises(TypeError, OrderedDict.update)
1622
Raymond Hettinger345c49b2011-01-01 23:51:55 +00001623 def test_abc(self):
1624 self.assertIsInstance(OrderedDict(), MutableMapping)
1625 self.assertTrue(issubclass(OrderedDict, MutableMapping))
1626
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001627 def test_clear(self):
1628 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1629 shuffle(pairs)
1630 od = OrderedDict(pairs)
1631 self.assertEqual(len(od), len(pairs))
1632 od.clear()
1633 self.assertEqual(len(od), 0)
1634
1635 def test_delitem(self):
1636 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1637 od = OrderedDict(pairs)
1638 del od['a']
Benjamin Peterson577473f2010-01-19 00:09:57 +00001639 self.assertNotIn('a', od)
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001640 with self.assertRaises(KeyError):
1641 del od['a']
1642 self.assertEqual(list(od.items()), pairs[:2] + pairs[3:])
1643
1644 def test_setitem(self):
1645 od = OrderedDict([('d', 1), ('b', 2), ('c', 3), ('a', 4), ('e', 5)])
1646 od['c'] = 10 # existing element
1647 od['f'] = 20 # new element
1648 self.assertEqual(list(od.items()),
1649 [('d', 1), ('b', 2), ('c', 10), ('a', 4), ('e', 5), ('f', 20)])
1650
1651 def test_iterators(self):
1652 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1653 shuffle(pairs)
1654 od = OrderedDict(pairs)
1655 self.assertEqual(list(od), [t[0] for t in pairs])
1656 self.assertEqual(list(od.keys()), [t[0] for t in pairs])
1657 self.assertEqual(list(od.values()), [t[1] for t in pairs])
1658 self.assertEqual(list(od.items()), pairs)
1659 self.assertEqual(list(reversed(od)),
1660 [t[0] for t in reversed(pairs)])
Serhiy Storchaka578c9212014-04-04 15:19:36 +03001661 self.assertEqual(list(reversed(od.keys())),
1662 [t[0] for t in reversed(pairs)])
1663 self.assertEqual(list(reversed(od.values())),
1664 [t[1] for t in reversed(pairs)])
1665 self.assertEqual(list(reversed(od.items())), list(reversed(pairs)))
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001666
Raymond Hettinger53d2c412014-05-03 21:58:45 -07001667 def test_detect_deletion_during_iteration(self):
1668 od = OrderedDict.fromkeys('abc')
1669 it = iter(od)
1670 key = next(it)
1671 del od[key]
1672 with self.assertRaises(Exception):
1673 # Note, the exact exception raised is not guaranteed
1674 # The only guarantee that the next() will not succeed
1675 next(it)
1676
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001677 def test_popitem(self):
1678 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1679 shuffle(pairs)
1680 od = OrderedDict(pairs)
1681 while pairs:
1682 self.assertEqual(od.popitem(), pairs.pop())
1683 with self.assertRaises(KeyError):
1684 od.popitem()
1685 self.assertEqual(len(od), 0)
1686
1687 def test_pop(self):
1688 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1689 shuffle(pairs)
1690 od = OrderedDict(pairs)
1691 shuffle(pairs)
1692 while pairs:
1693 k, v = pairs.pop()
1694 self.assertEqual(od.pop(k), v)
1695 with self.assertRaises(KeyError):
1696 od.pop('xyz')
1697 self.assertEqual(len(od), 0)
1698 self.assertEqual(od.pop(k, 12345), 12345)
1699
Raymond Hettinger345c49b2011-01-01 23:51:55 +00001700 # make sure pop still works when __missing__ is defined
1701 class Missing(OrderedDict):
1702 def __missing__(self, key):
1703 return 0
1704 m = Missing(a=1)
1705 self.assertEqual(m.pop('b', 5), 5)
1706 self.assertEqual(m.pop('a', 6), 1)
1707 self.assertEqual(m.pop('a', 6), 6)
1708 with self.assertRaises(KeyError):
1709 m.pop('a')
1710
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001711 def test_equality(self):
1712 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1713 shuffle(pairs)
1714 od1 = OrderedDict(pairs)
1715 od2 = OrderedDict(pairs)
1716 self.assertEqual(od1, od2) # same order implies equality
1717 pairs = pairs[2:] + pairs[:2]
1718 od2 = OrderedDict(pairs)
1719 self.assertNotEqual(od1, od2) # different order implies inequality
1720 # comparison to regular dict is not order sensitive
1721 self.assertEqual(od1, dict(od2))
1722 self.assertEqual(dict(od2), od1)
1723 # different length implied inequality
1724 self.assertNotEqual(od1, OrderedDict(pairs[:-1]))
1725
1726 def test_copying(self):
1727 # Check that ordered dicts are copyable, deepcopyable, picklable,
1728 # and have a repr/eval round-trip
1729 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1730 od = OrderedDict(pairs)
Serhiy Storchakabad12572014-12-15 14:03:42 +02001731 def check(dup):
1732 msg = "\ncopy: %s\nod: %s" % (dup, od)
1733 self.assertIsNot(dup, od, msg)
1734 self.assertEqual(dup, od)
1735 check(od.copy())
1736 check(copy.copy(od))
1737 check(copy.deepcopy(od))
1738 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
1739 with self.subTest(proto=proto):
1740 check(pickle.loads(pickle.dumps(od, proto)))
1741 check(eval(repr(od)))
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001742 update_test = OrderedDict()
1743 update_test.update(od)
Serhiy Storchakabad12572014-12-15 14:03:42 +02001744 check(update_test)
1745 check(OrderedDict(od))
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001746
Raymond Hettinger5b26fb52009-03-03 22:38:22 +00001747 def test_yaml_linkage(self):
1748 # Verify that __reduce__ is setup in a way that supports PyYAML's dump() feature.
1749 # In yaml, lists are native but tuples are not.
1750 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1751 od = OrderedDict(pairs)
1752 # yaml.dump(od) -->
1753 # '!!python/object/apply:__main__.OrderedDict\n- - [a, 1]\n - [b, 2]\n'
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001754 self.assertTrue(all(type(pair)==list for pair in od.__reduce__()[1]))
Raymond Hettinger5b26fb52009-03-03 22:38:22 +00001755
Raymond Hettingerb2121572009-03-03 22:50:04 +00001756 def test_reduce_not_too_fat(self):
1757 # do not save instance dictionary if not needed
1758 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1759 od = OrderedDict(pairs)
Serhiy Storchaka3ee6dab2013-05-21 12:47:57 +03001760 self.assertIsNone(od.__reduce__()[2])
Raymond Hettingerb2121572009-03-03 22:50:04 +00001761 od.x = 10
Serhiy Storchaka3ee6dab2013-05-21 12:47:57 +03001762 self.assertIsNotNone(od.__reduce__()[2])
1763
1764 def test_pickle_recursive(self):
1765 od = OrderedDict()
1766 od[1] = od
1767 for proto in range(-1, pickle.HIGHEST_PROTOCOL + 1):
1768 dup = pickle.loads(pickle.dumps(od, proto))
1769 self.assertIsNot(dup, od)
1770 self.assertEqual(list(dup.keys()), [1])
1771 self.assertIs(dup[1], dup)
Raymond Hettingerb2121572009-03-03 22:50:04 +00001772
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001773 def test_repr(self):
1774 od = OrderedDict([('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)])
1775 self.assertEqual(repr(od),
1776 "OrderedDict([('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)])")
1777 self.assertEqual(eval(repr(od)), od)
1778 self.assertEqual(repr(OrderedDict()), "OrderedDict()")
1779
Raymond Hettingerdc08a142010-09-12 05:15:22 +00001780 def test_repr_recursive(self):
1781 # See issue #9826
1782 od = OrderedDict.fromkeys('abc')
1783 od['x'] = od
1784 self.assertEqual(repr(od),
1785 "OrderedDict([('a', None), ('b', None), ('c', None), ('x', ...)])")
1786
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001787 def test_setdefault(self):
1788 pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
1789 shuffle(pairs)
1790 od = OrderedDict(pairs)
1791 pair_order = list(od.items())
1792 self.assertEqual(od.setdefault('a', 10), 3)
1793 # make sure order didn't change
1794 self.assertEqual(list(od.items()), pair_order)
1795 self.assertEqual(od.setdefault('x', 10), 10)
1796 # make sure 'x' is added to the end
1797 self.assertEqual(list(od.items())[-1], ('x', 10))
1798
Raymond Hettingera673b1f2010-12-31 23:16:17 +00001799 # make sure setdefault still works when __missing__ is defined
1800 class Missing(OrderedDict):
1801 def __missing__(self, key):
1802 return 0
1803 self.assertEqual(Missing().setdefault(5, 9), 9)
1804
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001805 def test_reinsert(self):
1806 # Given insert a, insert b, delete a, re-insert a,
1807 # verify that a is now later than b.
1808 od = OrderedDict()
1809 od['a'] = 1
1810 od['b'] = 2
1811 del od['a']
1812 od['a'] = 1
1813 self.assertEqual(list(od.items()), [('b', 2), ('a', 1)])
1814
Raymond Hettingerf45abc92010-09-06 21:26:09 +00001815 def test_move_to_end(self):
1816 od = OrderedDict.fromkeys('abcde')
1817 self.assertEqual(list(od), list('abcde'))
1818 od.move_to_end('c')
1819 self.assertEqual(list(od), list('abdec'))
1820 od.move_to_end('c', 0)
1821 self.assertEqual(list(od), list('cabde'))
1822 od.move_to_end('c', 0)
1823 self.assertEqual(list(od), list('cabde'))
1824 od.move_to_end('e')
1825 self.assertEqual(list(od), list('cabde'))
1826 with self.assertRaises(KeyError):
1827 od.move_to_end('x')
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001828
Raymond Hettinger35c87f22010-09-16 19:10:17 +00001829 def test_sizeof(self):
1830 # Wimpy test: Just verify the reported size is larger than a regular dict
1831 d = dict(a=1)
1832 od = OrderedDict(**d)
1833 self.assertGreater(sys.getsizeof(od), sys.getsizeof(d))
1834
Raymond Hettinger32062e92011-01-01 22:38:00 +00001835 def test_override_update(self):
1836 # Verify that subclasses can override update() without breaking __init__()
1837 class MyOD(OrderedDict):
1838 def update(self, *args, **kwds):
1839 raise Exception()
1840 items = [('a', 1), ('c', 3), ('b', 2)]
1841 self.assertEqual(list(MyOD(items).items()), items)
1842
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001843class GeneralMappingTests(mapping_tests.BasicTestMappingProtocol):
1844 type2test = OrderedDict
1845
Raymond Hettingerdc879f02009-03-19 20:30:56 +00001846 def test_popitem(self):
1847 d = self._empty_mapping()
1848 self.assertRaises(KeyError, d.popitem)
1849
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001850class MyOrderedDict(OrderedDict):
1851 pass
1852
1853class SubclassMappingTests(mapping_tests.BasicTestMappingProtocol):
1854 type2test = MyOrderedDict
1855
Raymond Hettingerdc879f02009-03-19 20:30:56 +00001856 def test_popitem(self):
1857 d = self._empty_mapping()
1858 self.assertRaises(KeyError, d.popitem)
Raymond Hettinger2d32f632009-03-02 21:24:57 +00001859
1860
Raymond Hettinger499e1932011-02-23 07:56:53 +00001861################################################################################
1862### Run tests
1863################################################################################
1864
Christian Heimes25bb7832008-01-11 16:17:00 +00001865import doctest, collections
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001866
Guido van Rossumd8faa362007-04-27 19:54:29 +00001867def test_main(verbose=None):
Benjamin Petersonad9d48d2008-04-02 21:49:44 +00001868 NamedTupleDocs = doctest.DocTestSuite(module=collections)
Raymond Hettingerb8baf632009-01-14 02:20:07 +00001869 test_classes = [TestNamedTuple, NamedTupleDocs, TestOneTrickPonyABCs,
Raymond Hettingerd0321312011-02-26 06:53:58 +00001870 TestCollectionABCs, TestCounter, TestChainMap,
Raymond Hettinger573b44c2015-05-22 16:56:32 -07001871 TestOrderedDict, GeneralMappingTests, SubclassMappingTests,
1872 TestUserObjects]
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001873 support.run_unittest(*test_classes)
1874 support.run_doctest(collections, verbose)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001875
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001876
Guido van Rossumd8faa362007-04-27 19:54:29 +00001877if __name__ == "__main__":
1878 test_main(verbose=True)