Issue26748: Enum classes should evaluate as True

This commit is contained in:
Ethan Furman 2016-04-13 23:52:09 -07:00
parent b9869dfe35
commit 5de67b106a
2 changed files with 19 additions and 0 deletions

View File

@ -193,6 +193,12 @@ class EnumMeta(type):
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):
"""Either returns an existing member, or creates a new enum class.

View File

@ -257,6 +257,19 @@ class TestEnum(unittest.TestCase):
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):