Issue26748: Enum classes should evaluate as True

ethanfurman · ethanfurman · commit de4e079d3dc3 · 2016-04-13T23:53:45.000-07:00
diff --git a/Lib/enum.py b/Lib/enum.py
@@ -206,6 +206,12 @@ def __new__(metacls, cls, bases, classdict):
             enum_class.__new__ = Enum.__new__
         return enum_class
 
+    def __bool__(self):
+        """
+        classes/types should always be True.
+        """
+        return True
+
     def __call__(cls, value, names=None, *, module=None, qualname=None, type=None, start=1):
         """Either returns an existing member, or creates a new enum class.
 
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
@@ -253,6 +253,19 @@ def spam(cls):
         with self.assertRaises(AttributeError):
             del Season.SPRING.name
 
+    def test_bool_of_class(self):
+        class Empty(Enum):
+            pass
+        self.assertTrue(bool(Empty))
+
+    def test_bool_of_member(self):
+        class Count(Enum):
+            zero = 0
+            one = 1
+            two = 2
+        for member in Count:
+            self.assertTrue(bool(member))
+
     def test_invalid_names(self):
         with self.assertRaises(ValueError):
             class Wrong(Enum):
