py: Implement __contains__ special method.

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)