Lib/test/test_contains.py

   1 from test.test_support import TestFailed, have_unicode
   2
   3 class base_set:
   4
   5     def __init__(self, el):
   6         self.el = el
   7
   8 class set(base_set):
   9
  10     def __contains__(self, el):
  11         return self.el == el
  12
  13 class seq(base_set):
  14
  15     def __getitem__(self, n):
  16         return [self.el][n]
  17
  18 def check(ok, *args):
  19     if not ok:
  20         raise TestFailed, " ".join(map(str, args))
  21
  22 a = base_set(1)
  23 b = set(1)
  24 c = seq(1)
  25
  26 check(1 in b, "1 not in set(1)")
  27 check(0 not in b, "0 in set(1)")
  28 check(1 in c, "1 not in seq(1)")
  29 check(0 not in c, "0 in seq(1)")
  30
  31 try:
  32     1 in a
  33     check(0, "in base_set did not raise error")
  34 except TypeError:
  35     pass
  36
  37 try:
  38     1 not in a
  39     check(0, "not in base_set did not raise error")
  40 except TypeError:
  41     pass
  42
  43 # Test char in string
  44
  45 check('c' in 'abc', "'c' not in 'abc'")
  46 check('d' not in 'abc', "'d' in 'abc'")
  47
  48 check('' in '', "'' not in ''")
  49 check('' in 'abc', "'' not in 'abc'")
  50
  51 try:
  52     None in 'abc'
  53     check(0, "None in 'abc' did not raise error")
  54 except TypeError:
  55     pass
  56
  57
  58 if have_unicode:
  59
  60     # Test char in Unicode
  61
  62     check('c' in unicode('abc'), "'c' not in u'abc'")
  63     check('d' not in unicode('abc'), "'d' in u'abc'")
  64
  65     check('' in unicode(''), "'' not in u''")
  66     check(unicode('') in '', "u'' not in ''")
  67     check(unicode('') in unicode(''), "u'' not in u''")
  68     check('' in unicode('abc'), "'' not in u'abc'")
  69     check(unicode('') in 'abc', "u'' not in 'abc'")
  70     check(unicode('') in unicode('abc'), "u'' not in u'abc'")
  71
  72     try:
  73         None in unicode('abc')
  74         check(0, "None in u'abc' did not raise error")
  75     except TypeError:
  76         pass
  77
  78     # Test Unicode char in Unicode
  79
  80     check(unicode('c') in unicode('abc'), "u'c' not in u'abc'")
  81     check(unicode('d') not in unicode('abc'), "u'd' in u'abc'")
  82
  83     # Test Unicode char in string
  84
  85     check(unicode('c') in 'abc', "u'c' not in 'abc'")
  86     check(unicode('d') not in 'abc', "u'd' in 'abc'")
  87
  88 # A collection of tests on builtin sequence types
  89 a = range(10)
  90 for i in a:
  91     check(i in a, "%s not in %s" % (`i`, `a`))
  92 check(16 not in a, "16 not in %s" % `a`)
  93 check(a not in a, "%s not in %s" % (`a`, `a`))
  94
  95 a = tuple(a)
  96 for i in a:
  97     check(i in a, "%s not in %s" % (`i`, `a`))
  98 check(16 not in a, "16 not in %s" % `a`)
  99 check(a not in a, "%s not in %s" % (`a`, `a`))
 100
 101 class Deviant1:
 102     """Behaves strangely when compared
 103
 104     This class is designed to make sure that the contains code
 105     works when the list is modified during the check.
 106     """
 107
 108     aList = range(15)
 109
 110     def __cmp__(self, other):
 111         if other == 12:
 112             self.aList.remove(12)
 113             self.aList.remove(13)
 114             self.aList.remove(14)
 115         return 1
 116
 117 check(Deviant1() not in Deviant1.aList, "Deviant1 failed")
 118
 119 class Deviant2:
 120     """Behaves strangely when compared
 121
 122     This class raises an exception during comparison.  That in
 123     turn causes the comparison to fail with a TypeError.
 124     """
 125
 126     def __cmp__(self, other):
 127         if other == 4:
 128             raise RuntimeError, "gotcha"
 129
 130 try:
 131     check(Deviant2() not in a, "oops")
 132 except TypeError:
 133     pass