Lib/test/test_userdict.py

   1 # Check every path through every method of UserDict
   2
   3 from test_support import verify, verbose
   4 from UserDict import UserDict
   5
   6 d0 = {}
   7 d1 = {"one": 1}
   8 d2 = {"one": 1, "two": 2}
   9
  10 # Test constructors
  11
  12 u = UserDict()
  13 u0 = UserDict(d0)
  14 u1 = UserDict(d1)
  15 u2 = UserDict(d2)
  16
  17 uu = UserDict(u)
  18 uu0 = UserDict(u0)
  19 uu1 = UserDict(u1)
  20 uu2 = UserDict(u2)
  21
  22 # Test __repr__
  23
  24 verify(str(u0) == str(d0))
  25 verify(repr(u1) == repr(d1))
  26 verify(`u2` == `d2`)
  27
  28 # Test __cmp__ and __len__
  29
  30 all = [d0, d1, d2, u, u0, u1, u2, uu, uu0, uu1, uu2]
  31 for a in all:
  32     for b in all:
  33         verify(cmp(a, b) == cmp(len(a), len(b)))
  34
  35 # Test __getitem__
  36
  37 verify(u2["one"] == 1)
  38 try:
  39     u1["two"]
  40 except KeyError:
  41     pass
  42 else:
  43     verify(0, "u1['two'] shouldn't exist")
  44
  45 # Test __setitem__
  46
  47 u3 = UserDict(u2)
  48 u3["two"] = 2
  49 u3["three"] = 3
  50
  51 # Test __delitem__
  52
  53 del u3["three"]
  54 try:
  55     del u3["three"]
  56 except KeyError:
  57     pass
  58 else:
  59     verify(0, "u3['three'] shouldn't exist")
  60
  61 # Test clear
  62
  63 u3.clear()
  64 verify(u3 == {})
  65
  66 # Test copy()
  67
  68 u2a = u2.copy()
  69 verify(u2a == u2)
  70
  71 class MyUserDict(UserDict):
  72     def display(self): print self
  73
  74 m2 = MyUserDict(u2)
  75 m2a = m2.copy()
  76 verify(m2a == m2)
  77
  78 # Test keys, items, values
  79
  80 verify(u2.keys() == d2.keys())
  81 verify(u2.items() == d2.items())
  82 verify(u2.values() == d2.values())
  83
  84 # Test has_key and "in".
  85
  86 for i in u2.keys():
  87     verify(u2.has_key(i) == 1)
  88     verify((i in u2) == 1)
  89     verify(u1.has_key(i) == d1.has_key(i))
  90     verify((i in u1) == (i in d1))
  91     verify(u0.has_key(i) == d0.has_key(i))
  92     verify((i in u0) == (i in d0))
  93
  94 # Test update
  95
  96 t = UserDict()
  97 t.update(u2)
  98 verify(t == u2)
  99
 100 # Test get
 101
 102 for i in u2.keys():
 103     verify(u2.get(i) == u2[i])
 104     verify(u1.get(i) == d1.get(i))
 105     verify(u0.get(i) == d0.get(i))
 106
 107 # Test "in" iteration.
 108 for i in xrange(20):
 109     u2[i] = str(i)
 110 ikeys = []
 111 for k in u2:
 112     ikeys.append(k)
 113 ikeys.sort()
 114 keys = u2.keys()
 115 keys.sort()
 116 verify(ikeys == keys)