commit c850f34220888838110cb07909fd80f3648ad2ba
author Ethan Furman <ethan@stoneleaf.us> Sun Sep 15 16:59:35 2013 -0700
committer Ethan Furman <ethan@stoneleaf.us> Sun Sep 15 16:59:35 2013 -0700
tree c193b3046dc2d68e32648f989c40049005aff415
parent 173a157e725579eec1f28f8c9d53d6761ba6c79f

Close #18693: Enum is now more help() friendly.

Lib/test/test_enum.py

diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 5d96d6d..65d0de7 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -98,7 +98,7 @@
         Season = self.Season
         self.assertEqual(
             set(dir(Season)),
-            set(['__class__', '__doc__', '__members__',
+            set(['__class__', '__doc__', '__members__', '__module__',
                 'SPRING', 'SUMMER', 'AUTUMN', 'WINTER']),
             )
 
@@ -106,9 +106,24 @@
         Season = self.Season
         self.assertEqual(
             set(dir(Season.WINTER)),
-            set(['__class__', '__doc__', 'name', 'value']),
+            set(['__class__', '__doc__', '__module__', 'name', 'value']),
             )
 
+    def test_dir_with_added_behavior(self):
+        class Test(Enum):
+            this = 'that'
+            these = 'those'
+            def wowser(self):
+                return ("Wowser! I'm %s!" % self.name)
+        self.assertEqual(
+                set(dir(Test)),
+                set(['__class__', '__doc__', '__members__', '__module__', 'this', 'these']),
+                )
+        self.assertEqual(
+                set(dir(Test.this)),
+                set(['__class__', '__doc__', '__module__', 'name', 'value', 'wowser']),
+                )
+
     def test_enum_in_enum_out(self):
         Season = self.Season
         self.assertIs(Season(Season.WINTER), Season.WINTER)