diff options
Diffstat (limited to 'tests/basics/class-contains.py')
-rw-r--r-- | tests/basics/class-contains.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tests/basics/class-contains.py b/tests/basics/class-contains.py new file mode 100644 index 0000000000..b6dd3661cd --- /dev/null +++ b/tests/basics/class-contains.py @@ -0,0 +1,23 @@ +# A contains everything +class A: + def __contains__(self, key): + return True + +a = A() +print(True in a) +print(1 in a) +print(() in a) + +# B contains given things +class B: + def __init__(self, items): + self.items = items + def __contains__(self, key): + return key in self.items + +b = B([]) +print(1 in b) +b = B([1, 2]) +print(1 in b) +print(2 in b) +print(3 in b) |