From 64a9972b40b8258a6e37498da6279f4e40a88e2d Mon Sep 17 00:00:00 2001 From: Ethan Furman Date: Sun, 22 Sep 2013 16:18:19 -0700 Subject: Close #19025: Better error message when trying to delete an Enum member. Also slight code reorg for PEP 8 guidelines. --- Lib/test/test_enum.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'Lib/test/test_enum.py') diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index 65d0de743a..d59c5e3ea3 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -172,6 +172,27 @@ class TestEnum(unittest.TestCase): with self.assertRaises(AttributeError): Season.WINTER = 'really cold' + def test_attribute_deletion(self): + class Season(Enum): + SPRING = 1 + SUMMER = 2 + AUTUMN = 3 + WINTER = 4 + + def spam(cls): + pass + + self.assertTrue(hasattr(Season, 'spam')) + del Season.spam + self.assertFalse(hasattr(Season, 'spam')) + + with self.assertRaises(AttributeError): + del Season.SPRING + with self.assertRaises(AttributeError): + del Season.DRY + with self.assertRaises(AttributeError): + del Season.SPRING.name + def test_invalid_names(self): with self.assertRaises(ValueError): class Wrong(Enum): -- cgit v1.2.1