commit 7219e27087baaa8a5693b5bef1b1357bddbffa53
author Ethan Furman <ethan@stoneleaf.us> Wed Sep 16 13:01:00 2020 -0700
committer GitHub <noreply@github.com> Wed Sep 16 13:01:00 2020 -0700
tree 9235d51826e2a14c26220a63a536085615604d16
parent fc23a9483ef0d7c98bea9f82392377d0b6ef7b18

Enum: make `Flag` and `IntFlag` members iterable (GH-22221)

Lib/test/test_enum.py

diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index ebf7604..59789fb 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -2350,6 +2350,12 @@
         self.assertFalse(W in RX)
         self.assertFalse(X in RW)
 
+    def test_member_iter(self):
+        Color = self.Color
+        self.assertEqual(list(Color.PURPLE), [Color.BLUE, Color.RED])
+        self.assertEqual(list(Color.BLUE), [Color.BLUE])
+        self.assertEqual(list(Color.GREEN), [Color.GREEN])
+
     def test_auto_number(self):
         class Color(Flag):
             red = auto()
@@ -2805,6 +2811,12 @@
         with self.assertRaises(TypeError):
             self.assertFalse('test' in RW)
 
+    def test_member_iter(self):
+        Color = self.Color
+        self.assertEqual(list(Color.PURPLE), [Color.BLUE, Color.RED])
+        self.assertEqual(list(Color.BLUE), [Color.BLUE])
+        self.assertEqual(list(Color.GREEN), [Color.GREEN])
+
     def test_bool(self):
         Perm = self.Perm
         for f in Perm: