Mercurial > cpython-lru_cache-weakref

diff Lib/test/test_enum.py @ 85724:353ced6ae182

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Close #18693: Enum is now more help() friendly.
author Ethan Furman <ethan@stoneleaf.us>
date Sun, 15 Sep 2013 16:59:35 -0700
parents 54ddd1124df8
children ed011b0d7daf
line wrap: on
line diff
--- a/Lib/test/test_enum.py	Sun Sep 15 18:15:56 2013 -0400
+++ b/Lib/test/test_enum.py	Sun Sep 15 16:59:35 2013 -0700
@@ -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)