summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/class-contains.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/basics/class-contains.py')
-rw-r--r--tests/basics/class-contains.py23
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)