Mercurial > cpython-lru_cache-weakref
annotate Lib/test/test_defaultdict.py @ 54294:dba767bc3155 legacy-trunk
remove mention of old ctypes version
| author | Benjamin Peterson <benjamin@python.org> |
|---|---|
| date | Sat, 23 May 2009 20:59:09 +0000 |
| parents | e94af0c7aa12 |
| children | dd4da75ebc97 6768b2350c57 |
| rev | line source |
|---|---|
| 36656 | 1 """Unit tests for collections.defaultdict.""" |
| 2 | |
| 3 import os | |
| 4 import copy | |
| 5 import tempfile | |
| 6 import unittest | |
|
39206
025c27b812f6
Add test_main() methods. These three tests were never run
Georg Brandl <georg@python.org>
parents:
36656
diff
changeset
|
7 from test import test_support |
| 36656 | 8 |
| 9 from collections import defaultdict | |
| 10 | |
| 11 def foobar(): | |
| 12 return list | |
| 13 | |
| 14 class TestDefaultDict(unittest.TestCase): | |
| 15 | |
| 16 def test_basic(self): | |
| 17 d1 = defaultdict() | |
| 18 self.assertEqual(d1.default_factory, None) | |
| 19 d1.default_factory = list | |
| 20 d1[12].append(42) | |
| 21 self.assertEqual(d1, {12: [42]}) | |
| 22 d1[12].append(24) | |
| 23 self.assertEqual(d1, {12: [42, 24]}) | |
| 24 d1[13] | |
| 25 d1[14] | |
| 26 self.assertEqual(d1, {12: [42, 24], 13: [], 14: []}) | |
| 27 self.assert_(d1[12] is not d1[13] is not d1[14]) | |
| 28 d2 = defaultdict(list, foo=1, bar=2) | |
| 29 self.assertEqual(d2.default_factory, list) | |
| 30 self.assertEqual(d2, {"foo": 1, "bar": 2}) | |
| 31 self.assertEqual(d2["foo"], 1) | |
| 32 self.assertEqual(d2["bar"], 2) | |
| 33 self.assertEqual(d2[42], []) | |
| 34 self.assert_("foo" in d2) | |
| 35 self.assert_("foo" in d2.keys()) | |
| 36 self.assert_("bar" in d2) | |
| 37 self.assert_("bar" in d2.keys()) | |
| 38 self.assert_(42 in d2) | |
| 39 self.assert_(42 in d2.keys()) | |
| 40 self.assert_(12 not in d2) | |
| 41 self.assert_(12 not in d2.keys()) | |
| 42 d2.default_factory = None | |
| 43 self.assertEqual(d2.default_factory, None) | |
| 44 try: | |
| 45 d2[15] | |
| 46 except KeyError, err: | |
| 47 self.assertEqual(err.args, (15,)) | |
| 48 else: | |
| 49 self.fail("d2[15] didn't raise KeyError") | |
|
40760
2900c7564055
Check for a common user error with defaultdict().
Raymond Hettinger <python@rcn.com>
parents:
39206
diff
changeset
|
50 self.assertRaises(TypeError, defaultdict, 1) |
| 36656 | 51 |
| 52 def test_missing(self): | |
| 53 d1 = defaultdict() | |
| 54 self.assertRaises(KeyError, d1.__missing__, 42) | |
| 55 d1.default_factory = list | |
| 56 self.assertEqual(d1.__missing__(42), []) | |
| 57 | |
| 58 def test_repr(self): | |
| 59 d1 = defaultdict() | |
| 60 self.assertEqual(d1.default_factory, None) | |
| 61 self.assertEqual(repr(d1), "defaultdict(None, {})") | |
| 62 d1[11] = 41 | |
| 63 self.assertEqual(repr(d1), "defaultdict(None, {11: 41})") | |
|
40760
2900c7564055
Check for a common user error with defaultdict().
Raymond Hettinger <python@rcn.com>
parents:
39206
diff
changeset
|
64 d2 = defaultdict(int) |
|
2900c7564055
Check for a common user error with defaultdict().
Raymond Hettinger <python@rcn.com>
parents:
39206
diff
changeset
|
65 self.assertEqual(d2.default_factory, int) |
| 36656 | 66 d2[12] = 42 |
|
40760
2900c7564055
Check for a common user error with defaultdict().
Raymond Hettinger <python@rcn.com>
parents:
39206
diff
changeset
|
67 self.assertEqual(repr(d2), "defaultdict(<type 'int'>, {12: 42})") |
| 36656 | 68 def foo(): return 43 |
| 69 d3 = defaultdict(foo) | |
| 70 self.assert_(d3.default_factory is foo) | |
| 71 d3[13] | |
| 72 self.assertEqual(repr(d3), "defaultdict(%s, {13: 43})" % repr(foo)) | |
| 73 | |
| 74 def test_print(self): | |
| 75 d1 = defaultdict() | |
| 76 def foo(): return 42 | |
| 77 d2 = defaultdict(foo, {1: 2}) | |
| 78 # NOTE: We can't use tempfile.[Named]TemporaryFile since this | |
| 79 # code must exercise the tp_print C code, which only gets | |
| 80 # invoked for *real* files. | |
| 81 tfn = tempfile.mktemp() | |
| 82 try: | |
| 83 f = open(tfn, "w+") | |
| 84 try: | |
| 85 print >>f, d1 | |
| 86 print >>f, d2 | |
| 87 f.seek(0) | |
| 88 self.assertEqual(f.readline(), repr(d1) + "\n") | |
| 89 self.assertEqual(f.readline(), repr(d2) + "\n") | |
| 90 finally: | |
| 91 f.close() | |
| 92 finally: | |
| 93 os.remove(tfn) | |
| 94 | |
| 95 def test_copy(self): | |
| 96 d1 = defaultdict() | |
| 97 d2 = d1.copy() | |
| 98 self.assertEqual(type(d2), defaultdict) | |
| 99 self.assertEqual(d2.default_factory, None) | |
| 100 self.assertEqual(d2, {}) | |
| 101 d1.default_factory = list | |
| 102 d3 = d1.copy() | |
| 103 self.assertEqual(type(d3), defaultdict) | |
| 104 self.assertEqual(d3.default_factory, list) | |
| 105 self.assertEqual(d3, {}) | |
| 106 d1[42] | |
| 107 d4 = d1.copy() | |
| 108 self.assertEqual(type(d4), defaultdict) | |
| 109 self.assertEqual(d4.default_factory, list) | |
| 110 self.assertEqual(d4, {42: []}) | |
| 111 d4[12] | |
| 112 self.assertEqual(d4, {42: [], 12: []}) | |
| 113 | |
| 114 def test_shallow_copy(self): | |
| 115 d1 = defaultdict(foobar, {1: 1}) | |
| 116 d2 = copy.copy(d1) | |
| 117 self.assertEqual(d2.default_factory, foobar) | |
| 118 self.assertEqual(d2, d1) | |
| 119 d1.default_factory = list | |
| 120 d2 = copy.copy(d1) | |
| 121 self.assertEqual(d2.default_factory, list) | |
| 122 self.assertEqual(d2, d1) | |
| 123 | |
| 124 def test_deep_copy(self): | |
| 125 d1 = defaultdict(foobar, {1: [1]}) | |
| 126 d2 = copy.deepcopy(d1) | |
| 127 self.assertEqual(d2.default_factory, foobar) | |
| 128 self.assertEqual(d2, d1) | |
| 129 self.assert_(d1[1] is not d2[1]) | |
| 130 d1.default_factory = list | |
| 131 d2 = copy.deepcopy(d1) | |
| 132 self.assertEqual(d2.default_factory, list) | |
| 133 self.assertEqual(d2, d1) | |
| 134 | |
|
41021
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
135 def test_keyerror_without_factory(self): |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
136 d1 = defaultdict() |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
137 try: |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
138 d1[(1,)] |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
139 except KeyError, err: |
|
41617
615795d52b9f
Deprecate BaseException.message as per PEP 352.
Brett Cannon <bcannon@gmail.com>
parents:
41021
diff
changeset
|
140 self.assertEqual(err.args[0], (1,)) |
|
41021
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
141 else: |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
142 self.fail("expected KeyError") |
|
09d08fcb8f7b
A test case for the defaultdict KeyError bug.
Georg Brandl <georg@python.org>
parents:
40760
diff
changeset
|
143 |
|
45231
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
144 def test_recursive_repr(self): |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
145 # Issue2045: stack overflow when default_factory is a bound method |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
146 class sub(defaultdict): |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
147 def __init__(self): |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
148 self.default_factory = self._factory |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
149 def _factory(self): |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
150 return [] |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
151 d = sub() |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
152 self.assert_(repr(d).startswith( |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
153 "defaultdict(<bound method sub._factory of defaultdict(...")) |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
154 |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
155 # NOTE: printing a subclass of a builtin type does not call its |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
156 # tp_print slot. So this part is essentially the same test as above. |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
157 tfn = tempfile.mktemp() |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
158 try: |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
159 f = open(tfn, "w+") |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
160 try: |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
161 print >>f, d |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
162 finally: |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
163 f.close() |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
164 finally: |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
165 os.remove(tfn) |
|
e94af0c7aa12
issue 2045: Infinite recursion when printing a subclass of defaultdict,
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents:
41617
diff
changeset
|
166 |
| 36656 | 167 |
|
39206
025c27b812f6
Add test_main() methods. These three tests were never run
Georg Brandl <georg@python.org>
parents:
36656
diff
changeset
|
168 def test_main(): |
|
025c27b812f6
Add test_main() methods. These three tests were never run
Georg Brandl <georg@python.org>
parents:
36656
diff
changeset
|
169 test_support.run_unittest(TestDefaultDict) |
|
025c27b812f6
Add test_main() methods. These three tests were never run
Georg Brandl <georg@python.org>
parents:
36656
diff
changeset
|
170 |
| 36656 | 171 if __name__ == "__main__": |
|
39206
025c27b812f6
Add test_main() methods. These three tests were never run
Georg Brandl <georg@python.org>
parents:
36656
diff
changeset
|
172 test_main() |