fix membership tests for primitive values, closes #117

Author: fgmacedo

choicesenum/enums.py

@@ -1,10 +1,23 @@
 # coding: utf-8
 from __future__ import absolute_import, unicode_literals
 
-from enum import Enum
+import six
+from enum import Enum, EnumMeta
 
 
-class ChoicesEnum(Enum):
+class ChoicesMetaClass(EnumMeta):
+
+    def __contains__(cls, member):
+        if not isinstance(member, cls):
+            try:
+                member = cls(member)
+            except Exception:
+                return False
+
+        return member._name_ in cls._member_map_
+
+
+class ChoicesEnum(six.with_metaclass(ChoicesMetaClass, Enum)):
 
     def __new__(cls, value, display=None):
         obj = object.__new__(cls)
@@ -90,7 +103,7 @@ def __json__(self):
 
     @property
     def display(self):
-        return self._display_ if self._display_ is not None else\
+        return self._display_ if self._display_ is not None else \
             self._name_.replace('_', ' ').capitalize()
 
     @property

tests/test_choicesenum.py

@@ -296,3 +296,9 @@ def test_should_return_default_value(colors):
     assert colors.get('#f00') == colors.RED
     assert colors.get('undefined_color') is None
     assert colors.get('undefined_color', colors.RED) == colors.RED
+
+
+def test_should_support_membership_vertification(colors):
+    assert colors.RED in colors
+    assert colors.RED.value in colors
+    assert 'non-existent-color' not in colors