|
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, "%r not in %r" % (i, a)) |
|
92 check(16 not in a, "16 not in %r" % (a,)) |
|
93 check(a not in a, "%s not in %r" % (a, a)) |
|
94 |
|
95 a = tuple(a) |
|
96 for i in a: |
|
97 check(i in a, "%r not in %r" % (i, a)) |
|
98 check(16 not in a, "16 not in %r" % (a,)) |
|
99 check(a not in a, "%r not in %r" % (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 |