summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_enum.py
diff options
context:
space:
mode:
authorEthan Furman <ethan@stoneleaf.us>2013-10-17 02:09:31 (GMT)
committerEthan Furman <ethan@stoneleaf.us>2013-10-17 02:09:31 (GMT)
commit1a1628819779110854d4e1ab490db2cab80e26b4 (patch)
treec4bc31e859c7b50932d2ef9b71e0d7f8aa2ab464 /Lib/test/test_enum.py
parent1f2eaa97eb9119bd2e98d9b62b5303850c78d2f6 (diff)
downloadcpython-1a1628819779110854d4e1ab490db2cab80e26b4.zip
cpython-1a1628819779110854d4e1ab490db2cab80e26b4.tar.gz
cpython-1a1628819779110854d4e1ab490db2cab80e26b4.tar.bz2
Close #19252: better test coverage for Enum. Thanks, CliffM
Diffstat (limited to 'Lib/test/test_enum.py')
-rw-r--r--Lib/test/test_enum.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 3d962f9..77477df 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -1139,8 +1139,10 @@ class TestEnum(unittest.TestCase):
green = ()
blue = ()
self.assertEqual(list(ColorInAList), [ColorInAList.red, ColorInAList.green, ColorInAList.blue])
- self.assertEqual(ColorInAList.red.value, [1])
- self.assertEqual(ColorInAList([1]), ColorInAList.red)
+ for enum, value in zip(ColorInAList, range(3)):
+ value += 1
+ self.assertEqual(enum.value, [value])
+ self.assertIs(ColorInAList([value]), enum)
def test_conflicting_types_resolved_in_new(self):
class LabelledIntEnum(int, Enum):