Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 1 | import unittest |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 2 | from test import support |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 3 | |
| 4 | class DictSetTest(unittest.TestCase): |
| 5 | |
Guido van Rossum | 3ac6741 | 2007-02-10 18:55:06 +0000 | [diff] [blame] | 6 | def test_constructors_not_callable(self): |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 7 | kt = type({}.keys()) |
Guido van Rossum | 3ac6741 | 2007-02-10 18:55:06 +0000 | [diff] [blame] | 8 | self.assertRaises(TypeError, kt, {}) |
| 9 | self.assertRaises(TypeError, kt) |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 10 | it = type({}.items()) |
Guido van Rossum | 3ac6741 | 2007-02-10 18:55:06 +0000 | [diff] [blame] | 11 | self.assertRaises(TypeError, it, {}) |
| 12 | self.assertRaises(TypeError, it) |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 13 | vt = type({}.values()) |
Guido van Rossum | 3ac6741 | 2007-02-10 18:55:06 +0000 | [diff] [blame] | 14 | self.assertRaises(TypeError, vt, {}) |
| 15 | self.assertRaises(TypeError, vt) |
| 16 | |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 17 | def test_dict_keys(self): |
| 18 | d = {1: 10, "a": "ABC"} |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 19 | keys = d.keys() |
Guido van Rossum | 83825ac | 2007-02-10 04:54:19 +0000 | [diff] [blame] | 20 | self.assertEqual(len(keys), 2) |
Guido van Rossum | d9214d1 | 2007-02-12 02:23:40 +0000 | [diff] [blame] | 21 | self.assertEqual(set(keys), {1, "a"}) |
| 22 | self.assertEqual(keys, {1, "a"}) |
| 23 | self.assertNotEqual(keys, {1, "a", "b"}) |
| 24 | self.assertNotEqual(keys, {1, "b"}) |
| 25 | self.assertNotEqual(keys, {1}) |
| 26 | self.assertNotEqual(keys, 42) |
Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 27 | self.assertIn(1, keys) |
| 28 | self.assertIn("a", keys) |
| 29 | self.assertNotIn(10, keys) |
| 30 | self.assertNotIn("Z", keys) |
Guido van Rossum | d9214d1 | 2007-02-12 02:23:40 +0000 | [diff] [blame] | 31 | self.assertEqual(d.keys(), d.keys()) |
| 32 | e = {1: 11, "a": "def"} |
| 33 | self.assertEqual(d.keys(), e.keys()) |
| 34 | del e["a"] |
| 35 | self.assertNotEqual(d.keys(), e.keys()) |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 36 | |
| 37 | def test_dict_items(self): |
| 38 | d = {1: 10, "a": "ABC"} |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 39 | items = d.items() |
Guido van Rossum | 83825ac | 2007-02-10 04:54:19 +0000 | [diff] [blame] | 40 | self.assertEqual(len(items), 2) |
Guido van Rossum | d9214d1 | 2007-02-12 02:23:40 +0000 | [diff] [blame] | 41 | self.assertEqual(set(items), {(1, 10), ("a", "ABC")}) |
| 42 | self.assertEqual(items, {(1, 10), ("a", "ABC")}) |
| 43 | self.assertNotEqual(items, {(1, 10), ("a", "ABC"), "junk"}) |
| 44 | self.assertNotEqual(items, {(1, 10), ("a", "def")}) |
| 45 | self.assertNotEqual(items, {(1, 10)}) |
| 46 | self.assertNotEqual(items, 42) |
Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 47 | self.assertIn((1, 10), items) |
| 48 | self.assertIn(("a", "ABC"), items) |
| 49 | self.assertNotIn((1, 11), items) |
| 50 | self.assertNotIn(1, items) |
| 51 | self.assertNotIn((), items) |
| 52 | self.assertNotIn((1,), items) |
| 53 | self.assertNotIn((1, 2, 3), items) |
Guido van Rossum | d9214d1 | 2007-02-12 02:23:40 +0000 | [diff] [blame] | 54 | self.assertEqual(d.items(), d.items()) |
| 55 | e = d.copy() |
| 56 | self.assertEqual(d.items(), e.items()) |
| 57 | e["a"] = "def" |
| 58 | self.assertNotEqual(d.items(), e.items()) |
| 59 | |
| 60 | def test_dict_mixed_keys_items(self): |
| 61 | d = {(1, 1): 11, (2, 2): 22} |
| 62 | e = {1: 1, 2: 2} |
| 63 | self.assertEqual(d.keys(), e.items()) |
| 64 | self.assertNotEqual(d.items(), e.keys()) |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 65 | |
| 66 | def test_dict_values(self): |
| 67 | d = {1: 10, "a": "ABC"} |
Guido van Rossum | cc2b016 | 2007-02-11 06:12:03 +0000 | [diff] [blame] | 68 | values = d.values() |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 69 | self.assertEqual(set(values), {10, "ABC"}) |
Guido van Rossum | 83825ac | 2007-02-10 04:54:19 +0000 | [diff] [blame] | 70 | self.assertEqual(len(values), 2) |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 71 | |
| 72 | def test_main(): |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 73 | support.run_unittest(DictSetTest) |
Guido van Rossum | b90c848 | 2007-02-10 01:11:45 +0000 | [diff] [blame] | 74 | |
| 75 | if __name__ == "__main__": |
| 76 | test_main() |