| Alexandre Vassalotti | 69eb516 | 2010-01-11 23:17:10 +0000 | [diff] [blame] | 1 | import unittest |
| 2 | from test import test_support |
| 3 | |
| 4 | class DictSetTest(unittest.TestCase): |
| 5 | |
| 6 | def test_constructors_not_callable(self): |
| 7 | kt = type({}.viewkeys()) |
| 8 | self.assertRaises(TypeError, kt, {}) |
| 9 | self.assertRaises(TypeError, kt) |
| 10 | it = type({}.viewitems()) |
| 11 | self.assertRaises(TypeError, it, {}) |
| 12 | self.assertRaises(TypeError, it) |
| 13 | vt = type({}.viewvalues()) |
| 14 | self.assertRaises(TypeError, vt, {}) |
| 15 | self.assertRaises(TypeError, vt) |
| 16 | |
| 17 | def test_dict_keys(self): |
| 18 | d = {1: 10, "a": "ABC"} |
| 19 | keys = d.viewkeys() |
| 20 | self.assertEqual(len(keys), 2) |
| 21 | self.assertEqual(set(keys), set([1, "a"])) |
| 22 | self.assertEqual(keys, set([1, "a"])) |
| 23 | self.assertNotEqual(keys, set([1, "a", "b"])) |
| 24 | self.assertNotEqual(keys, set([1, "b"])) |
| 25 | self.assertNotEqual(keys, set([1])) |
| 26 | self.assertNotEqual(keys, 42) |
| 27 | self.assert_(1 in keys) |
| 28 | self.assert_("a" in keys) |
| 29 | self.assert_(10 not in keys) |
| 30 | self.assert_("Z" not in keys) |
| 31 | self.assertEqual(d.viewkeys(), d.viewkeys()) |
| 32 | e = {1: 11, "a": "def"} |
| 33 | self.assertEqual(d.viewkeys(), e.viewkeys()) |
| 34 | del e["a"] |
| 35 | self.assertNotEqual(d.viewkeys(), e.viewkeys()) |
| 36 | |
| 37 | def test_dict_items(self): |
| 38 | d = {1: 10, "a": "ABC"} |
| 39 | items = d.viewitems() |
| 40 | self.assertEqual(len(items), 2) |
| 41 | self.assertEqual(set(items), set([(1, 10), ("a", "ABC")])) |
| 42 | self.assertEqual(items, set([(1, 10), ("a", "ABC")])) |
| 43 | self.assertNotEqual(items, set([(1, 10), ("a", "ABC"), "junk"])) |
| 44 | self.assertNotEqual(items, set([(1, 10), ("a", "def")])) |
| 45 | self.assertNotEqual(items, set([(1, 10)])) |
| 46 | self.assertNotEqual(items, 42) |
| 47 | self.assert_((1, 10) in items) |
| 48 | self.assert_(("a", "ABC") in items) |
| 49 | self.assert_((1, 11) not in items) |
| 50 | self.assert_(1 not in items) |
| 51 | self.assert_(() not in items) |
| 52 | self.assert_((1,) not in items) |
| 53 | self.assert_((1, 2, 3) not in items) |
| 54 | self.assertEqual(d.viewitems(), d.viewitems()) |
| 55 | e = d.copy() |
| 56 | self.assertEqual(d.viewitems(), e.viewitems()) |
| 57 | e["a"] = "def" |
| 58 | self.assertNotEqual(d.viewitems(), e.viewitems()) |
| 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.viewkeys(), e.viewitems()) |
| 64 | self.assertNotEqual(d.viewitems(), e.viewkeys()) |
| 65 | |
| 66 | def test_dict_values(self): |
| 67 | d = {1: 10, "a": "ABC"} |
| 68 | values = d.viewvalues() |
| 69 | self.assertEqual(set(values), set([10, "ABC"])) |
| 70 | self.assertEqual(len(values), 2) |
| 71 | |
| 72 | def test_dict_repr(self): |
| 73 | d = {1: 10, "a": "ABC"} |
| 74 | self.assertTrue(isinstance(repr(d), str)) |
| 75 | self.assertTrue(isinstance(repr(d.viewitems()), str)) |
| 76 | self.assertTrue(isinstance(repr(d.viewkeys()), str)) |
| 77 | self.assertTrue(isinstance(repr(d.viewvalues()), str)) |
| 78 | |
| 79 | |
| 80 | def test_main(): |
| 81 | test_support.run_unittest(DictSetTest) |
| 82 | |
| 83 | if __name__ == "__main__": |
| 84 | test_main() |