Add support for __contains__

zopefoundation · May 4, 2013 · 33de146 · 33de146
1 parent cc3dbe2
commit 33de146
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,6 +4,8 @@ Changelog
 3.0 (unreleased)
 ----------------
 
+- Add support for `__contains__`.
+
 - Rewrite tests as unit tests.
 
 2.13.0 (2010-03-30)

diff --git a/src/Record/__init__.py b/src/Record/__init__.py
@@ -70,3 +70,6 @@ def __setattr__(self, key, value):
             object.__setattr__(self, key, value)
         else:
             self.__setitem__(key, value)
+
+    def __contains__(self, key):
+        return key in self._schema
diff --git a/src/Record/tests.py b/src/Record/tests.py
@@ -67,3 +67,9 @@ def test_sequence(self):
         self.assertEqual(r[2], 7)
         self.assertEqual(r.c, 7)
         self.assertEqual(list(r), [1, 6, 7])
+
+    def test_contains(self):
+        r = R((1, 2, None))
+        self.assertTrue('a' in r)
+        self.assertTrue('c' in r)
+        self.assertFalse('d' in r)