blob: 8d33801ca094b6ae0fbf806218392e06003392a2 [file] [log] [blame]
Guido van Rossumb90c8482007-02-10 01:11:45 +00001import unittest
Guido van Rossumb90c8482007-02-10 01:11:45 +00002
3class DictSetTest(unittest.TestCase):
4
Guido van Rossum3ac67412007-02-10 18:55:06 +00005 def test_constructors_not_callable(self):
Guido van Rossumcc2b0162007-02-11 06:12:03 +00006 kt = type({}.keys())
Guido van Rossum3ac67412007-02-10 18:55:06 +00007 self.assertRaises(TypeError, kt, {})
8 self.assertRaises(TypeError, kt)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00009 it = type({}.items())
Guido van Rossum3ac67412007-02-10 18:55:06 +000010 self.assertRaises(TypeError, it, {})
11 self.assertRaises(TypeError, it)
Guido van Rossumcc2b0162007-02-11 06:12:03 +000012 vt = type({}.values())
Guido van Rossum3ac67412007-02-10 18:55:06 +000013 self.assertRaises(TypeError, vt, {})
14 self.assertRaises(TypeError, vt)
15
Guido van Rossumb90c8482007-02-10 01:11:45 +000016 def test_dict_keys(self):
17 d = {1: 10, "a": "ABC"}
Guido van Rossumcc2b0162007-02-11 06:12:03 +000018 keys = d.keys()
Guido van Rossum83825ac2007-02-10 04:54:19 +000019 self.assertEqual(len(keys), 2)
Guido van Rossumd9214d12007-02-12 02:23:40 +000020 self.assertEqual(set(keys), {1, "a"})
21 self.assertEqual(keys, {1, "a"})
22 self.assertNotEqual(keys, {1, "a", "b"})
23 self.assertNotEqual(keys, {1, "b"})
24 self.assertNotEqual(keys, {1})
25 self.assertNotEqual(keys, 42)
Benjamin Peterson577473f2010-01-19 00:09:57 +000026 self.assertIn(1, keys)
27 self.assertIn("a", keys)
28 self.assertNotIn(10, keys)
29 self.assertNotIn("Z", keys)
Guido van Rossumd9214d12007-02-12 02:23:40 +000030 self.assertEqual(d.keys(), d.keys())
31 e = {1: 11, "a": "def"}
32 self.assertEqual(d.keys(), e.keys())
33 del e["a"]
34 self.assertNotEqual(d.keys(), e.keys())
Guido van Rossumb90c8482007-02-10 01:11:45 +000035
36 def test_dict_items(self):
37 d = {1: 10, "a": "ABC"}
Guido van Rossumcc2b0162007-02-11 06:12:03 +000038 items = d.items()
Guido van Rossum83825ac2007-02-10 04:54:19 +000039 self.assertEqual(len(items), 2)
Guido van Rossumd9214d12007-02-12 02:23:40 +000040 self.assertEqual(set(items), {(1, 10), ("a", "ABC")})
41 self.assertEqual(items, {(1, 10), ("a", "ABC")})
42 self.assertNotEqual(items, {(1, 10), ("a", "ABC"), "junk"})
43 self.assertNotEqual(items, {(1, 10), ("a", "def")})
44 self.assertNotEqual(items, {(1, 10)})
45 self.assertNotEqual(items, 42)
Benjamin Peterson577473f2010-01-19 00:09:57 +000046 self.assertIn((1, 10), items)
47 self.assertIn(("a", "ABC"), items)
48 self.assertNotIn((1, 11), items)
49 self.assertNotIn(1, items)
50 self.assertNotIn((), items)
51 self.assertNotIn((1,), items)
52 self.assertNotIn((1, 2, 3), items)
Guido van Rossumd9214d12007-02-12 02:23:40 +000053 self.assertEqual(d.items(), d.items())
54 e = d.copy()
55 self.assertEqual(d.items(), e.items())
56 e["a"] = "def"
57 self.assertNotEqual(d.items(), e.items())
58
59 def test_dict_mixed_keys_items(self):
60 d = {(1, 1): 11, (2, 2): 22}
61 e = {1: 1, 2: 2}
62 self.assertEqual(d.keys(), e.items())
63 self.assertNotEqual(d.items(), e.keys())
Guido van Rossumb90c8482007-02-10 01:11:45 +000064
65 def test_dict_values(self):
66 d = {1: 10, "a": "ABC"}
Guido van Rossumcc2b0162007-02-11 06:12:03 +000067 values = d.values()
Guido van Rossumb90c8482007-02-10 01:11:45 +000068 self.assertEqual(set(values), {10, "ABC"})
Guido van Rossum83825ac2007-02-10 04:54:19 +000069 self.assertEqual(len(values), 2)
Guido van Rossumb90c8482007-02-10 01:11:45 +000070
Alexandre Vassalottia7142572010-05-04 03:41:49 +000071 def test_dict_repr(self):
72 d = {1: 10, "a": "ABC"}
73 self.assertIsInstance(repr(d), str)
74 r = repr(d.items())
75 self.assertIsInstance(r, str)
76 self.assertTrue(r == "dict_items([('a', 'ABC'), (1, 10)])" or
77 r == "dict_items([(1, 10), ('a', 'ABC')])")
78 r = repr(d.keys())
79 self.assertIsInstance(r, str)
80 self.assertTrue(r == "dict_keys(['a', 1])" or
81 r == "dict_keys([1, 'a'])")
82 r = repr(d.values())
83 self.assertIsInstance(r, str)
84 self.assertTrue(r == "dict_values(['ABC', 10])" or
85 r == "dict_values([10, 'ABC'])")
86
87 def test_keys_set_operations(self):
88 d1 = {'a': 1, 'b': 2}
89 d2 = {'b': 3, 'c': 2}
90 d3 = {'d': 4, 'e': 5}
91 self.assertEqual(d1.keys() & d1.keys(), {'a', 'b'})
92 self.assertEqual(d1.keys() & d2.keys(), {'b'})
93 self.assertEqual(d1.keys() & d3.keys(), set())
94 self.assertEqual(d1.keys() & set(d1.keys()), {'a', 'b'})
95 self.assertEqual(d1.keys() & set(d2.keys()), {'b'})
96 self.assertEqual(d1.keys() & set(d3.keys()), set())
97
98 self.assertEqual(d1.keys() | d1.keys(), {'a', 'b'})
99 self.assertEqual(d1.keys() | d2.keys(), {'a', 'b', 'c'})
100 self.assertEqual(d1.keys() | d3.keys(), {'a', 'b', 'd', 'e'})
101 self.assertEqual(d1.keys() | set(d1.keys()), {'a', 'b'})
102 self.assertEqual(d1.keys() | set(d2.keys()), {'a', 'b', 'c'})
103 self.assertEqual(d1.keys() | set(d3.keys()),
104 {'a', 'b', 'd', 'e'})
105
106 self.assertEqual(d1.keys() ^ d1.keys(), set())
107 self.assertEqual(d1.keys() ^ d2.keys(), {'a', 'c'})
108 self.assertEqual(d1.keys() ^ d3.keys(), {'a', 'b', 'd', 'e'})
109 self.assertEqual(d1.keys() ^ set(d1.keys()), set())
110 self.assertEqual(d1.keys() ^ set(d2.keys()), {'a', 'c'})
111 self.assertEqual(d1.keys() ^ set(d3.keys()),
112 {'a', 'b', 'd', 'e'})
113
Ezio Melotti25326352013-08-08 20:12:28 +0300114 self.assertEqual(d1.keys() - d1.keys(), set())
115 self.assertEqual(d1.keys() - d2.keys(), {'a'})
116 self.assertEqual(d1.keys() - d3.keys(), {'a', 'b'})
117 self.assertEqual(d1.keys() - set(d1.keys()), set())
118 self.assertEqual(d1.keys() - set(d2.keys()), {'a'})
119 self.assertEqual(d1.keys() - set(d3.keys()), {'a', 'b'})
120
Daniel Stutzbach045b3ba2010-09-02 15:06:06 +0000121 self.assertFalse(d1.keys().isdisjoint(d1.keys()))
122 self.assertFalse(d1.keys().isdisjoint(d2.keys()))
123 self.assertFalse(d1.keys().isdisjoint(list(d2.keys())))
124 self.assertFalse(d1.keys().isdisjoint(set(d2.keys())))
125 self.assertTrue(d1.keys().isdisjoint({'x', 'y', 'z'}))
126 self.assertTrue(d1.keys().isdisjoint(['x', 'y', 'z']))
127 self.assertTrue(d1.keys().isdisjoint(set(['x', 'y', 'z'])))
128 self.assertTrue(d1.keys().isdisjoint(set(['x', 'y'])))
129 self.assertTrue(d1.keys().isdisjoint(['x', 'y']))
130 self.assertTrue(d1.keys().isdisjoint({}))
131 self.assertTrue(d1.keys().isdisjoint(d3.keys()))
132
133 de = {}
134 self.assertTrue(de.keys().isdisjoint(set()))
135 self.assertTrue(de.keys().isdisjoint([]))
136 self.assertTrue(de.keys().isdisjoint(de.keys()))
137 self.assertTrue(de.keys().isdisjoint([1]))
138
Alexandre Vassalottia7142572010-05-04 03:41:49 +0000139 def test_items_set_operations(self):
140 d1 = {'a': 1, 'b': 2}
141 d2 = {'a': 2, 'b': 2}
142 d3 = {'d': 4, 'e': 5}
143 self.assertEqual(
144 d1.items() & d1.items(), {('a', 1), ('b', 2)})
145 self.assertEqual(d1.items() & d2.items(), {('b', 2)})
146 self.assertEqual(d1.items() & d3.items(), set())
147 self.assertEqual(d1.items() & set(d1.items()),
148 {('a', 1), ('b', 2)})
149 self.assertEqual(d1.items() & set(d2.items()), {('b', 2)})
150 self.assertEqual(d1.items() & set(d3.items()), set())
151
152 self.assertEqual(d1.items() | d1.items(),
153 {('a', 1), ('b', 2)})
154 self.assertEqual(d1.items() | d2.items(),
155 {('a', 1), ('a', 2), ('b', 2)})
156 self.assertEqual(d1.items() | d3.items(),
157 {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
158 self.assertEqual(d1.items() | set(d1.items()),
159 {('a', 1), ('b', 2)})
160 self.assertEqual(d1.items() | set(d2.items()),
161 {('a', 1), ('a', 2), ('b', 2)})
162 self.assertEqual(d1.items() | set(d3.items()),
163 {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
164
165 self.assertEqual(d1.items() ^ d1.items(), set())
166 self.assertEqual(d1.items() ^ d2.items(),
167 {('a', 1), ('a', 2)})
168 self.assertEqual(d1.items() ^ d3.items(),
169 {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
170
Ezio Melotti25326352013-08-08 20:12:28 +0300171 self.assertEqual(d1.items() - d1.items(), set())
172 self.assertEqual(d1.items() - d2.items(), {('a', 1)})
173 self.assertEqual(d1.items() - d3.items(), {('a', 1), ('b', 2)})
174 self.assertEqual(d1.items() - set(d1.items()), set())
175 self.assertEqual(d1.items() - set(d2.items()), {('a', 1)})
176 self.assertEqual(d1.items() - set(d3.items()), {('a', 1), ('b', 2)})
177
Daniel Stutzbach045b3ba2010-09-02 15:06:06 +0000178 self.assertFalse(d1.items().isdisjoint(d1.items()))
179 self.assertFalse(d1.items().isdisjoint(d2.items()))
180 self.assertFalse(d1.items().isdisjoint(list(d2.items())))
181 self.assertFalse(d1.items().isdisjoint(set(d2.items())))
182 self.assertTrue(d1.items().isdisjoint({'x', 'y', 'z'}))
183 self.assertTrue(d1.items().isdisjoint(['x', 'y', 'z']))
184 self.assertTrue(d1.items().isdisjoint(set(['x', 'y', 'z'])))
185 self.assertTrue(d1.items().isdisjoint(set(['x', 'y'])))
186 self.assertTrue(d1.items().isdisjoint({}))
187 self.assertTrue(d1.items().isdisjoint(d3.items()))
188
189 de = {}
190 self.assertTrue(de.items().isdisjoint(set()))
191 self.assertTrue(de.items().isdisjoint([]))
192 self.assertTrue(de.items().isdisjoint(de.items()))
193 self.assertTrue(de.items().isdisjoint([1]))
194
Benjamin Peterson611d9012013-05-19 19:39:38 -0700195 def test_recursive_repr(self):
196 d = {}
197 d[42] = d.values()
Yury Selivanovf488fb42015-07-03 01:04:23 -0400198 self.assertRaises(RecursionError, repr, d)
Benjamin Peterson611d9012013-05-19 19:39:38 -0700199
Alexandre Vassalottia7142572010-05-04 03:41:49 +0000200
Guido van Rossumb90c8482007-02-10 01:11:45 +0000201if __name__ == "__main__":
Zachary Ware38c707e2015-04-13 15:00:43 -0500202 unittest.main()