More efficient __contains__

1 files changed, 6 insertions, 0 deletions

diff --git a/frozendict/__init__.py b/frozendict/__init__.py
index 7036731..9598f10 100644
--- a/frozendict/__init__.py
+++ b/frozendict/__init__.py
@@ -12,6 +12,9 @@ class frozendict(collections.Mapping):
     def __getitem__(self, key):
         return self.__dict[key]
 
+    def __contains__(self, key):
+        return key in self.__dict
+
     def copy(self, **add_or_replace):
         return frozendict(self, **add_or_replace)
 
@@ -45,6 +48,9 @@ class FrozenOrderedDict(collections.Mapping):
     def __getitem__(self, key):
         return self.__dict[key]
 
+    def __contains__(self, key):
+        return key in self.__dict
+
     def copy(self, **add_or_replace):
         return FrozenOrderedDict(self, **add_or_replace)