diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2020-09-17 00:28:32 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-09-17 00:28:32 (GMT) |
commit | 007eddad3b809e1555391dc7e8168918933b6184 (patch) | |
tree | b2125194e5c981cec0d568fbe1317e8c88f05050 | |
parent | 3f4012117bf80aa7c005f8fa6fb8e1f8b1aef5d5 (diff) | |
download | cpython-007eddad3b809e1555391dc7e8168918933b6184.zip cpython-007eddad3b809e1555391dc7e8168918933b6184.tar.gz cpython-007eddad3b809e1555391dc7e8168918933b6184.tar.bz2 |
_auto_called cleanup (GH-22285)
(cherry picked from commit fc23a9483ef0d7c98bea9f82392377d0b6ef7b18)
Co-authored-by: Ethan Furman <ethan@stoneleaf.us>
-rw-r--r-- | Lib/enum.py | 2 | ||||
-rw-r--r-- | Lib/test/test_enum.py | 11 |
2 files changed, 12 insertions, 1 deletions
diff --git a/Lib/enum.py b/Lib/enum.py index 4c1acc6..c2adfd1 100644 --- a/Lib/enum.py +++ b/Lib/enum.py @@ -104,9 +104,9 @@ class _EnumDict(dict): # enum overwriting a descriptor? raise TypeError('%r already defined as: %r' % (key, self[key])) if isinstance(value, auto): - self._auto_called = True if value.value == _auto_null: value.value = self._generate_next_value(key, 1, len(self._member_names), self._last_values[:]) + self._auto_called = True value = value.value self._member_names.append(key) self._last_values.append(value) diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index 11e88db..daa44ae 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -1815,6 +1815,17 @@ class TestEnum(unittest.TestCase): def _generate_next_value_(name, start, count, last): return name + def test_auto_order_wierd(self): + weird_auto = auto() + weird_auto.value = 'pathological case' + class Color(Enum): + red = weird_auto + def _generate_next_value_(name, start, count, last): + return name + blue = auto() + self.assertEqual(list(Color), [Color.red, Color.blue]) + self.assertEqual(Color.red.value, 'pathological case') + self.assertEqual(Color.blue.value, 'blue') def test_duplicate_auto(self): class Dupes(Enum): |