| from test.test_support import have_unicode, run_unittest | |
| import unittest | |
| class base_set: | |
| def __init__(self, el): | |
| self.el = el | |
| class set(base_set): | |
| def __contains__(self, el): | |
| return self.el == el | |
| class seq(base_set): | |
| def __getitem__(self, n): | |
| return [self.el][n] | |
| class TestContains(unittest.TestCase): | |
| def test_common_tests(self): | |
| a = base_set(1) | |
| b = set(1) | |
| c = seq(1) | |
| self.assertIn(1, b) | |
| self.assertNotIn(0, b) | |
| self.assertIn(1, c) | |
| self.assertNotIn(0, c) | |
| self.assertRaises(TypeError, lambda: 1 in a) | |
| self.assertRaises(TypeError, lambda: 1 not in a) | |
| # test char in string | |
| self.assertIn('c', 'abc') | |
| self.assertNotIn('d', 'abc') | |
| self.assertIn('', '') | |
| self.assertIn('', 'abc') | |
| self.assertRaises(TypeError, lambda: None in 'abc') | |
| if have_unicode: | |
| def test_char_in_unicode(self): | |
| self.assertIn('c', unicode('abc')) | |
| self.assertNotIn('d', unicode('abc')) | |
| self.assertIn('', unicode('')) | |
| self.assertIn(unicode(''), '') | |
| self.assertIn(unicode(''), unicode('')) | |
| self.assertIn('', unicode('abc')) | |
| self.assertIn(unicode(''), 'abc') | |
| self.assertIn(unicode(''), unicode('abc')) | |
| self.assertRaises(TypeError, lambda: None in unicode('abc')) | |
| # test Unicode char in Unicode | |
| self.assertIn(unicode('c'), unicode('abc')) | |
| self.assertNotIn(unicode('d'), unicode('abc')) | |
| # test Unicode char in string | |
| self.assertIn(unicode('c'), 'abc') | |
| self.assertNotIn(unicode('d'), 'abc') | |
| def test_builtin_sequence_types(self): | |
| # a collection of tests on builtin sequence types | |
| a = range(10) | |
| for i in a: | |
| self.assertIn(i, a) | |
| self.assertNotIn(16, a) | |
| self.assertNotIn(a, a) | |
| a = tuple(a) | |
| for i in a: | |
| self.assertIn(i, a) | |
| self.assertNotIn(16, a) | |
| self.assertNotIn(a, a) | |
| class Deviant1: | |
| """Behaves strangely when compared | |
| This class is designed to make sure that the contains code | |
| works when the list is modified during the check. | |
| """ | |
| aList = range(15) | |
| def __cmp__(self, other): | |
| if other == 12: | |
| self.aList.remove(12) | |
| self.aList.remove(13) | |
| self.aList.remove(14) | |
| return 1 | |
| self.assertNotIn(Deviant1(), Deviant1.aList) | |
| class Deviant2: | |
| """Behaves strangely when compared | |
| This class raises an exception during comparison. That in | |
| turn causes the comparison to fail with a TypeError. | |
| """ | |
| def __cmp__(self, other): | |
| if other == 4: | |
| raise RuntimeError, "gotcha" | |
| try: | |
| self.assertNotIn(Deviant2(), a) | |
| except TypeError: | |
| pass | |
| def test_main(): | |
| run_unittest(TestContains) | |
| if __name__ == '__main__': | |
| test_main() |