1 | from test.test_support import have_unicode, run_unittest |
---|
2 | import unittest |
---|
3 | |
---|
4 | |
---|
5 | class base_set: |
---|
6 | def __init__(self, el): |
---|
7 | self.el = el |
---|
8 | |
---|
9 | class set(base_set): |
---|
10 | def __contains__(self, el): |
---|
11 | return self.el == el |
---|
12 | |
---|
13 | class seq(base_set): |
---|
14 | def __getitem__(self, n): |
---|
15 | return [self.el][n] |
---|
16 | |
---|
17 | |
---|
18 | class TestContains(unittest.TestCase): |
---|
19 | def test_common_tests(self): |
---|
20 | a = base_set(1) |
---|
21 | b = set(1) |
---|
22 | c = seq(1) |
---|
23 | self.assertIn(1, b) |
---|
24 | self.assertNotIn(0, b) |
---|
25 | self.assertIn(1, c) |
---|
26 | self.assertNotIn(0, c) |
---|
27 | self.assertRaises(TypeError, lambda: 1 in a) |
---|
28 | self.assertRaises(TypeError, lambda: 1 not in a) |
---|
29 | |
---|
30 | # test char in string |
---|
31 | self.assertIn('c', 'abc') |
---|
32 | self.assertNotIn('d', 'abc') |
---|
33 | |
---|
34 | self.assertIn('', '') |
---|
35 | self.assertIn('', 'abc') |
---|
36 | |
---|
37 | self.assertRaises(TypeError, lambda: None in 'abc') |
---|
38 | |
---|
39 | if have_unicode: |
---|
40 | def test_char_in_unicode(self): |
---|
41 | self.assertIn('c', unicode('abc')) |
---|
42 | self.assertNotIn('d', unicode('abc')) |
---|
43 | |
---|
44 | self.assertIn('', unicode('')) |
---|
45 | self.assertIn(unicode(''), '') |
---|
46 | self.assertIn(unicode(''), unicode('')) |
---|
47 | self.assertIn('', unicode('abc')) |
---|
48 | self.assertIn(unicode(''), 'abc') |
---|
49 | self.assertIn(unicode(''), unicode('abc')) |
---|
50 | |
---|
51 | self.assertRaises(TypeError, lambda: None in unicode('abc')) |
---|
52 | |
---|
53 | # test Unicode char in Unicode |
---|
54 | self.assertIn(unicode('c'), unicode('abc')) |
---|
55 | self.assertNotIn(unicode('d'), unicode('abc')) |
---|
56 | |
---|
57 | # test Unicode char in string |
---|
58 | self.assertIn(unicode('c'), 'abc') |
---|
59 | self.assertNotIn(unicode('d'), 'abc') |
---|
60 | |
---|
61 | def test_builtin_sequence_types(self): |
---|
62 | # a collection of tests on builtin sequence types |
---|
63 | a = range(10) |
---|
64 | for i in a: |
---|
65 | self.assertIn(i, a) |
---|
66 | self.assertNotIn(16, a) |
---|
67 | self.assertNotIn(a, a) |
---|
68 | |
---|
69 | a = tuple(a) |
---|
70 | for i in a: |
---|
71 | self.assertIn(i, a) |
---|
72 | self.assertNotIn(16, a) |
---|
73 | self.assertNotIn(a, a) |
---|
74 | |
---|
75 | class Deviant1: |
---|
76 | """Behaves strangely when compared |
---|
77 | |
---|
78 | This class is designed to make sure that the contains code |
---|
79 | works when the list is modified during the check. |
---|
80 | """ |
---|
81 | aList = range(15) |
---|
82 | def __cmp__(self, other): |
---|
83 | if other == 12: |
---|
84 | self.aList.remove(12) |
---|
85 | self.aList.remove(13) |
---|
86 | self.aList.remove(14) |
---|
87 | return 1 |
---|
88 | |
---|
89 | self.assertNotIn(Deviant1(), Deviant1.aList) |
---|
90 | |
---|
91 | class Deviant2: |
---|
92 | """Behaves strangely when compared |
---|
93 | |
---|
94 | This class raises an exception during comparison. That in |
---|
95 | turn causes the comparison to fail with a TypeError. |
---|
96 | """ |
---|
97 | def __cmp__(self, other): |
---|
98 | if other == 4: |
---|
99 | raise RuntimeError, "gotcha" |
---|
100 | |
---|
101 | try: |
---|
102 | self.assertNotIn(Deviant2(), a) |
---|
103 | except TypeError: |
---|
104 | pass |
---|
105 | |
---|
106 | |
---|
107 | def test_main(): |
---|
108 | run_unittest(TestContains) |
---|
109 | |
---|
110 | if __name__ == '__main__': |
---|
111 | test_main() |
---|