Lib/test/test_dictviews.py

   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.assertIn(1, keys)
  28         self.assertIn("a", keys)
  29         self.assertNotIn(10, keys)
  30         self.assertNotIn("Z", 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.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)
  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.assertIsInstance(repr(d), str)
  75         r = repr(d.viewitems())
  76         self.assertIsInstance(r, str)
  77         self.assertTrue(r == "dict_items([('a', 'ABC'), (1, 10)])" or
  78                         r == "dict_items([(1, 10), ('a', 'ABC')])")
  79         r = repr(d.viewkeys())
  80         self.assertIsInstance(r, str)
  81         self.assertTrue(r == "dict_keys(['a', 1])" or
  82                         r == "dict_keys([1, 'a'])")
  83         r = repr(d.viewvalues())
  84         self.assertIsInstance(r, str)
  85         self.assertTrue(r == "dict_values(['ABC', 10])" or
  86                         r == "dict_values([10, 'ABC'])")
  87
  88     def test_keys_set_operations(self):
  89         d1 = {'a': 1, 'b': 2}
  90         d2 = {'b': 3, 'c': 2}
  91         d3 = {'d': 4, 'e': 5}
  92         self.assertEqual(d1.viewkeys() & d1.viewkeys(), {'a', 'b'})
  93         self.assertEqual(d1.viewkeys() & d2.viewkeys(), {'b'})
  94         self.assertEqual(d1.viewkeys() & d3.viewkeys(), set())
  95         self.assertEqual(d1.viewkeys() & set(d1.viewkeys()), {'a', 'b'})
  96         self.assertEqual(d1.viewkeys() & set(d2.viewkeys()), {'b'})
  97         self.assertEqual(d1.viewkeys() & set(d3.viewkeys()), set())
  98
  99         self.assertEqual(d1.viewkeys() | d1.viewkeys(), {'a', 'b'})
 100         self.assertEqual(d1.viewkeys() | d2.viewkeys(), {'a', 'b', 'c'})
 101         self.assertEqual(d1.viewkeys() | d3.viewkeys(), {'a', 'b', 'd', 'e'})
 102         self.assertEqual(d1.viewkeys() | set(d1.viewkeys()), {'a', 'b'})
 103         self.assertEqual(d1.viewkeys() | set(d2.viewkeys()), {'a', 'b', 'c'})
 104         self.assertEqual(d1.viewkeys() | set(d3.viewkeys()),
 105                          {'a', 'b', 'd', 'e'})
 106
 107         self.assertEqual(d1.viewkeys() ^ d1.viewkeys(), set())
 108         self.assertEqual(d1.viewkeys() ^ d2.viewkeys(), {'a', 'c'})
 109         self.assertEqual(d1.viewkeys() ^ d3.viewkeys(), {'a', 'b', 'd', 'e'})
 110         self.assertEqual(d1.viewkeys() ^ set(d1.viewkeys()), set())
 111         self.assertEqual(d1.viewkeys() ^ set(d2.viewkeys()), {'a', 'c'})
 112         self.assertEqual(d1.viewkeys() ^ set(d3.viewkeys()),
 113                          {'a', 'b', 'd', 'e'})
 114
 115     def test_items_set_operations(self):
 116         d1 = {'a': 1, 'b': 2}
 117         d2 = {'a': 2, 'b': 2}
 118         d3 = {'d': 4, 'e': 5}
 119         self.assertEqual(
 120             d1.viewitems() & d1.viewitems(), {('a', 1), ('b', 2)})
 121         self.assertEqual(d1.viewitems() & d2.viewitems(), {('b', 2)})
 122         self.assertEqual(d1.viewitems() & d3.viewitems(), set())
 123         self.assertEqual(d1.viewitems() & set(d1.viewitems()),
 124                          {('a', 1), ('b', 2)})
 125         self.assertEqual(d1.viewitems() & set(d2.viewitems()), {('b', 2)})
 126         self.assertEqual(d1.viewitems() & set(d3.viewitems()), set())
 127
 128         self.assertEqual(d1.viewitems() | d1.viewitems(),
 129                          {('a', 1), ('b', 2)})
 130         self.assertEqual(d1.viewitems() | d2.viewitems(),
 131                          {('a', 1), ('a', 2), ('b', 2)})
 132         self.assertEqual(d1.viewitems() | d3.viewitems(),
 133                          {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
 134         self.assertEqual(d1.viewitems() | set(d1.viewitems()),
 135                          {('a', 1), ('b', 2)})
 136         self.assertEqual(d1.viewitems() | set(d2.viewitems()),
 137                          {('a', 1), ('a', 2), ('b', 2)})
 138         self.assertEqual(d1.viewitems() | set(d3.viewitems()),
 139                          {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
 140
 141         self.assertEqual(d1.viewitems() ^ d1.viewitems(), set())
 142         self.assertEqual(d1.viewitems() ^ d2.viewitems(),
 143                          {('a', 1), ('a', 2)})
 144         self.assertEqual(d1.viewitems() ^ d3.viewitems(),
 145                          {('a', 1), ('b', 2), ('d', 4), ('e', 5)})
 146
 147
 148
 149
 150 def test_main():
 151     test_support.run_unittest(DictSetTest)
 152
 153 if __name__ == "__main__":
 154     test_main()