2013-06-14 20:55:46 -03:00
|
|
|
import enum
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
import doctest
|
2013-10-22 00:45:55 -03:00
|
|
|
import inspect
|
2021-01-26 16:52:52 -04:00
|
|
|
import os
|
2013-10-22 00:45:55 -03:00
|
|
|
import pydoc
|
2018-09-12 14:28:53 -03:00
|
|
|
import sys
|
2013-06-14 20:55:46 -03:00
|
|
|
import unittest
|
2017-09-07 13:56:24 -03:00
|
|
|
import threading
|
2013-06-14 20:55:46 -03:00
|
|
|
from collections import OrderedDict
|
2021-03-31 01:17:26 -03:00
|
|
|
from enum import Enum, IntEnum, StrEnum, EnumType, Flag, IntFlag, unique, auto
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
from enum import STRICT, CONFORM, EJECT, KEEP
|
2013-10-22 00:45:55 -03:00
|
|
|
from io import StringIO
|
2014-02-06 21:28:50 -04:00
|
|
|
from pickle import dumps, loads, PicklingError, HIGHEST_PROTOCOL
|
2015-11-14 08:46:42 -04:00
|
|
|
from test import support
|
2019-08-08 02:43:18 -03:00
|
|
|
from test.support import ALWAYS_EQ
|
2020-05-27 19:10:27 -03:00
|
|
|
from test.support import threading_helper
|
2018-01-22 11:56:37 -04:00
|
|
|
from datetime import timedelta
|
2017-01-24 16:12:06 -04:00
|
|
|
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
def load_tests(loader, tests, ignore):
|
|
|
|
tests.addTests(doctest.DocTestSuite(enum))
|
2021-03-03 13:54:30 -04:00
|
|
|
if os.path.exists('Doc/library/enum.rst'):
|
2021-01-26 16:52:52 -04:00
|
|
|
tests.addTests(doctest.DocFileSuite(
|
|
|
|
'../../Doc/library/enum.rst',
|
|
|
|
optionflags=doctest.ELLIPSIS|doctest.NORMALIZE_WHITESPACE,
|
|
|
|
))
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
return tests
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
# for pickle tests
|
|
|
|
try:
|
|
|
|
class Stooges(Enum):
|
|
|
|
LARRY = 1
|
|
|
|
CURLY = 2
|
|
|
|
MOE = 3
|
|
|
|
except Exception as exc:
|
|
|
|
Stooges = exc
|
|
|
|
|
|
|
|
try:
|
|
|
|
class IntStooges(int, Enum):
|
|
|
|
LARRY = 1
|
|
|
|
CURLY = 2
|
|
|
|
MOE = 3
|
|
|
|
except Exception as exc:
|
|
|
|
IntStooges = exc
|
|
|
|
|
|
|
|
try:
|
|
|
|
class FloatStooges(float, Enum):
|
|
|
|
LARRY = 1.39
|
|
|
|
CURLY = 2.72
|
|
|
|
MOE = 3.142596
|
|
|
|
except Exception as exc:
|
|
|
|
FloatStooges = exc
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
try:
|
|
|
|
class FlagStooges(Flag):
|
|
|
|
LARRY = 1
|
|
|
|
CURLY = 2
|
|
|
|
MOE = 3
|
|
|
|
except Exception as exc:
|
|
|
|
FlagStooges = exc
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
# for pickle test and subclass tests
|
2020-09-21 21:23:13 -03:00
|
|
|
class Name(StrEnum):
|
|
|
|
BDFL = 'Guido van Rossum'
|
|
|
|
FLUFL = 'Barry Warsaw'
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
try:
|
|
|
|
Question = Enum('Question', 'who what when where why', module=__name__)
|
|
|
|
except Exception as exc:
|
|
|
|
Question = exc
|
|
|
|
|
|
|
|
try:
|
|
|
|
Answer = Enum('Answer', 'him this then there because')
|
|
|
|
except Exception as exc:
|
|
|
|
Answer = exc
|
|
|
|
|
2014-02-08 15:36:27 -04:00
|
|
|
try:
|
|
|
|
Theory = Enum('Theory', 'rule law supposition', qualname='spanish_inquisition')
|
|
|
|
except Exception as exc:
|
|
|
|
Theory = exc
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
# for doctests
|
|
|
|
try:
|
|
|
|
class Fruit(Enum):
|
2016-11-21 13:22:05 -04:00
|
|
|
TOMATO = 1
|
|
|
|
BANANA = 2
|
|
|
|
CHERRY = 3
|
2013-06-14 20:55:46 -03:00
|
|
|
except Exception:
|
|
|
|
pass
|
|
|
|
|
2015-03-31 10:56:49 -03:00
|
|
|
def test_pickle_dump_load(assertion, source, target=None):
|
2014-02-06 21:28:50 -04:00
|
|
|
if target is None:
|
|
|
|
target = source
|
2015-03-31 10:56:49 -03:00
|
|
|
for protocol in range(HIGHEST_PROTOCOL + 1):
|
2014-02-06 21:28:50 -04:00
|
|
|
assertion(loads(dumps(source, protocol=protocol)), target)
|
|
|
|
|
2015-03-31 10:56:49 -03:00
|
|
|
def test_pickle_exception(assertion, exception, obj):
|
|
|
|
for protocol in range(HIGHEST_PROTOCOL + 1):
|
2014-02-06 21:28:50 -04:00
|
|
|
with assertion(exception):
|
|
|
|
dumps(obj, protocol=protocol)
|
2013-10-06 21:19:54 -03:00
|
|
|
|
|
|
|
class TestHelpers(unittest.TestCase):
|
|
|
|
# _is_descriptor, _is_sunder, _is_dunder
|
|
|
|
|
|
|
|
def test_is_descriptor(self):
|
|
|
|
class foo:
|
|
|
|
pass
|
|
|
|
for attr in ('__get__','__set__','__delete__'):
|
|
|
|
obj = foo()
|
|
|
|
self.assertFalse(enum._is_descriptor(obj))
|
|
|
|
setattr(obj, attr, 1)
|
|
|
|
self.assertTrue(enum._is_descriptor(obj))
|
|
|
|
|
|
|
|
def test_is_sunder(self):
|
|
|
|
for s in ('_a_', '_aa_'):
|
|
|
|
self.assertTrue(enum._is_sunder(s))
|
|
|
|
|
|
|
|
for s in ('a', 'a_', '_a', '__a', 'a__', '__a__', '_a__', '__a_', '_',
|
|
|
|
'__', '___', '____', '_____',):
|
|
|
|
self.assertFalse(enum._is_sunder(s))
|
|
|
|
|
|
|
|
def test_is_dunder(self):
|
|
|
|
for s in ('__a__', '__aa__'):
|
|
|
|
self.assertTrue(enum._is_dunder(s))
|
|
|
|
for s in ('a', 'a_', '_a', '__a', 'a__', '_a_', '_a__', '__a_', '_',
|
|
|
|
'__', '___', '____', '_____',):
|
|
|
|
self.assertFalse(enum._is_dunder(s))
|
|
|
|
|
2018-09-21 23:03:09 -03:00
|
|
|
# for subclassing tests
|
|
|
|
|
|
|
|
class classproperty:
|
|
|
|
|
|
|
|
def __init__(self, fget=None, fset=None, fdel=None, doc=None):
|
|
|
|
self.fget = fget
|
|
|
|
self.fset = fset
|
|
|
|
self.fdel = fdel
|
|
|
|
if doc is None and fget is not None:
|
|
|
|
doc = fget.__doc__
|
|
|
|
self.__doc__ = doc
|
|
|
|
|
|
|
|
def __get__(self, instance, ownerclass):
|
|
|
|
return self.fget(ownerclass)
|
|
|
|
|
|
|
|
|
2016-09-11 03:36:59 -03:00
|
|
|
# tests
|
2013-10-06 21:19:54 -03:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
class TestEnum(unittest.TestCase):
|
2014-02-08 15:36:27 -04:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def setUp(self):
|
|
|
|
class Season(Enum):
|
|
|
|
SPRING = 1
|
|
|
|
SUMMER = 2
|
|
|
|
AUTUMN = 3
|
|
|
|
WINTER = 4
|
|
|
|
self.Season = Season
|
|
|
|
|
2013-08-31 23:17:41 -03:00
|
|
|
class Konstants(float, Enum):
|
|
|
|
E = 2.7182818
|
|
|
|
PI = 3.1415926
|
|
|
|
TAU = 2 * PI
|
|
|
|
self.Konstants = Konstants
|
|
|
|
|
|
|
|
class Grades(IntEnum):
|
|
|
|
A = 5
|
|
|
|
B = 4
|
|
|
|
C = 3
|
|
|
|
D = 2
|
|
|
|
F = 0
|
|
|
|
self.Grades = Grades
|
|
|
|
|
|
|
|
class Directional(str, Enum):
|
|
|
|
EAST = 'east'
|
|
|
|
WEST = 'west'
|
|
|
|
NORTH = 'north'
|
|
|
|
SOUTH = 'south'
|
|
|
|
self.Directional = Directional
|
|
|
|
|
|
|
|
from datetime import date
|
|
|
|
class Holiday(date, Enum):
|
|
|
|
NEW_YEAR = 2013, 1, 1
|
|
|
|
IDES_OF_MARCH = 2013, 3, 15
|
|
|
|
self.Holiday = Holiday
|
|
|
|
|
2013-08-12 10:51:41 -03:00
|
|
|
def test_dir_on_class(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertEqual(
|
|
|
|
set(dir(Season)),
|
2013-09-15 20:59:35 -03:00
|
|
|
set(['__class__', '__doc__', '__members__', '__module__',
|
2013-08-12 10:51:41 -03:00
|
|
|
'SPRING', 'SUMMER', 'AUTUMN', 'WINTER']),
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_dir_on_item(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertEqual(
|
|
|
|
set(dir(Season.WINTER)),
|
2013-09-15 20:59:35 -03:00
|
|
|
set(['__class__', '__doc__', '__module__', 'name', 'value']),
|
2013-08-12 10:51:41 -03:00
|
|
|
)
|
|
|
|
|
2013-09-15 20:59:35 -03:00
|
|
|
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']),
|
|
|
|
)
|
|
|
|
|
2014-10-14 12:58:32 -03:00
|
|
|
def test_dir_on_sub_with_behavior_on_super(self):
|
|
|
|
# see issue22506
|
|
|
|
class SuperEnum(Enum):
|
|
|
|
def invisible(self):
|
|
|
|
return "did you see me?"
|
|
|
|
class SubEnum(SuperEnum):
|
|
|
|
sample = 5
|
|
|
|
self.assertEqual(
|
|
|
|
set(dir(SubEnum.sample)),
|
|
|
|
set(['__class__', '__doc__', '__module__', 'name', 'value', 'invisible']),
|
|
|
|
)
|
|
|
|
|
2020-09-21 10:11:06 -03:00
|
|
|
def test_dir_on_sub_with_behavior_including_instance_dict_on_super(self):
|
|
|
|
# see issue40084
|
|
|
|
class SuperEnum(IntEnum):
|
|
|
|
def __new__(cls, value, description=""):
|
|
|
|
obj = int.__new__(cls, value)
|
|
|
|
obj._value_ = value
|
|
|
|
obj.description = description
|
|
|
|
return obj
|
|
|
|
class SubEnum(SuperEnum):
|
|
|
|
sample = 5
|
|
|
|
self.assertTrue({'description'} <= set(dir(SubEnum.sample)))
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_enum_in_enum_out(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertIs(Season(Season.WINTER), Season.WINTER)
|
|
|
|
|
|
|
|
def test_enum_value(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertEqual(Season.SPRING.value, 1)
|
|
|
|
|
|
|
|
def test_intenum_value(self):
|
|
|
|
self.assertEqual(IntStooges.CURLY.value, 2)
|
|
|
|
|
|
|
|
def test_enum(self):
|
|
|
|
Season = self.Season
|
|
|
|
lst = list(Season)
|
|
|
|
self.assertEqual(len(lst), len(Season))
|
|
|
|
self.assertEqual(len(Season), 4, Season)
|
|
|
|
self.assertEqual(
|
|
|
|
[Season.SPRING, Season.SUMMER, Season.AUTUMN, Season.WINTER], lst)
|
|
|
|
|
|
|
|
for i, season in enumerate('SPRING SUMMER AUTUMN WINTER'.split(), 1):
|
|
|
|
e = Season(i)
|
|
|
|
self.assertEqual(e, getattr(Season, season))
|
|
|
|
self.assertEqual(e.value, i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, season)
|
|
|
|
self.assertIn(e, Season)
|
|
|
|
self.assertIs(type(e), Season)
|
|
|
|
self.assertIsInstance(e, Season)
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(e), season)
|
|
|
|
self.assertEqual(repr(e), 'Season.{0}'.format(season))
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_value_name(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertEqual(Season.SPRING.name, 'SPRING')
|
|
|
|
self.assertEqual(Season.SPRING.value, 1)
|
|
|
|
with self.assertRaises(AttributeError):
|
|
|
|
Season.SPRING.name = 'invierno'
|
|
|
|
with self.assertRaises(AttributeError):
|
|
|
|
Season.SPRING.value = 2
|
|
|
|
|
2013-09-06 11:16:48 -03:00
|
|
|
def test_changing_member(self):
|
|
|
|
Season = self.Season
|
|
|
|
with self.assertRaises(AttributeError):
|
|
|
|
Season.WINTER = 'really cold'
|
|
|
|
|
2013-09-22 20:18:19 -03:00
|
|
|
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
|
|
|
|
|
2016-04-14 03:52:09 -03:00
|
|
|
def test_bool_of_class(self):
|
|
|
|
class Empty(Enum):
|
|
|
|
pass
|
|
|
|
self.assertTrue(bool(Empty))
|
|
|
|
|
|
|
|
def test_bool_of_member(self):
|
|
|
|
class Count(Enum):
|
|
|
|
zero = 0
|
|
|
|
one = 1
|
|
|
|
two = 2
|
|
|
|
for member in Count:
|
|
|
|
self.assertTrue(bool(member))
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_invalid_names(self):
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Wrong(Enum):
|
|
|
|
mro = 9
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Wrong(Enum):
|
|
|
|
_create_= 11
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Wrong(Enum):
|
|
|
|
_get_mixins_ = 9
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Wrong(Enum):
|
|
|
|
_find_new_ = 1
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Wrong(Enum):
|
|
|
|
_any_name_ = 9
|
|
|
|
|
2015-09-18 01:49:12 -03:00
|
|
|
def test_bool(self):
|
2016-01-15 19:01:33 -04:00
|
|
|
# plain Enum members are always True
|
2015-09-18 01:49:12 -03:00
|
|
|
class Logic(Enum):
|
|
|
|
true = True
|
|
|
|
false = False
|
|
|
|
self.assertTrue(Logic.true)
|
2016-01-15 19:01:33 -04:00
|
|
|
self.assertTrue(Logic.false)
|
|
|
|
# unless overridden
|
|
|
|
class RealLogic(Enum):
|
|
|
|
true = True
|
|
|
|
false = False
|
|
|
|
def __bool__(self):
|
|
|
|
return bool(self._value_)
|
|
|
|
self.assertTrue(RealLogic.true)
|
|
|
|
self.assertFalse(RealLogic.false)
|
|
|
|
# mixed Enums depend on mixed-in type
|
|
|
|
class IntLogic(int, Enum):
|
|
|
|
true = 1
|
|
|
|
false = 0
|
|
|
|
self.assertTrue(IntLogic.true)
|
|
|
|
self.assertFalse(IntLogic.false)
|
2015-09-18 01:49:12 -03:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_contains(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertIn(Season.AUTUMN, Season)
|
2018-09-10 15:21:04 -03:00
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
3 in Season
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
'AUTUMN' in Season
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
val = Season(3)
|
|
|
|
self.assertIn(val, Season)
|
|
|
|
|
|
|
|
class OtherEnum(Enum):
|
|
|
|
one = 1; two = 2
|
|
|
|
self.assertNotIn(OtherEnum.two, Season)
|
|
|
|
|
|
|
|
def test_comparisons(self):
|
|
|
|
Season = self.Season
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
Season.SPRING < Season.WINTER
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
Season.SPRING > 4
|
|
|
|
|
|
|
|
self.assertNotEqual(Season.SPRING, 1)
|
|
|
|
|
|
|
|
class Part(Enum):
|
|
|
|
SPRING = 1
|
|
|
|
CLIP = 2
|
|
|
|
BARREL = 3
|
|
|
|
|
|
|
|
self.assertNotEqual(Season.SPRING, Part.SPRING)
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
Season.SPRING < Part.CLIP
|
|
|
|
|
|
|
|
def test_enum_duplicates(self):
|
|
|
|
class Season(Enum):
|
|
|
|
SPRING = 1
|
|
|
|
SUMMER = 2
|
|
|
|
AUTUMN = FALL = 3
|
|
|
|
WINTER = 4
|
|
|
|
ANOTHER_SPRING = 1
|
|
|
|
lst = list(Season)
|
|
|
|
self.assertEqual(
|
|
|
|
lst,
|
|
|
|
[Season.SPRING, Season.SUMMER,
|
|
|
|
Season.AUTUMN, Season.WINTER,
|
|
|
|
])
|
|
|
|
self.assertIs(Season.FALL, Season.AUTUMN)
|
|
|
|
self.assertEqual(Season.FALL.value, 3)
|
|
|
|
self.assertEqual(Season.AUTUMN.value, 3)
|
|
|
|
self.assertIs(Season(3), Season.AUTUMN)
|
|
|
|
self.assertIs(Season(1), Season.SPRING)
|
|
|
|
self.assertEqual(Season.FALL.name, 'AUTUMN')
|
|
|
|
self.assertEqual(
|
|
|
|
[k for k,v in Season.__members__.items() if v.name != k],
|
|
|
|
['FALL', 'ANOTHER_SPRING'],
|
|
|
|
)
|
|
|
|
|
2013-09-15 16:34:36 -03:00
|
|
|
def test_duplicate_name(self):
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
red = 4
|
|
|
|
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
def red(self):
|
|
|
|
return 'red'
|
|
|
|
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class Color(Enum):
|
|
|
|
@property
|
|
|
|
def red(self):
|
|
|
|
return 'redder'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
|
2020-09-13 17:27:51 -03:00
|
|
|
def test_reserved__sunder_(self):
|
2020-09-15 16:27:06 -03:00
|
|
|
with self.assertRaisesRegex(
|
|
|
|
ValueError,
|
2021-03-31 01:17:26 -03:00
|
|
|
'_sunder_ names, such as ._bad_., are reserved',
|
2020-09-15 16:27:06 -03:00
|
|
|
):
|
2020-09-13 17:27:51 -03:00
|
|
|
class Bad(Enum):
|
|
|
|
_bad_ = 1
|
2013-09-15 16:34:36 -03:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_enum_with_value_name(self):
|
|
|
|
class Huh(Enum):
|
|
|
|
name = 1
|
|
|
|
value = 2
|
|
|
|
self.assertEqual(
|
|
|
|
list(Huh),
|
|
|
|
[Huh.name, Huh.value],
|
|
|
|
)
|
|
|
|
self.assertIs(type(Huh.name), Huh)
|
|
|
|
self.assertEqual(Huh.name.name, 'name')
|
|
|
|
self.assertEqual(Huh.name.value, 1)
|
2013-08-31 23:17:41 -03:00
|
|
|
|
|
|
|
def test_format_enum(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertEqual('{}'.format(Season.SPRING),
|
|
|
|
'{}'.format(str(Season.SPRING)))
|
|
|
|
self.assertEqual( '{:}'.format(Season.SPRING),
|
|
|
|
'{:}'.format(str(Season.SPRING)))
|
|
|
|
self.assertEqual('{:20}'.format(Season.SPRING),
|
|
|
|
'{:20}'.format(str(Season.SPRING)))
|
|
|
|
self.assertEqual('{:^20}'.format(Season.SPRING),
|
|
|
|
'{:^20}'.format(str(Season.SPRING)))
|
|
|
|
self.assertEqual('{:>20}'.format(Season.SPRING),
|
|
|
|
'{:>20}'.format(str(Season.SPRING)))
|
|
|
|
self.assertEqual('{:<20}'.format(Season.SPRING),
|
|
|
|
'{:<20}'.format(str(Season.SPRING)))
|
|
|
|
|
2019-07-04 15:28:37 -03:00
|
|
|
def test_str_override_enum(self):
|
|
|
|
class EnumWithStrOverrides(Enum):
|
|
|
|
one = auto()
|
|
|
|
two = auto()
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
return 'Str!'
|
|
|
|
self.assertEqual(str(EnumWithStrOverrides.one), 'Str!')
|
|
|
|
self.assertEqual('{}'.format(EnumWithStrOverrides.one), 'Str!')
|
|
|
|
|
|
|
|
def test_format_override_enum(self):
|
|
|
|
class EnumWithFormatOverride(Enum):
|
|
|
|
one = 1.0
|
|
|
|
two = 2.0
|
|
|
|
def __format__(self, spec):
|
|
|
|
return 'Format!!'
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(EnumWithFormatOverride.one), 'one')
|
2019-07-04 15:28:37 -03:00
|
|
|
self.assertEqual('{}'.format(EnumWithFormatOverride.one), 'Format!!')
|
|
|
|
|
|
|
|
def test_str_and_format_override_enum(self):
|
|
|
|
class EnumWithStrFormatOverrides(Enum):
|
|
|
|
one = auto()
|
|
|
|
two = auto()
|
|
|
|
def __str__(self):
|
|
|
|
return 'Str!'
|
|
|
|
def __format__(self, spec):
|
|
|
|
return 'Format!'
|
|
|
|
self.assertEqual(str(EnumWithStrFormatOverrides.one), 'Str!')
|
|
|
|
self.assertEqual('{}'.format(EnumWithStrFormatOverrides.one), 'Format!')
|
|
|
|
|
|
|
|
def test_str_override_mixin(self):
|
|
|
|
class MixinEnumWithStrOverride(float, Enum):
|
|
|
|
one = 1.0
|
|
|
|
two = 2.0
|
|
|
|
def __str__(self):
|
|
|
|
return 'Overridden!'
|
|
|
|
self.assertEqual(str(MixinEnumWithStrOverride.one), 'Overridden!')
|
|
|
|
self.assertEqual('{}'.format(MixinEnumWithStrOverride.one), 'Overridden!')
|
|
|
|
|
|
|
|
def test_str_and_format_override_mixin(self):
|
|
|
|
class MixinWithStrFormatOverrides(float, Enum):
|
|
|
|
one = 1.0
|
|
|
|
two = 2.0
|
|
|
|
def __str__(self):
|
|
|
|
return 'Str!'
|
|
|
|
def __format__(self, spec):
|
|
|
|
return 'Format!'
|
|
|
|
self.assertEqual(str(MixinWithStrFormatOverrides.one), 'Str!')
|
|
|
|
self.assertEqual('{}'.format(MixinWithStrFormatOverrides.one), 'Format!')
|
|
|
|
|
|
|
|
def test_format_override_mixin(self):
|
2013-08-31 23:17:41 -03:00
|
|
|
class TestFloat(float, Enum):
|
|
|
|
one = 1.0
|
|
|
|
two = 2.0
|
|
|
|
def __format__(self, spec):
|
|
|
|
return 'TestFloat success!'
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(TestFloat.one), 'one')
|
2013-08-31 23:17:41 -03:00
|
|
|
self.assertEqual('{}'.format(TestFloat.one), 'TestFloat success!')
|
|
|
|
|
|
|
|
def assertFormatIsValue(self, spec, member):
|
|
|
|
self.assertEqual(spec.format(member), spec.format(member.value))
|
|
|
|
|
|
|
|
def test_format_enum_date(self):
|
|
|
|
Holiday = self.Holiday
|
|
|
|
self.assertFormatIsValue('{}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:20}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:^20}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:>20}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:<20}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:%Y %m}', Holiday.IDES_OF_MARCH)
|
|
|
|
self.assertFormatIsValue('{:%Y %m %M:00}', Holiday.IDES_OF_MARCH)
|
|
|
|
|
|
|
|
def test_format_enum_float(self):
|
|
|
|
Konstants = self.Konstants
|
|
|
|
self.assertFormatIsValue('{}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:20}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:^20}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:>20}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:<20}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:n}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:5.2}', Konstants.TAU)
|
|
|
|
self.assertFormatIsValue('{:f}', Konstants.TAU)
|
|
|
|
|
|
|
|
def test_format_enum_int(self):
|
|
|
|
Grades = self.Grades
|
|
|
|
self.assertFormatIsValue('{}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:20}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:^20}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:>20}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:<20}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:+}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:08X}', Grades.C)
|
|
|
|
self.assertFormatIsValue('{:b}', Grades.C)
|
|
|
|
|
|
|
|
def test_format_enum_str(self):
|
|
|
|
Directional = self.Directional
|
|
|
|
self.assertFormatIsValue('{}', Directional.WEST)
|
|
|
|
self.assertFormatIsValue('{:}', Directional.WEST)
|
|
|
|
self.assertFormatIsValue('{:20}', Directional.WEST)
|
|
|
|
self.assertFormatIsValue('{:^20}', Directional.WEST)
|
|
|
|
self.assertFormatIsValue('{:>20}', Directional.WEST)
|
|
|
|
self.assertFormatIsValue('{:<20}', Directional.WEST)
|
|
|
|
|
2020-09-15 20:28:25 -03:00
|
|
|
def test_object_str_override(self):
|
|
|
|
class Colors(Enum):
|
|
|
|
RED, GREEN, BLUE = 1, 2, 3
|
|
|
|
def __repr__(self):
|
|
|
|
return "test.%s" % (self._name_, )
|
|
|
|
__str__ = object.__str__
|
|
|
|
self.assertEqual(str(Colors.RED), 'test.RED')
|
|
|
|
|
2020-09-15 19:56:26 -03:00
|
|
|
def test_enum_str_override(self):
|
|
|
|
class MyStrEnum(Enum):
|
|
|
|
def __str__(self):
|
|
|
|
return 'MyStr'
|
|
|
|
class MyMethodEnum(Enum):
|
|
|
|
def hello(self):
|
|
|
|
return 'Hello! My name is %s' % self.name
|
|
|
|
class Test1Enum(MyMethodEnum, int, MyStrEnum):
|
|
|
|
One = 1
|
|
|
|
Two = 2
|
2020-12-08 15:14:10 -04:00
|
|
|
self.assertTrue(Test1Enum._member_type_ is int)
|
2020-09-15 19:56:26 -03:00
|
|
|
self.assertEqual(str(Test1Enum.One), 'MyStr')
|
2020-12-08 15:14:10 -04:00
|
|
|
self.assertEqual(format(Test1Enum.One, ''), 'MyStr')
|
2020-09-15 19:56:26 -03:00
|
|
|
#
|
|
|
|
class Test2Enum(MyStrEnum, MyMethodEnum):
|
|
|
|
One = 1
|
|
|
|
Two = 2
|
|
|
|
self.assertEqual(str(Test2Enum.One), 'MyStr')
|
2020-12-08 15:14:10 -04:00
|
|
|
self.assertEqual(format(Test1Enum.One, ''), 'MyStr')
|
2020-09-15 19:56:26 -03:00
|
|
|
|
|
|
|
def test_inherited_data_type(self):
|
|
|
|
class HexInt(int):
|
|
|
|
def __repr__(self):
|
|
|
|
return hex(self)
|
|
|
|
class MyEnum(HexInt, enum.Enum):
|
|
|
|
A = 1
|
|
|
|
B = 2
|
|
|
|
C = 3
|
2021-03-31 01:17:26 -03:00
|
|
|
def __repr__(self):
|
|
|
|
return '<%s.%s: %r>' % (self.__class__.__name__, self._name_, self._value_)
|
2020-09-15 19:56:26 -03:00
|
|
|
self.assertEqual(repr(MyEnum.A), '<MyEnum.A: 0x1>')
|
|
|
|
|
|
|
|
def test_too_many_data_types(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'too many data types'):
|
|
|
|
class Huh(str, int, Enum):
|
|
|
|
One = 1
|
|
|
|
|
|
|
|
class MyStr(str):
|
|
|
|
def hello(self):
|
|
|
|
return 'hello, %s' % self
|
|
|
|
class MyInt(int):
|
|
|
|
def repr(self):
|
|
|
|
return hex(self)
|
|
|
|
with self.assertRaisesRegex(TypeError, 'too many data types'):
|
|
|
|
class Huh(MyStr, MyInt, Enum):
|
|
|
|
One = 1
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_hash(self):
|
|
|
|
Season = self.Season
|
|
|
|
dates = {}
|
|
|
|
dates[Season.WINTER] = '1225'
|
|
|
|
dates[Season.SPRING] = '0315'
|
|
|
|
dates[Season.SUMMER] = '0704'
|
|
|
|
dates[Season.AUTUMN] = '1031'
|
|
|
|
self.assertEqual(dates[Season.AUTUMN], '1031')
|
|
|
|
|
|
|
|
def test_intenum_from_scratch(self):
|
|
|
|
class phy(int, Enum):
|
|
|
|
pi = 3
|
|
|
|
tau = 2 * pi
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
|
|
|
|
|
|
|
def test_intenum_inherited(self):
|
|
|
|
class IntEnum(int, Enum):
|
|
|
|
pass
|
|
|
|
class phy(IntEnum):
|
|
|
|
pi = 3
|
|
|
|
tau = 2 * pi
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
|
|
|
|
|
|
|
def test_floatenum_from_scratch(self):
|
|
|
|
class phy(float, Enum):
|
2013-08-31 23:17:41 -03:00
|
|
|
pi = 3.1415926
|
2013-06-14 20:55:46 -03:00
|
|
|
tau = 2 * pi
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
|
|
|
|
|
|
|
def test_floatenum_inherited(self):
|
|
|
|
class FloatEnum(float, Enum):
|
|
|
|
pass
|
|
|
|
class phy(FloatEnum):
|
2013-08-31 23:17:41 -03:00
|
|
|
pi = 3.1415926
|
2013-06-14 20:55:46 -03:00
|
|
|
tau = 2 * pi
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
|
|
|
|
|
|
|
def test_strenum_from_scratch(self):
|
|
|
|
class phy(str, Enum):
|
|
|
|
pi = 'Pi'
|
|
|
|
tau = 'Tau'
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
|
|
|
|
2020-09-21 21:23:13 -03:00
|
|
|
def test_strenum_inherited_methods(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
class phy(StrEnum):
|
|
|
|
pi = 'Pi'
|
|
|
|
tau = 'Tau'
|
|
|
|
self.assertTrue(phy.pi < phy.tau)
|
2020-09-21 21:23:13 -03:00
|
|
|
self.assertEqual(phy.pi.upper(), 'PI')
|
|
|
|
self.assertEqual(phy.tau.count('a'), 1)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_intenum(self):
|
|
|
|
class WeekDay(IntEnum):
|
|
|
|
SUNDAY = 1
|
|
|
|
MONDAY = 2
|
|
|
|
TUESDAY = 3
|
|
|
|
WEDNESDAY = 4
|
|
|
|
THURSDAY = 5
|
|
|
|
FRIDAY = 6
|
|
|
|
SATURDAY = 7
|
|
|
|
|
|
|
|
self.assertEqual(['a', 'b', 'c'][WeekDay.MONDAY], 'c')
|
|
|
|
self.assertEqual([i for i in range(WeekDay.TUESDAY)], [0, 1, 2])
|
|
|
|
|
|
|
|
lst = list(WeekDay)
|
|
|
|
self.assertEqual(len(lst), len(WeekDay))
|
|
|
|
self.assertEqual(len(WeekDay), 7)
|
|
|
|
target = 'SUNDAY MONDAY TUESDAY WEDNESDAY THURSDAY FRIDAY SATURDAY'
|
|
|
|
target = target.split()
|
|
|
|
for i, weekday in enumerate(target, 1):
|
|
|
|
e = WeekDay(i)
|
|
|
|
self.assertEqual(e, i)
|
|
|
|
self.assertEqual(int(e), i)
|
|
|
|
self.assertEqual(e.name, weekday)
|
|
|
|
self.assertIn(e, WeekDay)
|
|
|
|
self.assertEqual(lst.index(e)+1, i)
|
|
|
|
self.assertTrue(0 < e < 8)
|
|
|
|
self.assertIs(type(e), WeekDay)
|
|
|
|
self.assertIsInstance(e, int)
|
|
|
|
self.assertIsInstance(e, Enum)
|
|
|
|
|
|
|
|
def test_intenum_duplicates(self):
|
|
|
|
class WeekDay(IntEnum):
|
|
|
|
SUNDAY = 1
|
|
|
|
MONDAY = 2
|
|
|
|
TUESDAY = TEUSDAY = 3
|
|
|
|
WEDNESDAY = 4
|
|
|
|
THURSDAY = 5
|
|
|
|
FRIDAY = 6
|
|
|
|
SATURDAY = 7
|
|
|
|
self.assertIs(WeekDay.TEUSDAY, WeekDay.TUESDAY)
|
|
|
|
self.assertEqual(WeekDay(3).name, 'TUESDAY')
|
|
|
|
self.assertEqual([k for k,v in WeekDay.__members__.items()
|
|
|
|
if v.name != k], ['TEUSDAY', ])
|
|
|
|
|
2016-05-12 04:37:58 -03:00
|
|
|
def test_intenum_from_bytes(self):
|
|
|
|
self.assertIs(IntStooges.from_bytes(b'\x00\x03', 'big'), IntStooges.MOE)
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
IntStooges.from_bytes(b'\x00\x05', 'big')
|
|
|
|
|
|
|
|
def test_floatenum_fromhex(self):
|
|
|
|
h = float.hex(FloatStooges.MOE.value)
|
|
|
|
self.assertIs(FloatStooges.fromhex(h), FloatStooges.MOE)
|
|
|
|
h = float.hex(FloatStooges.MOE.value + 0.01)
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
FloatStooges.fromhex(h)
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_pickle_enum(self):
|
|
|
|
if isinstance(Stooges, Exception):
|
|
|
|
raise Stooges
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, Stooges.CURLY)
|
|
|
|
test_pickle_dump_load(self.assertIs, Stooges)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_pickle_int(self):
|
|
|
|
if isinstance(IntStooges, Exception):
|
|
|
|
raise IntStooges
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, IntStooges.CURLY)
|
|
|
|
test_pickle_dump_load(self.assertIs, IntStooges)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_pickle_float(self):
|
|
|
|
if isinstance(FloatStooges, Exception):
|
|
|
|
raise FloatStooges
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, FloatStooges.CURLY)
|
|
|
|
test_pickle_dump_load(self.assertIs, FloatStooges)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_pickle_enum_function(self):
|
|
|
|
if isinstance(Answer, Exception):
|
|
|
|
raise Answer
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, Answer.him)
|
|
|
|
test_pickle_dump_load(self.assertIs, Answer)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_pickle_enum_function_with_module(self):
|
|
|
|
if isinstance(Question, Exception):
|
|
|
|
raise Question
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, Question.who)
|
|
|
|
test_pickle_dump_load(self.assertIs, Question)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
2014-02-08 15:36:27 -04:00
|
|
|
def test_enum_function_with_qualname(self):
|
|
|
|
if isinstance(Theory, Exception):
|
|
|
|
raise Theory
|
|
|
|
self.assertEqual(Theory.__qualname__, 'spanish_inquisition')
|
|
|
|
|
|
|
|
def test_class_nested_enum_and_pickle_protocol_four(self):
|
|
|
|
# would normally just have this directly in the class namespace
|
|
|
|
class NestedEnum(Enum):
|
|
|
|
twigs = 'common'
|
|
|
|
shiny = 'rare'
|
|
|
|
|
|
|
|
self.__class__.NestedEnum = NestedEnum
|
|
|
|
self.NestedEnum.__qualname__ = '%s.NestedEnum' % self.__class__.__name__
|
2015-03-31 10:56:49 -03:00
|
|
|
test_pickle_dump_load(self.assertIs, self.NestedEnum.twigs)
|
2014-02-08 15:36:27 -04:00
|
|
|
|
2015-03-18 21:27:57 -03:00
|
|
|
def test_pickle_by_name(self):
|
|
|
|
class ReplaceGlobalInt(IntEnum):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
ReplaceGlobalInt.__reduce_ex__ = enum._reduce_ex_by_name
|
|
|
|
for proto in range(HIGHEST_PROTOCOL):
|
|
|
|
self.assertEqual(ReplaceGlobalInt.TWO.__reduce_ex__(proto), 'TWO')
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_exploding_pickle(self):
|
2014-02-08 15:36:27 -04:00
|
|
|
BadPickle = Enum(
|
|
|
|
'BadPickle', 'dill sweet bread-n-butter', module=__name__)
|
2013-06-14 20:55:46 -03:00
|
|
|
globals()['BadPickle'] = BadPickle
|
2014-02-08 15:36:27 -04:00
|
|
|
# now break BadPickle to test exception raising
|
|
|
|
enum._make_class_unpicklable(BadPickle)
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_exception(self.assertRaises, TypeError, BadPickle.dill)
|
|
|
|
test_pickle_exception(self.assertRaises, PicklingError, BadPickle)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_string_enum(self):
|
|
|
|
class SkillLevel(str, Enum):
|
|
|
|
master = 'what is the sound of one hand clapping?'
|
|
|
|
journeyman = 'why did the chicken cross the road?'
|
|
|
|
apprentice = 'knock, knock!'
|
|
|
|
self.assertEqual(SkillLevel.apprentice, 'knock, knock!')
|
|
|
|
|
|
|
|
def test_getattr_getitem(self):
|
|
|
|
class Period(Enum):
|
|
|
|
morning = 1
|
|
|
|
noon = 2
|
|
|
|
evening = 3
|
|
|
|
night = 4
|
|
|
|
self.assertIs(Period(2), Period.noon)
|
|
|
|
self.assertIs(getattr(Period, 'night'), Period.night)
|
|
|
|
self.assertIs(Period['morning'], Period.morning)
|
|
|
|
|
|
|
|
def test_getattr_dunder(self):
|
|
|
|
Season = self.Season
|
|
|
|
self.assertTrue(getattr(Season, '__eq__'))
|
|
|
|
|
|
|
|
def test_iteration_order(self):
|
|
|
|
class Season(Enum):
|
|
|
|
SUMMER = 2
|
|
|
|
WINTER = 4
|
|
|
|
AUTUMN = 3
|
|
|
|
SPRING = 1
|
|
|
|
self.assertEqual(
|
|
|
|
list(Season),
|
|
|
|
[Season.SUMMER, Season.WINTER, Season.AUTUMN, Season.SPRING],
|
|
|
|
)
|
|
|
|
|
2013-09-14 22:11:24 -03:00
|
|
|
def test_reversed_iteration_order(self):
|
|
|
|
self.assertEqual(
|
|
|
|
list(reversed(self.Season)),
|
|
|
|
[self.Season.WINTER, self.Season.AUTUMN, self.Season.SUMMER,
|
|
|
|
self.Season.SPRING]
|
|
|
|
)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_string(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', 'june july august')
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_string_with_start(self):
|
2014-09-17 00:35:55 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', 'june july august', start=10)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 10):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_string_list(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', ['june', 'july', 'august'])
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_string_list_with_start(self):
|
2014-09-17 00:35:55 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', ['june', 'july', 'august'], start=20)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 20):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_iterable(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = Enum(
|
|
|
|
'SummerMonth',
|
|
|
|
(('june', 1), ('july', 2), ('august', 3))
|
|
|
|
)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_from_dict(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = Enum(
|
|
|
|
'SummerMonth',
|
|
|
|
OrderedDict((('june', 1), ('july', 2), ('august', 3)))
|
|
|
|
)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(int(e.value), i)
|
|
|
|
self.assertNotEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_type(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', 'june july august', type=int)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_type_with_start(self):
|
2014-09-17 00:35:55 -03:00
|
|
|
SummerMonth = Enum('SummerMonth', 'june july august', type=int, start=30)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 30):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_type_from_subclass(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
SummerMonth = IntEnum('SummerMonth', 'june july august')
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 1):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2016-07-27 22:11:04 -03:00
|
|
|
def test_programmatic_function_type_from_subclass_with_start(self):
|
2014-09-17 00:35:55 -03:00
|
|
|
SummerMonth = IntEnum('SummerMonth', 'june july august', start=40)
|
|
|
|
lst = list(SummerMonth)
|
|
|
|
self.assertEqual(len(lst), len(SummerMonth))
|
|
|
|
self.assertEqual(len(SummerMonth), 3, SummerMonth)
|
|
|
|
self.assertEqual(
|
|
|
|
[SummerMonth.june, SummerMonth.july, SummerMonth.august],
|
|
|
|
lst,
|
|
|
|
)
|
|
|
|
for i, month in enumerate('june july august'.split(), 40):
|
|
|
|
e = SummerMonth(i)
|
|
|
|
self.assertEqual(e, i)
|
|
|
|
self.assertEqual(e.name, month)
|
|
|
|
self.assertIn(e, SummerMonth)
|
|
|
|
self.assertIs(type(e), SummerMonth)
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_subclassing(self):
|
|
|
|
if isinstance(Name, Exception):
|
|
|
|
raise Name
|
|
|
|
self.assertEqual(Name.BDFL, 'Guido van Rossum')
|
|
|
|
self.assertTrue(Name.BDFL, Name('Guido van Rossum'))
|
|
|
|
self.assertIs(Name.BDFL, getattr(Name, 'BDFL'))
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, Name.BDFL)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_extending(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class MoreColor(Color):
|
|
|
|
cyan = 4
|
|
|
|
magenta = 5
|
|
|
|
yellow = 6
|
2020-09-16 11:11:57 -03:00
|
|
|
with self.assertRaisesRegex(TypeError, "EvenMoreColor: cannot extend enumeration 'Color'"):
|
|
|
|
class EvenMoreColor(Color, IntEnum):
|
|
|
|
chartruese = 7
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_exclude_methods(self):
|
|
|
|
class whatever(Enum):
|
|
|
|
this = 'that'
|
|
|
|
these = 'those'
|
|
|
|
def really(self):
|
|
|
|
return 'no, not %s' % self.value
|
|
|
|
self.assertIsNot(type(whatever.really), whatever)
|
|
|
|
self.assertEqual(whatever.this.really(), 'no, not that')
|
|
|
|
|
|
|
|
def test_wrong_inheritance_order(self):
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class Wrong(Enum, str):
|
|
|
|
NotHere = 'error before this point'
|
|
|
|
|
|
|
|
def test_intenum_transitivity(self):
|
|
|
|
class number(IntEnum):
|
|
|
|
one = 1
|
|
|
|
two = 2
|
|
|
|
three = 3
|
|
|
|
class numero(IntEnum):
|
|
|
|
uno = 1
|
|
|
|
dos = 2
|
|
|
|
tres = 3
|
|
|
|
self.assertEqual(number.one, numero.uno)
|
|
|
|
self.assertEqual(number.two, numero.dos)
|
|
|
|
self.assertEqual(number.three, numero.tres)
|
|
|
|
|
|
|
|
def test_wrong_enum_in_call(self):
|
|
|
|
class Monochrome(Enum):
|
|
|
|
black = 0
|
|
|
|
white = 1
|
|
|
|
class Gender(Enum):
|
|
|
|
male = 0
|
|
|
|
female = 1
|
|
|
|
self.assertRaises(ValueError, Monochrome, Gender.male)
|
|
|
|
|
|
|
|
def test_wrong_enum_in_mixed_call(self):
|
|
|
|
class Monochrome(IntEnum):
|
|
|
|
black = 0
|
|
|
|
white = 1
|
|
|
|
class Gender(Enum):
|
|
|
|
male = 0
|
|
|
|
female = 1
|
|
|
|
self.assertRaises(ValueError, Monochrome, Gender.male)
|
|
|
|
|
|
|
|
def test_mixed_enum_in_call_1(self):
|
|
|
|
class Monochrome(IntEnum):
|
|
|
|
black = 0
|
|
|
|
white = 1
|
|
|
|
class Gender(IntEnum):
|
|
|
|
male = 0
|
|
|
|
female = 1
|
|
|
|
self.assertIs(Monochrome(Gender.female), Monochrome.white)
|
|
|
|
|
|
|
|
def test_mixed_enum_in_call_2(self):
|
|
|
|
class Monochrome(Enum):
|
|
|
|
black = 0
|
|
|
|
white = 1
|
|
|
|
class Gender(IntEnum):
|
|
|
|
male = 0
|
|
|
|
female = 1
|
|
|
|
self.assertIs(Monochrome(Gender.male), Monochrome.black)
|
|
|
|
|
|
|
|
def test_flufl_enum(self):
|
|
|
|
class Fluflnum(Enum):
|
|
|
|
def __int__(self):
|
|
|
|
return int(self.value)
|
|
|
|
class MailManOptions(Fluflnum):
|
|
|
|
option1 = 1
|
|
|
|
option2 = 2
|
|
|
|
option3 = 3
|
|
|
|
self.assertEqual(int(MailManOptions.option1), 1)
|
|
|
|
|
2013-08-04 12:42:23 -03:00
|
|
|
def test_introspection(self):
|
|
|
|
class Number(IntEnum):
|
|
|
|
one = 100
|
|
|
|
two = 200
|
|
|
|
self.assertIs(Number.one._member_type_, int)
|
|
|
|
self.assertIs(Number._member_type_, int)
|
|
|
|
class String(str, Enum):
|
|
|
|
yarn = 'soft'
|
|
|
|
rope = 'rough'
|
|
|
|
wire = 'hard'
|
|
|
|
self.assertIs(String.yarn._member_type_, str)
|
|
|
|
self.assertIs(String._member_type_, str)
|
|
|
|
class Plain(Enum):
|
|
|
|
vanilla = 'white'
|
|
|
|
one = 1
|
|
|
|
self.assertIs(Plain.vanilla._member_type_, object)
|
|
|
|
self.assertIs(Plain._member_type_, object)
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_no_such_enum_member(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
Color(4)
|
|
|
|
with self.assertRaises(KeyError):
|
|
|
|
Color['chartreuse']
|
|
|
|
|
|
|
|
def test_new_repr(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
def __repr__(self):
|
|
|
|
return "don't you just love shades of %s?" % self.name
|
|
|
|
self.assertEqual(
|
|
|
|
repr(Color.blue),
|
|
|
|
"don't you just love shades of blue?",
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_inherited_repr(self):
|
|
|
|
class MyEnum(Enum):
|
|
|
|
def __repr__(self):
|
|
|
|
return "My name is %s." % self.name
|
|
|
|
class MyIntEnum(int, MyEnum):
|
|
|
|
this = 1
|
|
|
|
that = 2
|
|
|
|
theother = 3
|
|
|
|
self.assertEqual(repr(MyIntEnum.that), "My name is that.")
|
|
|
|
|
|
|
|
def test_multiple_mixin_mro(self):
|
|
|
|
class auto_enum(type(Enum)):
|
|
|
|
def __new__(metacls, cls, bases, classdict):
|
|
|
|
temp = type(classdict)()
|
2020-12-09 21:12:11 -04:00
|
|
|
temp._cls_name = cls
|
2013-06-14 20:55:46 -03:00
|
|
|
names = set(classdict._member_names)
|
|
|
|
i = 0
|
|
|
|
for k in classdict._member_names:
|
|
|
|
v = classdict[k]
|
|
|
|
if v is Ellipsis:
|
|
|
|
v = i
|
|
|
|
else:
|
|
|
|
i = v
|
|
|
|
i += 1
|
|
|
|
temp[k] = v
|
|
|
|
for k, v in classdict.items():
|
|
|
|
if k not in names:
|
|
|
|
temp[k] = v
|
|
|
|
return super(auto_enum, metacls).__new__(
|
|
|
|
metacls, cls, bases, temp)
|
|
|
|
|
|
|
|
class AutoNumberedEnum(Enum, metaclass=auto_enum):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class AutoIntEnum(IntEnum, metaclass=auto_enum):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class TestAutoNumber(AutoNumberedEnum):
|
|
|
|
a = ...
|
|
|
|
b = 3
|
|
|
|
c = ...
|
|
|
|
|
|
|
|
class TestAutoInt(AutoIntEnum):
|
|
|
|
a = ...
|
|
|
|
b = 3
|
|
|
|
c = ...
|
|
|
|
|
|
|
|
def test_subclasses_with_getnewargs(self):
|
|
|
|
class NamedInt(int):
|
2014-02-06 21:28:50 -04:00
|
|
|
__qualname__ = 'NamedInt' # needed for pickle protocol 4
|
2013-06-14 20:55:46 -03:00
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
def __getnewargs__(self):
|
|
|
|
return self._args
|
|
|
|
@property
|
2014-02-08 15:36:27 -04:00
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
2020-09-15 16:27:06 -03:00
|
|
|
temp,
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
|
|
|
__qualname__ = 'NEI' # needed for pickle protocol 4
|
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
|
|
|
test_pickle_dump_load(self.assertEqual, NI5, 5)
|
|
|
|
self.assertEqual(NEI.y.value, 2)
|
|
|
|
test_pickle_dump_load(self.assertIs, NEI.y)
|
2014-02-18 16:37:12 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI)
|
2014-02-08 15:36:27 -04:00
|
|
|
|
|
|
|
def test_subclasses_with_getnewargs_ex(self):
|
|
|
|
class NamedInt(int):
|
|
|
|
__qualname__ = 'NamedInt' # needed for pickle protocol 4
|
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
def __getnewargs_ex__(self):
|
|
|
|
return self._args, {}
|
|
|
|
@property
|
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
2020-09-15 16:27:06 -03:00
|
|
|
temp,
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
|
|
|
__qualname__ = 'NEI' # needed for pickle protocol 4
|
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
2015-03-31 10:56:49 -03:00
|
|
|
test_pickle_dump_load(self.assertEqual, NI5, 5)
|
2014-02-08 15:36:27 -04:00
|
|
|
self.assertEqual(NEI.y.value, 2)
|
2015-03-31 10:56:49 -03:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI.y)
|
2014-02-18 16:37:12 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI)
|
2014-02-08 15:36:27 -04:00
|
|
|
|
|
|
|
def test_subclasses_with_reduce(self):
|
|
|
|
class NamedInt(int):
|
|
|
|
__qualname__ = 'NamedInt' # needed for pickle protocol 4
|
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
def __reduce__(self):
|
|
|
|
return self.__class__, self._args
|
|
|
|
@property
|
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
2020-09-15 16:27:06 -03:00
|
|
|
temp,
|
|
|
|
)
|
2014-02-08 15:36:27 -04:00
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
|
|
|
__qualname__ = 'NEI' # needed for pickle protocol 4
|
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
|
|
|
test_pickle_dump_load(self.assertEqual, NI5, 5)
|
|
|
|
self.assertEqual(NEI.y.value, 2)
|
|
|
|
test_pickle_dump_load(self.assertIs, NEI.y)
|
2014-02-18 16:37:12 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI)
|
2014-02-08 15:36:27 -04:00
|
|
|
|
|
|
|
def test_subclasses_with_reduce_ex(self):
|
|
|
|
class NamedInt(int):
|
|
|
|
__qualname__ = 'NamedInt' # needed for pickle protocol 4
|
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
def __reduce_ex__(self, proto):
|
|
|
|
return self.__class__, self._args
|
|
|
|
@property
|
2013-06-14 20:55:46 -03:00
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2013-06-14 20:55:46 -03:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
2020-09-15 16:27:06 -03:00
|
|
|
temp,
|
|
|
|
)
|
2013-06-14 20:55:46 -03:00
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
2014-02-06 21:28:50 -04:00
|
|
|
__qualname__ = 'NEI' # needed for pickle protocol 4
|
2013-06-14 20:55:46 -03:00
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertEqual, NI5, 5)
|
2013-06-14 20:55:46 -03:00
|
|
|
self.assertEqual(NEI.y.value, 2)
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI.y)
|
2014-02-18 16:37:12 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, NEI)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
2014-02-18 16:37:12 -04:00
|
|
|
def test_subclasses_without_direct_pickle_support(self):
|
2013-06-14 20:55:46 -03:00
|
|
|
class NamedInt(int):
|
2014-02-06 21:28:50 -04:00
|
|
|
__qualname__ = 'NamedInt'
|
2013-06-14 20:55:46 -03:00
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
@property
|
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2013-06-14 20:55:46 -03:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
|
|
|
temp )
|
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
2014-02-06 21:28:50 -04:00
|
|
|
__qualname__ = 'NEI'
|
2013-06-14 20:55:46 -03:00
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
|
|
|
self.assertEqual(NEI.y.value, 2)
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_exception(self.assertRaises, TypeError, NEI.x)
|
|
|
|
test_pickle_exception(self.assertRaises, PicklingError, NEI)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
2014-02-18 16:37:12 -04:00
|
|
|
def test_subclasses_without_direct_pickle_support_using_name(self):
|
|
|
|
class NamedInt(int):
|
|
|
|
__qualname__ = 'NamedInt'
|
|
|
|
def __new__(cls, *args):
|
|
|
|
_args = args
|
|
|
|
name, *args = args
|
|
|
|
if len(args) == 0:
|
|
|
|
raise TypeError("name and value must be specified")
|
|
|
|
self = int.__new__(cls, *args)
|
|
|
|
self._intname = name
|
|
|
|
self._args = _args
|
|
|
|
return self
|
|
|
|
@property
|
|
|
|
def __name__(self):
|
|
|
|
return self._intname
|
|
|
|
def __repr__(self):
|
|
|
|
# repr() is updated to include the name and type info
|
2020-09-15 16:27:06 -03:00
|
|
|
return "{}({!r}, {})".format(
|
|
|
|
type(self).__name__,
|
|
|
|
self.__name__,
|
|
|
|
int.__repr__(self),
|
|
|
|
)
|
2014-02-18 16:37:12 -04:00
|
|
|
def __str__(self):
|
|
|
|
# str() is unchanged, even if it relies on the repr() fallback
|
|
|
|
base = int
|
|
|
|
base_str = base.__str__
|
|
|
|
if base_str.__objclass__ is object:
|
|
|
|
return base.__repr__(self)
|
|
|
|
return base_str(self)
|
|
|
|
# for simplicity, we only define one operator that
|
|
|
|
# propagates expressions
|
|
|
|
def __add__(self, other):
|
|
|
|
temp = int(self) + int( other)
|
|
|
|
if isinstance(self, NamedInt) and isinstance(other, NamedInt):
|
|
|
|
return NamedInt(
|
|
|
|
'({0} + {1})'.format(self.__name__, other.__name__),
|
2020-09-15 16:27:06 -03:00
|
|
|
temp,
|
|
|
|
)
|
2014-02-18 16:37:12 -04:00
|
|
|
else:
|
|
|
|
return temp
|
|
|
|
|
|
|
|
class NEI(NamedInt, Enum):
|
|
|
|
__qualname__ = 'NEI'
|
|
|
|
x = ('the-x', 1)
|
|
|
|
y = ('the-y', 2)
|
|
|
|
def __reduce_ex__(self, proto):
|
|
|
|
return getattr, (self.__class__, self._name_)
|
|
|
|
|
|
|
|
self.assertIs(NEI.__new__, Enum.__new__)
|
|
|
|
self.assertEqual(repr(NEI.x + NEI.y), "NamedInt('(the-x + the-y)', 3)")
|
|
|
|
globals()['NamedInt'] = NamedInt
|
|
|
|
globals()['NEI'] = NEI
|
|
|
|
NI5 = NamedInt('test', 5)
|
|
|
|
self.assertEqual(NI5, 5)
|
|
|
|
self.assertEqual(NEI.y.value, 2)
|
|
|
|
test_pickle_dump_load(self.assertIs, NEI.y)
|
|
|
|
test_pickle_dump_load(self.assertIs, NEI)
|
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_tuple_subclass(self):
|
|
|
|
class SomeTuple(tuple, Enum):
|
2014-02-06 21:28:50 -04:00
|
|
|
__qualname__ = 'SomeTuple' # needed for pickle protocol 4
|
2013-06-14 20:55:46 -03:00
|
|
|
first = (1, 'for the money')
|
|
|
|
second = (2, 'for the show')
|
|
|
|
third = (3, 'for the music')
|
|
|
|
self.assertIs(type(SomeTuple.first), SomeTuple)
|
|
|
|
self.assertIsInstance(SomeTuple.second, tuple)
|
|
|
|
self.assertEqual(SomeTuple.third, (3, 'for the music'))
|
|
|
|
globals()['SomeTuple'] = SomeTuple
|
2014-02-06 21:28:50 -04:00
|
|
|
test_pickle_dump_load(self.assertIs, SomeTuple.first)
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_duplicate_values_give_unique_enum_items(self):
|
|
|
|
class AutoNumber(Enum):
|
|
|
|
first = ()
|
|
|
|
second = ()
|
|
|
|
third = ()
|
|
|
|
def __new__(cls):
|
|
|
|
value = len(cls.__members__) + 1
|
|
|
|
obj = object.__new__(cls)
|
2013-07-19 23:47:21 -03:00
|
|
|
obj._value_ = value
|
2013-06-14 20:55:46 -03:00
|
|
|
return obj
|
|
|
|
def __int__(self):
|
2013-07-19 23:47:21 -03:00
|
|
|
return int(self._value_)
|
2013-06-14 20:55:46 -03:00
|
|
|
self.assertEqual(
|
|
|
|
list(AutoNumber),
|
|
|
|
[AutoNumber.first, AutoNumber.second, AutoNumber.third],
|
|
|
|
)
|
|
|
|
self.assertEqual(int(AutoNumber.second), 2)
|
2013-07-19 23:35:56 -03:00
|
|
|
self.assertEqual(AutoNumber.third.value, 3)
|
2013-06-14 20:55:46 -03:00
|
|
|
self.assertIs(AutoNumber(1), AutoNumber.first)
|
|
|
|
|
|
|
|
def test_inherited_new_from_enhanced_enum(self):
|
|
|
|
class AutoNumber(Enum):
|
|
|
|
def __new__(cls):
|
|
|
|
value = len(cls.__members__) + 1
|
|
|
|
obj = object.__new__(cls)
|
2013-07-19 23:47:21 -03:00
|
|
|
obj._value_ = value
|
2013-06-14 20:55:46 -03:00
|
|
|
return obj
|
|
|
|
def __int__(self):
|
2013-07-19 23:47:21 -03:00
|
|
|
return int(self._value_)
|
2013-06-14 20:55:46 -03:00
|
|
|
class Color(AutoNumber):
|
|
|
|
red = ()
|
|
|
|
green = ()
|
|
|
|
blue = ()
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.green, Color.blue])
|
|
|
|
self.assertEqual(list(map(int, Color)), [1, 2, 3])
|
|
|
|
|
|
|
|
def test_inherited_new_from_mixed_enum(self):
|
|
|
|
class AutoNumber(IntEnum):
|
|
|
|
def __new__(cls):
|
|
|
|
value = len(cls.__members__) + 1
|
|
|
|
obj = int.__new__(cls, value)
|
2013-07-19 23:47:21 -03:00
|
|
|
obj._value_ = value
|
2013-06-14 20:55:46 -03:00
|
|
|
return obj
|
|
|
|
class Color(AutoNumber):
|
|
|
|
red = ()
|
|
|
|
green = ()
|
|
|
|
blue = ()
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.green, Color.blue])
|
|
|
|
self.assertEqual(list(map(int, Color)), [1, 2, 3])
|
|
|
|
|
2013-11-13 18:25:45 -04:00
|
|
|
def test_equality(self):
|
|
|
|
class OrdinaryEnum(Enum):
|
|
|
|
a = 1
|
2019-08-08 02:43:18 -03:00
|
|
|
self.assertEqual(ALWAYS_EQ, OrdinaryEnum.a)
|
|
|
|
self.assertEqual(OrdinaryEnum.a, ALWAYS_EQ)
|
2013-11-13 18:25:45 -04:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_ordered_mixin(self):
|
|
|
|
class OrderedEnum(Enum):
|
|
|
|
def __ge__(self, other):
|
|
|
|
if self.__class__ is other.__class__:
|
2013-07-19 23:47:21 -03:00
|
|
|
return self._value_ >= other._value_
|
2013-06-14 20:55:46 -03:00
|
|
|
return NotImplemented
|
|
|
|
def __gt__(self, other):
|
|
|
|
if self.__class__ is other.__class__:
|
2013-07-19 23:47:21 -03:00
|
|
|
return self._value_ > other._value_
|
2013-06-14 20:55:46 -03:00
|
|
|
return NotImplemented
|
|
|
|
def __le__(self, other):
|
|
|
|
if self.__class__ is other.__class__:
|
2013-07-19 23:47:21 -03:00
|
|
|
return self._value_ <= other._value_
|
2013-06-14 20:55:46 -03:00
|
|
|
return NotImplemented
|
|
|
|
def __lt__(self, other):
|
|
|
|
if self.__class__ is other.__class__:
|
2013-07-19 23:47:21 -03:00
|
|
|
return self._value_ < other._value_
|
2013-06-14 20:55:46 -03:00
|
|
|
return NotImplemented
|
|
|
|
class Grade(OrderedEnum):
|
|
|
|
A = 5
|
|
|
|
B = 4
|
|
|
|
C = 3
|
|
|
|
D = 2
|
|
|
|
F = 1
|
|
|
|
self.assertGreater(Grade.A, Grade.B)
|
|
|
|
self.assertLessEqual(Grade.F, Grade.C)
|
|
|
|
self.assertLess(Grade.D, Grade.A)
|
|
|
|
self.assertGreaterEqual(Grade.B, Grade.B)
|
2013-11-13 18:25:45 -04:00
|
|
|
self.assertEqual(Grade.B, Grade.B)
|
|
|
|
self.assertNotEqual(Grade.C, Grade.D)
|
2013-07-19 23:47:21 -03:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
def test_extending2(self):
|
|
|
|
class Shade(Enum):
|
|
|
|
def shade(self):
|
|
|
|
print(self.name)
|
|
|
|
class Color(Shade):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class MoreColor(Color):
|
|
|
|
cyan = 4
|
|
|
|
magenta = 5
|
|
|
|
yellow = 6
|
|
|
|
|
|
|
|
def test_extending3(self):
|
|
|
|
class Shade(Enum):
|
|
|
|
def shade(self):
|
|
|
|
return self.name
|
|
|
|
class Color(Shade):
|
|
|
|
def hex(self):
|
|
|
|
return '%s hexlified!' % self.value
|
|
|
|
class MoreColor(Color):
|
|
|
|
cyan = 4
|
|
|
|
magenta = 5
|
|
|
|
yellow = 6
|
|
|
|
self.assertEqual(MoreColor.magenta.hex(), '5 hexlified!')
|
|
|
|
|
2018-09-12 14:28:53 -03:00
|
|
|
def test_subclass_duplicate_name(self):
|
|
|
|
class Base(Enum):
|
|
|
|
def test(self):
|
|
|
|
pass
|
|
|
|
class Test(Base):
|
|
|
|
test = 1
|
|
|
|
self.assertIs(type(Test.test), Test)
|
|
|
|
|
|
|
|
def test_subclass_duplicate_name_dynamic(self):
|
|
|
|
from types import DynamicClassAttribute
|
|
|
|
class Base(Enum):
|
|
|
|
@DynamicClassAttribute
|
|
|
|
def test(self):
|
|
|
|
return 'dynamic'
|
|
|
|
class Test(Base):
|
|
|
|
test = 1
|
|
|
|
self.assertEqual(Test.test.test, 'dynamic')
|
2021-01-13 03:47:57 -04:00
|
|
|
class Base2(Enum):
|
|
|
|
@enum.property
|
|
|
|
def flash(self):
|
|
|
|
return 'flashy dynamic'
|
|
|
|
class Test(Base2):
|
|
|
|
flash = 1
|
|
|
|
self.assertEqual(Test.flash.flash, 'flashy dynamic')
|
2013-06-14 20:55:46 -03:00
|
|
|
|
|
|
|
def test_no_duplicates(self):
|
|
|
|
class UniqueEnum(Enum):
|
|
|
|
def __init__(self, *args):
|
|
|
|
cls = self.__class__
|
|
|
|
if any(self.value == e.value for e in cls):
|
|
|
|
a = self.name
|
|
|
|
e = cls(self.value).name
|
|
|
|
raise ValueError(
|
|
|
|
"aliases not allowed in UniqueEnum: %r --> %r"
|
|
|
|
% (a, e)
|
|
|
|
)
|
|
|
|
class Color(UniqueEnum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
class Color(UniqueEnum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
grene = 2
|
|
|
|
|
|
|
|
def test_init(self):
|
|
|
|
class Planet(Enum):
|
|
|
|
MERCURY = (3.303e+23, 2.4397e6)
|
|
|
|
VENUS = (4.869e+24, 6.0518e6)
|
|
|
|
EARTH = (5.976e+24, 6.37814e6)
|
|
|
|
MARS = (6.421e+23, 3.3972e6)
|
|
|
|
JUPITER = (1.9e+27, 7.1492e7)
|
|
|
|
SATURN = (5.688e+26, 6.0268e7)
|
|
|
|
URANUS = (8.686e+25, 2.5559e7)
|
|
|
|
NEPTUNE = (1.024e+26, 2.4746e7)
|
|
|
|
def __init__(self, mass, radius):
|
|
|
|
self.mass = mass # in kilograms
|
|
|
|
self.radius = radius # in meters
|
|
|
|
@property
|
|
|
|
def surface_gravity(self):
|
|
|
|
# universal gravitational constant (m3 kg-1 s-2)
|
|
|
|
G = 6.67300E-11
|
|
|
|
return G * self.mass / (self.radius * self.radius)
|
|
|
|
self.assertEqual(round(Planet.EARTH.surface_gravity, 2), 9.80)
|
|
|
|
self.assertEqual(Planet.EARTH.value, (5.976e+24, 6.37814e6))
|
|
|
|
|
2018-01-22 11:56:37 -04:00
|
|
|
def test_ignore(self):
|
|
|
|
class Period(timedelta, Enum):
|
|
|
|
'''
|
|
|
|
different lengths of time
|
|
|
|
'''
|
|
|
|
def __new__(cls, value, period):
|
|
|
|
obj = timedelta.__new__(cls, value)
|
|
|
|
obj._value_ = value
|
|
|
|
obj.period = period
|
|
|
|
return obj
|
|
|
|
_ignore_ = 'Period i'
|
|
|
|
Period = vars()
|
|
|
|
for i in range(13):
|
|
|
|
Period['month_%d' % i] = i*30, 'month'
|
|
|
|
for i in range(53):
|
|
|
|
Period['week_%d' % i] = i*7, 'week'
|
|
|
|
for i in range(32):
|
|
|
|
Period['day_%d' % i] = i, 'day'
|
|
|
|
OneDay = day_1
|
|
|
|
OneWeek = week_1
|
|
|
|
OneMonth = month_1
|
|
|
|
self.assertFalse(hasattr(Period, '_ignore_'))
|
|
|
|
self.assertFalse(hasattr(Period, 'Period'))
|
|
|
|
self.assertFalse(hasattr(Period, 'i'))
|
|
|
|
self.assertTrue(isinstance(Period.day_1, timedelta))
|
|
|
|
self.assertTrue(Period.month_1 is Period.day_30)
|
|
|
|
self.assertTrue(Period.week_4 is Period.day_28)
|
|
|
|
|
2013-07-19 23:35:56 -03:00
|
|
|
def test_nonhash_value(self):
|
|
|
|
class AutoNumberInAList(Enum):
|
|
|
|
def __new__(cls):
|
|
|
|
value = [len(cls.__members__) + 1]
|
|
|
|
obj = object.__new__(cls)
|
2013-07-19 23:47:21 -03:00
|
|
|
obj._value_ = value
|
2013-07-19 23:35:56 -03:00
|
|
|
return obj
|
|
|
|
class ColorInAList(AutoNumberInAList):
|
|
|
|
red = ()
|
|
|
|
green = ()
|
|
|
|
blue = ()
|
|
|
|
self.assertEqual(list(ColorInAList), [ColorInAList.red, ColorInAList.green, ColorInAList.blue])
|
2013-10-16 23:09:31 -03:00
|
|
|
for enum, value in zip(ColorInAList, range(3)):
|
|
|
|
value += 1
|
|
|
|
self.assertEqual(enum.value, [value])
|
|
|
|
self.assertIs(ColorInAList([value]), enum)
|
2013-07-19 23:35:56 -03:00
|
|
|
|
2013-07-25 17:50:45 -03:00
|
|
|
def test_conflicting_types_resolved_in_new(self):
|
|
|
|
class LabelledIntEnum(int, Enum):
|
|
|
|
def __new__(cls, *args):
|
|
|
|
value, label = args
|
|
|
|
obj = int.__new__(cls, value)
|
|
|
|
obj.label = label
|
|
|
|
obj._value_ = value
|
|
|
|
return obj
|
|
|
|
|
|
|
|
class LabelledList(LabelledIntEnum):
|
|
|
|
unprocessed = (1, "Unprocessed")
|
|
|
|
payment_complete = (2, "Payment Complete")
|
|
|
|
|
|
|
|
self.assertEqual(list(LabelledList), [LabelledList.unprocessed, LabelledList.payment_complete])
|
|
|
|
self.assertEqual(LabelledList.unprocessed, 1)
|
|
|
|
self.assertEqual(LabelledList(1), LabelledList.unprocessed)
|
2013-07-19 23:35:56 -03:00
|
|
|
|
2016-09-11 03:36:59 -03:00
|
|
|
def test_auto_number(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = auto()
|
|
|
|
blue = auto()
|
|
|
|
green = auto()
|
|
|
|
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
|
|
|
|
self.assertEqual(Color.red.value, 1)
|
|
|
|
self.assertEqual(Color.blue.value, 2)
|
|
|
|
self.assertEqual(Color.green.value, 3)
|
|
|
|
|
|
|
|
def test_auto_name(self):
|
|
|
|
class Color(Enum):
|
|
|
|
def _generate_next_value_(name, start, count, last):
|
|
|
|
return name
|
|
|
|
red = auto()
|
|
|
|
blue = auto()
|
|
|
|
green = auto()
|
|
|
|
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
|
|
|
|
self.assertEqual(Color.red.value, 'red')
|
|
|
|
self.assertEqual(Color.blue.value, 'blue')
|
|
|
|
self.assertEqual(Color.green.value, 'green')
|
|
|
|
|
|
|
|
def test_auto_name_inherit(self):
|
|
|
|
class AutoNameEnum(Enum):
|
|
|
|
def _generate_next_value_(name, start, count, last):
|
|
|
|
return name
|
|
|
|
class Color(AutoNameEnum):
|
|
|
|
red = auto()
|
|
|
|
blue = auto()
|
|
|
|
green = auto()
|
|
|
|
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
|
|
|
|
self.assertEqual(Color.red.value, 'red')
|
|
|
|
self.assertEqual(Color.blue.value, 'blue')
|
|
|
|
self.assertEqual(Color.green.value, 'green')
|
|
|
|
|
|
|
|
def test_auto_garbage(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 'red'
|
|
|
|
blue = auto()
|
|
|
|
self.assertEqual(Color.blue.value, 1)
|
|
|
|
|
|
|
|
def test_auto_garbage_corrected(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 'red'
|
|
|
|
blue = 2
|
|
|
|
green = auto()
|
|
|
|
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
|
|
|
|
self.assertEqual(Color.red.value, 'red')
|
|
|
|
self.assertEqual(Color.blue.value, 2)
|
|
|
|
self.assertEqual(Color.green.value, 3)
|
|
|
|
|
2020-04-28 14:20:55 -03:00
|
|
|
def test_auto_order(self):
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
class Color(Enum):
|
|
|
|
red = auto()
|
|
|
|
green = auto()
|
|
|
|
blue = auto()
|
|
|
|
def _generate_next_value_(name, start, count, last):
|
|
|
|
return name
|
|
|
|
|
2020-09-16 16:37:54 -03:00
|
|
|
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')
|
2020-04-28 14:20:55 -03:00
|
|
|
|
2016-09-18 17:15:41 -03:00
|
|
|
def test_duplicate_auto(self):
|
|
|
|
class Dupes(Enum):
|
|
|
|
first = primero = auto()
|
|
|
|
second = auto()
|
|
|
|
third = auto()
|
|
|
|
self.assertEqual([Dupes.first, Dupes.second, Dupes.third], list(Dupes))
|
|
|
|
|
2020-09-16 14:26:50 -03:00
|
|
|
def test_default_missing(self):
|
|
|
|
class Color(Enum):
|
|
|
|
RED = 1
|
|
|
|
GREEN = 2
|
|
|
|
BLUE = 3
|
|
|
|
try:
|
|
|
|
Color(7)
|
|
|
|
except ValueError as exc:
|
|
|
|
self.assertTrue(exc.__context__ is None)
|
|
|
|
else:
|
|
|
|
raise Exception('Exception not raised.')
|
|
|
|
|
2018-09-12 15:43:34 -03:00
|
|
|
def test_missing(self):
|
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
@classmethod
|
|
|
|
def _missing_(cls, item):
|
|
|
|
if item == 'three':
|
|
|
|
return cls.blue
|
|
|
|
elif item == 'bad return':
|
|
|
|
# trigger internal error
|
|
|
|
return 5
|
|
|
|
elif item == 'error out':
|
|
|
|
raise ZeroDivisionError
|
|
|
|
else:
|
|
|
|
# trigger not found
|
|
|
|
return None
|
|
|
|
self.assertIs(Color('three'), Color.blue)
|
2020-09-16 14:26:50 -03:00
|
|
|
try:
|
|
|
|
Color(7)
|
|
|
|
except ValueError as exc:
|
|
|
|
self.assertTrue(exc.__context__ is None)
|
|
|
|
else:
|
|
|
|
raise Exception('Exception not raised.')
|
2018-09-12 15:43:34 -03:00
|
|
|
try:
|
|
|
|
Color('bad return')
|
|
|
|
except TypeError as exc:
|
|
|
|
self.assertTrue(isinstance(exc.__context__, ValueError))
|
|
|
|
else:
|
|
|
|
raise Exception('Exception not raised.')
|
|
|
|
try:
|
|
|
|
Color('error out')
|
|
|
|
except ZeroDivisionError as exc:
|
|
|
|
self.assertTrue(isinstance(exc.__context__, ValueError))
|
|
|
|
else:
|
|
|
|
raise Exception('Exception not raised.')
|
|
|
|
|
2021-04-12 12:51:20 -03:00
|
|
|
def test_missing_exceptions_reset(self):
|
|
|
|
import weakref
|
|
|
|
#
|
|
|
|
class TestEnum(enum.Enum):
|
|
|
|
VAL1 = 'val1'
|
|
|
|
VAL2 = 'val2'
|
|
|
|
#
|
|
|
|
class Class1:
|
|
|
|
def __init__(self):
|
|
|
|
# Gracefully handle an exception of our own making
|
|
|
|
try:
|
|
|
|
raise ValueError()
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
#
|
|
|
|
class Class2:
|
|
|
|
def __init__(self):
|
|
|
|
# Gracefully handle an exception of Enum's making
|
|
|
|
try:
|
|
|
|
TestEnum('invalid_value')
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
# No strong refs here so these are free to die.
|
|
|
|
class_1_ref = weakref.ref(Class1())
|
|
|
|
class_2_ref = weakref.ref(Class2())
|
|
|
|
#
|
|
|
|
# The exception raised by Enum creates a reference loop and thus
|
|
|
|
# Class2 instances will stick around until the next gargage collection
|
|
|
|
# cycle, unlike Class1.
|
|
|
|
self.assertIs(class_1_ref(), None)
|
|
|
|
self.assertIs(class_2_ref(), None)
|
|
|
|
|
2018-09-21 23:03:09 -03:00
|
|
|
def test_multiple_mixin(self):
|
|
|
|
class MaxMixin:
|
|
|
|
@classproperty
|
|
|
|
def MAX(cls):
|
|
|
|
max = len(cls)
|
|
|
|
cls.MAX = max
|
|
|
|
return max
|
|
|
|
class StrMixin:
|
|
|
|
def __str__(self):
|
|
|
|
return self._name_.lower()
|
|
|
|
class SomeEnum(Enum):
|
|
|
|
def behavior(self):
|
|
|
|
return 'booyah'
|
|
|
|
class AnotherEnum(Enum):
|
|
|
|
def behavior(self):
|
|
|
|
return 'nuhuh!'
|
|
|
|
def social(self):
|
|
|
|
return "what's up?"
|
|
|
|
class Color(MaxMixin, Enum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 3)
|
|
|
|
self.assertEqual(Color.MAX, 3)
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Color.BLUE), 'BLUE')
|
2018-09-21 23:03:09 -03:00
|
|
|
class Color(MaxMixin, StrMixin, Enum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 3)
|
|
|
|
self.assertEqual(Color.MAX, 3)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
class Color(StrMixin, MaxMixin, Enum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 3)
|
|
|
|
self.assertEqual(Color.MAX, 3)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
class CoolColor(StrMixin, SomeEnum, Enum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(CoolColor.RED.value, 1)
|
|
|
|
self.assertEqual(CoolColor.GREEN.value, 2)
|
|
|
|
self.assertEqual(CoolColor.BLUE.value, 3)
|
|
|
|
self.assertEqual(str(CoolColor.BLUE), 'blue')
|
|
|
|
self.assertEqual(CoolColor.RED.behavior(), 'booyah')
|
|
|
|
class CoolerColor(StrMixin, AnotherEnum, Enum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(CoolerColor.RED.value, 1)
|
|
|
|
self.assertEqual(CoolerColor.GREEN.value, 2)
|
|
|
|
self.assertEqual(CoolerColor.BLUE.value, 3)
|
|
|
|
self.assertEqual(str(CoolerColor.BLUE), 'blue')
|
|
|
|
self.assertEqual(CoolerColor.RED.behavior(), 'nuhuh!')
|
|
|
|
self.assertEqual(CoolerColor.RED.social(), "what's up?")
|
|
|
|
class CoolestColor(StrMixin, SomeEnum, AnotherEnum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(CoolestColor.RED.value, 1)
|
|
|
|
self.assertEqual(CoolestColor.GREEN.value, 2)
|
|
|
|
self.assertEqual(CoolestColor.BLUE.value, 3)
|
|
|
|
self.assertEqual(str(CoolestColor.BLUE), 'blue')
|
|
|
|
self.assertEqual(CoolestColor.RED.behavior(), 'booyah')
|
|
|
|
self.assertEqual(CoolestColor.RED.social(), "what's up?")
|
|
|
|
class ConfusedColor(StrMixin, AnotherEnum, SomeEnum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(ConfusedColor.RED.value, 1)
|
|
|
|
self.assertEqual(ConfusedColor.GREEN.value, 2)
|
|
|
|
self.assertEqual(ConfusedColor.BLUE.value, 3)
|
|
|
|
self.assertEqual(str(ConfusedColor.BLUE), 'blue')
|
|
|
|
self.assertEqual(ConfusedColor.RED.behavior(), 'nuhuh!')
|
|
|
|
self.assertEqual(ConfusedColor.RED.social(), "what's up?")
|
|
|
|
class ReformedColor(StrMixin, IntEnum, SomeEnum, AnotherEnum):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(ReformedColor.RED.value, 1)
|
|
|
|
self.assertEqual(ReformedColor.GREEN.value, 2)
|
|
|
|
self.assertEqual(ReformedColor.BLUE.value, 3)
|
|
|
|
self.assertEqual(str(ReformedColor.BLUE), 'blue')
|
|
|
|
self.assertEqual(ReformedColor.RED.behavior(), 'booyah')
|
|
|
|
self.assertEqual(ConfusedColor.RED.social(), "what's up?")
|
|
|
|
self.assertTrue(issubclass(ReformedColor, int))
|
|
|
|
|
2018-10-06 03:29:36 -03:00
|
|
|
def test_multiple_inherited_mixin(self):
|
|
|
|
@unique
|
|
|
|
class Decision1(StrEnum):
|
|
|
|
REVERT = "REVERT"
|
|
|
|
REVERT_ALL = "REVERT_ALL"
|
|
|
|
RETRY = "RETRY"
|
|
|
|
class MyEnum(StrEnum):
|
|
|
|
pass
|
|
|
|
@unique
|
|
|
|
class Decision2(MyEnum):
|
|
|
|
REVERT = "REVERT"
|
|
|
|
REVERT_ALL = "REVERT_ALL"
|
|
|
|
RETRY = "RETRY"
|
|
|
|
|
2020-12-07 04:17:31 -04:00
|
|
|
def test_multiple_mixin_inherited(self):
|
|
|
|
class MyInt(int):
|
|
|
|
def __new__(cls, value):
|
|
|
|
return super().__new__(cls, value)
|
|
|
|
|
|
|
|
class HexMixin:
|
|
|
|
def __repr__(self):
|
|
|
|
return hex(self)
|
|
|
|
|
|
|
|
class MyIntEnum(HexMixin, MyInt, enum.Enum):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class Foo(MyIntEnum):
|
|
|
|
TEST = 1
|
|
|
|
self.assertTrue(isinstance(Foo.TEST, MyInt))
|
|
|
|
self.assertEqual(repr(Foo.TEST), "0x1")
|
|
|
|
|
|
|
|
class Fee(MyIntEnum):
|
|
|
|
TEST = 1
|
|
|
|
def __new__(cls, value):
|
|
|
|
value += 1
|
|
|
|
member = int.__new__(cls, value)
|
|
|
|
member._value_ = value
|
|
|
|
return member
|
|
|
|
self.assertEqual(Fee.TEST, 2)
|
|
|
|
|
2019-01-24 15:43:13 -04:00
|
|
|
def test_empty_globals(self):
|
|
|
|
# bpo-35717: sys._getframe(2).f_globals['__name__'] fails with KeyError
|
|
|
|
# when using compile and exec because f_globals is empty
|
|
|
|
code = "from enum import Enum; Enum('Animal', 'ANT BEE CAT DOG')"
|
|
|
|
code = compile(code, "<string>", "exec")
|
|
|
|
global_ns = {}
|
|
|
|
local_ls = {}
|
|
|
|
exec(code, global_ns, local_ls)
|
|
|
|
|
2020-09-21 21:23:13 -03:00
|
|
|
def test_strenum(self):
|
|
|
|
class GoodStrEnum(StrEnum):
|
|
|
|
one = '1'
|
|
|
|
two = '2'
|
|
|
|
three = b'3', 'ascii'
|
|
|
|
four = b'4', 'latin1', 'strict'
|
2020-09-22 17:00:07 -03:00
|
|
|
self.assertEqual(GoodStrEnum.one, '1')
|
|
|
|
self.assertEqual(str(GoodStrEnum.one), '1')
|
|
|
|
self.assertEqual(GoodStrEnum.one, str(GoodStrEnum.one))
|
|
|
|
self.assertEqual(GoodStrEnum.one, '{}'.format(GoodStrEnum.one))
|
|
|
|
#
|
|
|
|
class DumbMixin:
|
|
|
|
def __str__(self):
|
|
|
|
return "don't do this"
|
|
|
|
class DumbStrEnum(DumbMixin, StrEnum):
|
|
|
|
five = '5'
|
|
|
|
six = '6'
|
|
|
|
seven = '7'
|
|
|
|
self.assertEqual(DumbStrEnum.seven, '7')
|
|
|
|
self.assertEqual(str(DumbStrEnum.seven), "don't do this")
|
|
|
|
#
|
|
|
|
class EnumMixin(Enum):
|
|
|
|
def hello(self):
|
|
|
|
print('hello from %s' % (self, ))
|
|
|
|
class HelloEnum(EnumMixin, StrEnum):
|
|
|
|
eight = '8'
|
|
|
|
self.assertEqual(HelloEnum.eight, '8')
|
|
|
|
self.assertEqual(HelloEnum.eight, str(HelloEnum.eight))
|
|
|
|
#
|
|
|
|
class GoodbyeMixin:
|
|
|
|
def goodbye(self):
|
|
|
|
print('%s wishes you a fond farewell')
|
|
|
|
class GoodbyeEnum(GoodbyeMixin, EnumMixin, StrEnum):
|
|
|
|
nine = '9'
|
|
|
|
self.assertEqual(GoodbyeEnum.nine, '9')
|
|
|
|
self.assertEqual(GoodbyeEnum.nine, str(GoodbyeEnum.nine))
|
|
|
|
#
|
2020-09-21 21:23:13 -03:00
|
|
|
with self.assertRaisesRegex(TypeError, '1 is not a string'):
|
|
|
|
class FirstFailedStrEnum(StrEnum):
|
|
|
|
one = 1
|
|
|
|
two = '2'
|
|
|
|
with self.assertRaisesRegex(TypeError, "2 is not a string"):
|
|
|
|
class SecondFailedStrEnum(StrEnum):
|
|
|
|
one = '1'
|
|
|
|
two = 2,
|
|
|
|
three = '3'
|
|
|
|
with self.assertRaisesRegex(TypeError, '2 is not a string'):
|
|
|
|
class ThirdFailedStrEnum(StrEnum):
|
|
|
|
one = '1'
|
|
|
|
two = 2
|
|
|
|
with self.assertRaisesRegex(TypeError, 'encoding must be a string, not %r' % (sys.getdefaultencoding, )):
|
|
|
|
class ThirdFailedStrEnum(StrEnum):
|
|
|
|
one = '1'
|
|
|
|
two = b'2', sys.getdefaultencoding
|
|
|
|
with self.assertRaisesRegex(TypeError, 'errors must be a string, not 9'):
|
|
|
|
class ThirdFailedStrEnum(StrEnum):
|
|
|
|
one = '1'
|
|
|
|
two = b'2', 'ascii', 9
|
2021-01-13 03:47:57 -04:00
|
|
|
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
def test_missing_value_error(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, "_value_ not set in __new__"):
|
|
|
|
class Combined(str, Enum):
|
|
|
|
#
|
|
|
|
def __new__(cls, value, sequence):
|
|
|
|
enum = str.__new__(cls, value)
|
|
|
|
if '(' in value:
|
|
|
|
fis_name, segment = value.split('(', 1)
|
|
|
|
segment = segment.strip(' )')
|
|
|
|
else:
|
|
|
|
fis_name = value
|
|
|
|
segment = None
|
|
|
|
enum.fis_name = fis_name
|
|
|
|
enum.segment = segment
|
|
|
|
enum.sequence = sequence
|
|
|
|
return enum
|
|
|
|
#
|
|
|
|
def __repr__(self):
|
|
|
|
return "<%s.%s>" % (self.__class__.__name__, self._name_)
|
|
|
|
#
|
|
|
|
key_type = 'An$(1,2)', 0
|
|
|
|
company_id = 'An$(3,2)', 1
|
|
|
|
code = 'An$(5,1)', 2
|
|
|
|
description = 'Bn$', 3
|
2013-06-14 20:55:46 -03:00
|
|
|
|
2020-12-09 21:12:11 -04:00
|
|
|
@unittest.skipUnless(
|
|
|
|
sys.version_info[:2] == (3, 9),
|
|
|
|
'private variables are now normal attributes',
|
|
|
|
)
|
|
|
|
def test_warning_for_private_variables(self):
|
|
|
|
with self.assertWarns(DeprecationWarning):
|
|
|
|
class Private(Enum):
|
|
|
|
__corporal = 'Radar'
|
|
|
|
self.assertEqual(Private._Private__corporal.value, 'Radar')
|
|
|
|
try:
|
|
|
|
with self.assertWarns(DeprecationWarning):
|
|
|
|
class Private(Enum):
|
|
|
|
__major_ = 'Hoolihan'
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
def test_private_variable_is_normal_attribute(self):
|
|
|
|
class Private(Enum):
|
|
|
|
__corporal = 'Radar'
|
|
|
|
__major_ = 'Hoolihan'
|
|
|
|
self.assertEqual(Private._Private__corporal, 'Radar')
|
|
|
|
self.assertEqual(Private._Private__major_, 'Hoolihan')
|
|
|
|
|
2021-02-08 21:32:38 -04:00
|
|
|
@unittest.skipUnless(
|
2021-03-03 13:54:30 -04:00
|
|
|
sys.version_info[:2] < (3, 12),
|
2021-02-08 21:32:38 -04:00
|
|
|
'member-member access now raises an exception',
|
|
|
|
)
|
|
|
|
def test_warning_for_member_from_member_access(self):
|
|
|
|
with self.assertWarns(DeprecationWarning):
|
|
|
|
class Di(Enum):
|
|
|
|
YES = 1
|
|
|
|
NO = 0
|
|
|
|
nope = Di.YES.NO
|
|
|
|
self.assertIs(Di.NO, nope)
|
|
|
|
|
|
|
|
@unittest.skipUnless(
|
2021-03-03 13:54:30 -04:00
|
|
|
sys.version_info[:2] >= (3, 12),
|
2021-02-08 21:32:38 -04:00
|
|
|
'member-member access currently issues a warning',
|
|
|
|
)
|
|
|
|
def test_exception_for_member_from_member_access(self):
|
|
|
|
with self.assertRaisesRegex(AttributeError, "Di: no instance attribute .NO."):
|
|
|
|
class Di(Enum):
|
|
|
|
YES = 1
|
|
|
|
NO = 0
|
|
|
|
nope = Di.YES.NO
|
|
|
|
|
2020-12-10 16:20:06 -04:00
|
|
|
def test_strenum_auto(self):
|
|
|
|
class Strings(StrEnum):
|
|
|
|
ONE = auto()
|
|
|
|
TWO = auto()
|
|
|
|
self.assertEqual([Strings.ONE, Strings.TWO], ['one', 'two'])
|
|
|
|
|
2020-12-09 21:12:11 -04:00
|
|
|
|
2020-12-10 17:07:00 -04:00
|
|
|
def test_dynamic_members_with_static_methods(self):
|
|
|
|
#
|
|
|
|
foo_defines = {'FOO_CAT': 'aloof', 'BAR_DOG': 'friendly', 'FOO_HORSE': 'big'}
|
|
|
|
class Foo(Enum):
|
|
|
|
vars().update({
|
|
|
|
k: v
|
|
|
|
for k, v in foo_defines.items()
|
|
|
|
if k.startswith('FOO_')
|
|
|
|
})
|
|
|
|
def upper(self):
|
|
|
|
return self.value.upper()
|
|
|
|
self.assertEqual(list(Foo), [Foo.FOO_CAT, Foo.FOO_HORSE])
|
|
|
|
self.assertEqual(Foo.FOO_CAT.value, 'aloof')
|
|
|
|
self.assertEqual(Foo.FOO_HORSE.upper(), 'BIG')
|
|
|
|
#
|
|
|
|
with self.assertRaisesRegex(TypeError, "'FOO_CAT' already defined as: 'aloof'"):
|
|
|
|
class FooBar(Enum):
|
|
|
|
vars().update({
|
|
|
|
k: v
|
|
|
|
for k, v in foo_defines.items()
|
|
|
|
if k.startswith('FOO_')
|
|
|
|
},
|
|
|
|
**{'FOO_CAT': 'small'},
|
|
|
|
)
|
|
|
|
def upper(self):
|
|
|
|
return self.value.upper()
|
|
|
|
|
|
|
|
|
2016-08-20 11:19:31 -03:00
|
|
|
class TestOrder(unittest.TestCase):
|
|
|
|
|
|
|
|
def test_same_members(self):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
|
|
|
|
def test_same_members_with_aliases(self):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
verde = green
|
|
|
|
|
|
|
|
def test_same_members_wrong_order(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'member order does not match _order_'):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue'
|
|
|
|
red = 1
|
|
|
|
blue = 3
|
|
|
|
green = 2
|
|
|
|
|
|
|
|
def test_order_has_extra_members(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'member order does not match _order_'):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue purple'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
|
|
|
|
def test_order_has_extra_members_with_aliases(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'member order does not match _order_'):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue purple'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
verde = green
|
|
|
|
|
|
|
|
def test_enum_has_extra_members(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'member order does not match _order_'):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
purple = 4
|
|
|
|
|
|
|
|
def test_enum_has_extra_members_with_aliases(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'member order does not match _order_'):
|
|
|
|
class Color(Enum):
|
|
|
|
_order_ = 'red green blue'
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
purple = 4
|
|
|
|
verde = green
|
|
|
|
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class TestFlag(unittest.TestCase):
|
2016-08-31 04:12:15 -03:00
|
|
|
"""Tests of the Flags."""
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class Perm(Flag):
|
2016-08-31 04:12:15 -03:00
|
|
|
R, W, X = 4, 2, 1
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class Open(Flag):
|
2016-08-31 04:12:15 -03:00
|
|
|
RO = 0
|
|
|
|
WO = 1
|
|
|
|
RW = 2
|
|
|
|
AC = 3
|
|
|
|
CE = 1<<19
|
|
|
|
|
2018-09-10 15:21:04 -03:00
|
|
|
class Color(Flag):
|
|
|
|
BLACK = 0
|
|
|
|
RED = 1
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
ROJO = 1
|
2018-09-10 15:21:04 -03:00
|
|
|
GREEN = 2
|
|
|
|
BLUE = 4
|
|
|
|
PURPLE = RED|BLUE
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
WHITE = RED|GREEN|BLUE
|
|
|
|
BLANCO = RED|GREEN|BLUE
|
2018-09-10 15:21:04 -03:00
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_str(self):
|
|
|
|
Perm = self.Perm
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Perm.R), 'R')
|
|
|
|
self.assertEqual(str(Perm.W), 'W')
|
|
|
|
self.assertEqual(str(Perm.X), 'X')
|
|
|
|
self.assertEqual(str(Perm.R | Perm.W), 'R|W')
|
|
|
|
self.assertEqual(str(Perm.R | Perm.W | Perm.X), 'R|W|X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(Perm(0)), 'Perm(0)')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(~Perm.R), 'W|X')
|
|
|
|
self.assertEqual(str(~Perm.W), 'R|X')
|
|
|
|
self.assertEqual(str(~Perm.X), 'R|W')
|
|
|
|
self.assertEqual(str(~(Perm.R | Perm.W)), 'X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(~(Perm.R | Perm.W | Perm.X)), 'Perm(0)')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Perm(~0)), 'R|W|X')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
|
|
|
Open = self.Open
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Open.RO), 'RO')
|
|
|
|
self.assertEqual(str(Open.WO), 'WO')
|
|
|
|
self.assertEqual(str(Open.AC), 'AC')
|
|
|
|
self.assertEqual(str(Open.RO | Open.CE), 'CE')
|
|
|
|
self.assertEqual(str(Open.WO | Open.CE), 'WO|CE')
|
|
|
|
self.assertEqual(str(~Open.RO), 'WO|RW|CE')
|
|
|
|
self.assertEqual(str(~Open.WO), 'RW|CE')
|
|
|
|
self.assertEqual(str(~Open.AC), 'CE')
|
|
|
|
self.assertEqual(str(~(Open.RO | Open.CE)), 'AC')
|
|
|
|
self.assertEqual(str(~(Open.WO | Open.CE)), 'RW')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
|
|
|
def test_repr(self):
|
|
|
|
Perm = self.Perm
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Perm.R), 'Perm.R')
|
|
|
|
self.assertEqual(repr(Perm.W), 'Perm.W')
|
|
|
|
self.assertEqual(repr(Perm.X), 'Perm.X')
|
|
|
|
self.assertEqual(repr(Perm.R | Perm.W), 'Perm.R|Perm.W')
|
|
|
|
self.assertEqual(repr(Perm.R | Perm.W | Perm.X), 'Perm.R|Perm.W|Perm.X')
|
|
|
|
self.assertEqual(repr(Perm(0)), '0x0')
|
|
|
|
self.assertEqual(repr(~Perm.R), 'Perm.W|Perm.X')
|
|
|
|
self.assertEqual(repr(~Perm.W), 'Perm.R|Perm.X')
|
|
|
|
self.assertEqual(repr(~Perm.X), 'Perm.R|Perm.W')
|
|
|
|
self.assertEqual(repr(~(Perm.R | Perm.W)), 'Perm.X')
|
|
|
|
self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '0x0')
|
|
|
|
self.assertEqual(repr(Perm(~0)), 'Perm.R|Perm.W|Perm.X')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
|
|
|
Open = self.Open
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Open.RO), 'Open.RO')
|
|
|
|
self.assertEqual(repr(Open.WO), 'Open.WO')
|
|
|
|
self.assertEqual(repr(Open.AC), 'Open.AC')
|
|
|
|
self.assertEqual(repr(Open.RO | Open.CE), 'Open.CE')
|
|
|
|
self.assertEqual(repr(Open.WO | Open.CE), 'Open.WO|Open.CE')
|
|
|
|
self.assertEqual(repr(~Open.RO), 'Open.WO|Open.RW|Open.CE')
|
|
|
|
self.assertEqual(repr(~Open.WO), 'Open.RW|Open.CE')
|
|
|
|
self.assertEqual(repr(~Open.AC), 'Open.CE')
|
|
|
|
self.assertEqual(repr(~(Open.RO | Open.CE)), 'Open.AC')
|
|
|
|
self.assertEqual(repr(~(Open.WO | Open.CE)), 'Open.RW')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
2020-12-08 15:14:10 -04:00
|
|
|
def test_format(self):
|
|
|
|
Perm = self.Perm
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(format(Perm.R, ''), 'R')
|
|
|
|
self.assertEqual(format(Perm.R | Perm.X, ''), 'R|X')
|
2020-12-08 15:14:10 -04:00
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_or(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for i in Perm:
|
|
|
|
for j in Perm:
|
|
|
|
self.assertEqual((i | j), Perm(i.value | j.value))
|
|
|
|
self.assertEqual((i | j).value, i.value | j.value)
|
|
|
|
self.assertIs(type(i | j), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i | i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO | Open.CE, Open.CE)
|
|
|
|
|
|
|
|
def test_and(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
RW = Perm.R | Perm.W
|
|
|
|
RX = Perm.R | Perm.X
|
|
|
|
WX = Perm.W | Perm.X
|
|
|
|
RWX = Perm.R | Perm.W | Perm.X
|
|
|
|
values = list(Perm) + [RW, RX, WX, RWX, Perm(0)]
|
|
|
|
for i in values:
|
|
|
|
for j in values:
|
|
|
|
self.assertEqual((i & j).value, i.value & j.value)
|
|
|
|
self.assertIs(type(i & j), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i & i, i)
|
|
|
|
self.assertIs(i & RWX, i)
|
|
|
|
self.assertIs(RWX & i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO & Open.CE, Open.RO)
|
|
|
|
|
|
|
|
def test_xor(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for i in Perm:
|
|
|
|
for j in Perm:
|
|
|
|
self.assertEqual((i ^ j).value, i.value ^ j.value)
|
|
|
|
self.assertIs(type(i ^ j), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i ^ Perm(0), i)
|
|
|
|
self.assertIs(Perm(0) ^ i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO ^ Open.CE, Open.CE)
|
|
|
|
self.assertIs(Open.CE ^ Open.CE, Open.RO)
|
|
|
|
|
|
|
|
def test_invert(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
RW = Perm.R | Perm.W
|
|
|
|
RX = Perm.R | Perm.X
|
|
|
|
WX = Perm.W | Perm.X
|
|
|
|
RWX = Perm.R | Perm.W | Perm.X
|
|
|
|
values = list(Perm) + [RW, RX, WX, RWX, Perm(0)]
|
|
|
|
for i in values:
|
|
|
|
self.assertIs(type(~i), Perm)
|
|
|
|
self.assertEqual(~~i, i)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(~~i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.WO & ~Open.WO, Open.RO)
|
|
|
|
self.assertIs((Open.WO|Open.CE) & ~Open.WO, Open.CE)
|
|
|
|
|
2016-09-02 20:32:32 -03:00
|
|
|
def test_bool(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for f in Perm:
|
|
|
|
self.assertTrue(f)
|
|
|
|
Open = self.Open
|
|
|
|
for f in Open:
|
|
|
|
self.assertEqual(bool(f.value), bool(f))
|
|
|
|
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
def test_boundary(self):
|
|
|
|
self.assertIs(enum.Flag._boundary_, STRICT)
|
|
|
|
class Iron(Flag, boundary=STRICT):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Iron._boundary_, STRICT)
|
|
|
|
#
|
|
|
|
class Water(Flag, boundary=CONFORM):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Water._boundary_, CONFORM)
|
|
|
|
#
|
|
|
|
class Space(Flag, boundary=EJECT):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Space._boundary_, EJECT)
|
|
|
|
#
|
|
|
|
class Bizarre(Flag, boundary=KEEP):
|
|
|
|
b = 3
|
|
|
|
c = 4
|
|
|
|
d = 6
|
|
|
|
#
|
|
|
|
self.assertRaisesRegex(ValueError, 'invalid value: 7', Iron, 7)
|
|
|
|
self.assertIs(Water(7), Water.ONE|Water.TWO)
|
|
|
|
self.assertIs(Water(~9), Water.TWO)
|
|
|
|
self.assertEqual(Space(7), 7)
|
|
|
|
self.assertTrue(type(Space(7)) is int)
|
|
|
|
self.assertEqual(list(Bizarre), [Bizarre.c])
|
|
|
|
self.assertIs(Bizarre(3), Bizarre.b)
|
|
|
|
self.assertIs(Bizarre(6), Bizarre.d)
|
|
|
|
|
|
|
|
def test_iter(self):
|
|
|
|
Color = self.Color
|
|
|
|
Open = self.Open
|
|
|
|
self.assertEqual(list(Color), [Color.RED, Color.GREEN, Color.BLUE])
|
|
|
|
self.assertEqual(list(Open), [Open.WO, Open.RW, Open.CE])
|
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_programatic_function_string(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = Flag('Perm', 'R W X')
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_string_with_start(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = Flag('Perm', 'R W X', start=8)
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 8<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_string_list(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = Flag('Perm', ['R', 'W', 'X'])
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_iterable(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = Flag('Perm', (('R', 2), ('W', 8), ('X', 32)))
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<(2*i+1)
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_from_dict(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = Flag('Perm', OrderedDict((('R', 2), ('W', 8), ('X', 32))))
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<(2*i+1)
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
def test_pickle(self):
|
|
|
|
if isinstance(FlagStooges, Exception):
|
|
|
|
raise FlagStooges
|
|
|
|
test_pickle_dump_load(self.assertIs, FlagStooges.CURLY|FlagStooges.MOE)
|
|
|
|
test_pickle_dump_load(self.assertIs, FlagStooges)
|
|
|
|
|
2018-09-10 15:21:04 -03:00
|
|
|
def test_contains(self):
|
|
|
|
Open = self.Open
|
|
|
|
Color = self.Color
|
|
|
|
self.assertFalse(Color.BLACK in Open)
|
|
|
|
self.assertFalse(Open.RO in Color)
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
'BLACK' in Color
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
'RO' in Open
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
1 in Color
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
1 in Open
|
|
|
|
|
|
|
|
def test_member_contains(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = self.Perm
|
|
|
|
R, W, X = Perm
|
|
|
|
RW = R | W
|
|
|
|
RX = R | X
|
|
|
|
WX = W | X
|
|
|
|
RWX = R | W | X
|
|
|
|
self.assertTrue(R in RW)
|
|
|
|
self.assertTrue(R in RX)
|
|
|
|
self.assertTrue(R in RWX)
|
|
|
|
self.assertTrue(W in RW)
|
|
|
|
self.assertTrue(W in WX)
|
|
|
|
self.assertTrue(W in RWX)
|
|
|
|
self.assertTrue(X in RX)
|
|
|
|
self.assertTrue(X in WX)
|
|
|
|
self.assertTrue(X in RWX)
|
|
|
|
self.assertFalse(R in WX)
|
|
|
|
self.assertFalse(W in RX)
|
|
|
|
self.assertFalse(X in RW)
|
|
|
|
|
2020-09-16 17:01:00 -03:00
|
|
|
def test_member_iter(self):
|
|
|
|
Color = self.Color
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(list(Color.BLACK), [])
|
|
|
|
self.assertEqual(list(Color.PURPLE), [Color.RED, Color.BLUE])
|
2020-09-16 17:01:00 -03:00
|
|
|
self.assertEqual(list(Color.BLUE), [Color.BLUE])
|
|
|
|
self.assertEqual(list(Color.GREEN), [Color.GREEN])
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(list(Color.WHITE), [Color.RED, Color.GREEN, Color.BLUE])
|
|
|
|
self.assertEqual(list(Color.WHITE), [Color.RED, Color.GREEN, Color.BLUE])
|
|
|
|
|
|
|
|
def test_member_length(self):
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.BLACK), 0)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.GREEN), 1)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.PURPLE), 2)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.BLANCO), 3)
|
|
|
|
|
|
|
|
def test_number_reset_and_order_cleanup(self):
|
|
|
|
class Confused(Flag):
|
|
|
|
_order_ = 'ONE TWO FOUR DOS EIGHT SIXTEEN'
|
|
|
|
ONE = auto()
|
|
|
|
TWO = auto()
|
|
|
|
FOUR = auto()
|
|
|
|
DOS = 2
|
|
|
|
EIGHT = auto()
|
|
|
|
SIXTEEN = auto()
|
|
|
|
self.assertEqual(
|
|
|
|
list(Confused),
|
|
|
|
[Confused.ONE, Confused.TWO, Confused.FOUR, Confused.EIGHT, Confused.SIXTEEN])
|
|
|
|
self.assertIs(Confused.TWO, Confused.DOS)
|
|
|
|
self.assertEqual(Confused.DOS._value_, 2)
|
|
|
|
self.assertEqual(Confused.EIGHT._value_, 8)
|
|
|
|
self.assertEqual(Confused.SIXTEEN._value_, 16)
|
|
|
|
|
|
|
|
def test_aliases(self):
|
|
|
|
Color = self.Color
|
|
|
|
self.assertEqual(Color(1).name, 'RED')
|
|
|
|
self.assertEqual(Color['ROJO'].name, 'RED')
|
|
|
|
self.assertEqual(Color(7).name, 'WHITE')
|
|
|
|
self.assertEqual(Color['BLANCO'].name, 'WHITE')
|
|
|
|
self.assertIs(Color.BLANCO, Color.WHITE)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open['AC'], Open.AC)
|
2020-09-16 17:01:00 -03:00
|
|
|
|
2016-09-11 03:36:59 -03:00
|
|
|
def test_auto_number(self):
|
|
|
|
class Color(Flag):
|
|
|
|
red = auto()
|
|
|
|
blue = auto()
|
|
|
|
green = auto()
|
|
|
|
|
|
|
|
self.assertEqual(list(Color), [Color.red, Color.blue, Color.green])
|
|
|
|
self.assertEqual(Color.red.value, 1)
|
|
|
|
self.assertEqual(Color.blue.value, 2)
|
|
|
|
self.assertEqual(Color.green.value, 4)
|
|
|
|
|
|
|
|
def test_auto_number_garbage(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, 'Invalid Flag value: .not an int.'):
|
|
|
|
class Color(Flag):
|
|
|
|
red = 'not an int'
|
|
|
|
blue = auto()
|
|
|
|
|
2016-09-18 17:15:41 -03:00
|
|
|
def test_duplicate_auto(self):
|
|
|
|
class Dupes(Enum):
|
|
|
|
first = primero = auto()
|
|
|
|
second = auto()
|
|
|
|
third = auto()
|
|
|
|
self.assertEqual([Dupes.first, Dupes.second, Dupes.third], list(Dupes))
|
|
|
|
|
|
|
|
def test_bizarre(self):
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
with self.assertRaisesRegex(TypeError, "invalid Flag 'Bizarre' -- missing values: 1, 2"):
|
|
|
|
class Bizarre(Flag):
|
|
|
|
b = 3
|
|
|
|
c = 4
|
|
|
|
d = 6
|
2016-09-18 17:15:41 -03:00
|
|
|
|
2018-09-21 23:03:09 -03:00
|
|
|
def test_multiple_mixin(self):
|
|
|
|
class AllMixin:
|
|
|
|
@classproperty
|
|
|
|
def ALL(cls):
|
|
|
|
members = list(cls)
|
|
|
|
all_value = None
|
|
|
|
if members:
|
|
|
|
all_value = members[0]
|
|
|
|
for member in members[1:]:
|
|
|
|
all_value |= member
|
|
|
|
cls.ALL = all_value
|
|
|
|
return all_value
|
|
|
|
class StrMixin:
|
|
|
|
def __str__(self):
|
|
|
|
return self._name_.lower()
|
|
|
|
class Color(AllMixin, Flag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Color.BLUE), 'BLUE')
|
2018-09-21 23:03:09 -03:00
|
|
|
class Color(AllMixin, StrMixin, Flag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
class Color(StrMixin, AllMixin, Flag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
|
2020-05-27 19:10:27 -03:00
|
|
|
@threading_helper.reap_threads
|
2017-01-24 16:12:06 -04:00
|
|
|
def test_unique_composite(self):
|
|
|
|
# override __eq__ to be identity only
|
|
|
|
class TestFlag(Flag):
|
|
|
|
one = auto()
|
|
|
|
two = auto()
|
|
|
|
three = auto()
|
|
|
|
four = auto()
|
|
|
|
five = auto()
|
|
|
|
six = auto()
|
|
|
|
seven = auto()
|
|
|
|
eight = auto()
|
|
|
|
def __eq__(self, other):
|
|
|
|
return self is other
|
|
|
|
def __hash__(self):
|
|
|
|
return hash(self._value_)
|
|
|
|
# have multiple threads competing to complete the composite members
|
|
|
|
seen = set()
|
|
|
|
failed = False
|
|
|
|
def cycle_enum():
|
|
|
|
nonlocal failed
|
|
|
|
try:
|
|
|
|
for i in range(256):
|
|
|
|
seen.add(TestFlag(i))
|
|
|
|
except Exception:
|
|
|
|
failed = True
|
|
|
|
threads = [
|
|
|
|
threading.Thread(target=cycle_enum)
|
|
|
|
for _ in range(8)
|
|
|
|
]
|
2020-05-27 19:10:27 -03:00
|
|
|
with threading_helper.start_threads(threads):
|
2017-01-24 16:12:06 -04:00
|
|
|
pass
|
|
|
|
# check that only 248 members were created
|
|
|
|
self.assertFalse(
|
|
|
|
failed,
|
|
|
|
'at least one thread failed while creating composite members')
|
|
|
|
self.assertEqual(256, len(seen), 'too many composite members created')
|
|
|
|
|
2020-12-09 20:41:22 -04:00
|
|
|
def test_init_subclass(self):
|
|
|
|
class MyEnum(Flag):
|
|
|
|
def __init_subclass__(cls, **kwds):
|
|
|
|
super().__init_subclass__(**kwds)
|
|
|
|
self.assertFalse(cls.__dict__.get('_test', False))
|
|
|
|
cls._test1 = 'MyEnum'
|
|
|
|
#
|
|
|
|
class TheirEnum(MyEnum):
|
|
|
|
def __init_subclass__(cls, **kwds):
|
|
|
|
super(TheirEnum, cls).__init_subclass__(**kwds)
|
|
|
|
cls._test2 = 'TheirEnum'
|
|
|
|
class WhoseEnum(TheirEnum):
|
|
|
|
def __init_subclass__(cls, **kwds):
|
|
|
|
pass
|
|
|
|
class NoEnum(WhoseEnum):
|
|
|
|
ONE = 1
|
|
|
|
self.assertEqual(TheirEnum.__dict__['_test1'], 'MyEnum')
|
|
|
|
self.assertEqual(WhoseEnum.__dict__['_test1'], 'MyEnum')
|
|
|
|
self.assertEqual(WhoseEnum.__dict__['_test2'], 'TheirEnum')
|
|
|
|
self.assertFalse(NoEnum.__dict__.get('_test1', False))
|
|
|
|
self.assertFalse(NoEnum.__dict__.get('_test2', False))
|
|
|
|
#
|
|
|
|
class OurEnum(MyEnum):
|
|
|
|
def __init_subclass__(cls, **kwds):
|
|
|
|
cls._test2 = 'OurEnum'
|
|
|
|
class WhereEnum(OurEnum):
|
|
|
|
def __init_subclass__(cls, **kwds):
|
|
|
|
pass
|
|
|
|
class NeverEnum(WhereEnum):
|
|
|
|
ONE = 1
|
|
|
|
self.assertEqual(OurEnum.__dict__['_test1'], 'MyEnum')
|
|
|
|
self.assertFalse(WhereEnum.__dict__.get('_test1', False))
|
|
|
|
self.assertEqual(WhereEnum.__dict__['_test2'], 'OurEnum')
|
|
|
|
self.assertFalse(NeverEnum.__dict__.get('_test1', False))
|
|
|
|
self.assertFalse(NeverEnum.__dict__.get('_test2', False))
|
|
|
|
|
2016-09-11 03:36:59 -03:00
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class TestIntFlag(unittest.TestCase):
|
2016-08-31 04:12:15 -03:00
|
|
|
"""Tests of the IntFlags."""
|
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class Perm(IntFlag):
|
2016-08-31 04:12:15 -03:00
|
|
|
R = 1 << 2
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
W = 1 << 1
|
|
|
|
X = 1 << 0
|
2016-08-31 04:12:15 -03:00
|
|
|
|
2016-09-02 03:55:19 -03:00
|
|
|
class Open(IntFlag):
|
2016-08-31 04:12:15 -03:00
|
|
|
RO = 0
|
|
|
|
WO = 1
|
|
|
|
RW = 2
|
|
|
|
AC = 3
|
|
|
|
CE = 1<<19
|
|
|
|
|
2018-09-10 15:21:04 -03:00
|
|
|
class Color(IntFlag):
|
|
|
|
BLACK = 0
|
|
|
|
RED = 1
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
ROJO = 1
|
2018-09-10 15:21:04 -03:00
|
|
|
GREEN = 2
|
|
|
|
BLUE = 4
|
|
|
|
PURPLE = RED|BLUE
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
WHITE = RED|GREEN|BLUE
|
|
|
|
BLANCO = RED|GREEN|BLUE
|
|
|
|
|
|
|
|
class Skip(IntFlag):
|
|
|
|
FIRST = 1
|
|
|
|
SECOND = 2
|
|
|
|
EIGHTH = 8
|
2018-09-10 15:21:04 -03:00
|
|
|
|
2016-09-18 17:15:41 -03:00
|
|
|
def test_type(self):
|
|
|
|
Perm = self.Perm
|
2020-12-08 15:14:10 -04:00
|
|
|
self.assertTrue(Perm._member_type_ is int)
|
2016-09-18 17:15:41 -03:00
|
|
|
Open = self.Open
|
|
|
|
for f in Perm:
|
|
|
|
self.assertTrue(isinstance(f, Perm))
|
|
|
|
self.assertEqual(f, f.value)
|
|
|
|
self.assertTrue(isinstance(Perm.W | Perm.X, Perm))
|
|
|
|
self.assertEqual(Perm.W | Perm.X, 3)
|
|
|
|
for f in Open:
|
|
|
|
self.assertTrue(isinstance(f, Open))
|
|
|
|
self.assertEqual(f, f.value)
|
|
|
|
self.assertTrue(isinstance(Open.WO | Open.RW, Open))
|
|
|
|
self.assertEqual(Open.WO | Open.RW, 3)
|
|
|
|
|
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_str(self):
|
|
|
|
Perm = self.Perm
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Perm.R), 'R')
|
|
|
|
self.assertEqual(str(Perm.W), 'W')
|
|
|
|
self.assertEqual(str(Perm.X), 'X')
|
|
|
|
self.assertEqual(str(Perm.R | Perm.W), 'R|W')
|
|
|
|
self.assertEqual(str(Perm.R | Perm.W | Perm.X), 'R|W|X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(Perm.R | 8), '12')
|
|
|
|
self.assertEqual(str(Perm(0)), 'Perm(0)')
|
|
|
|
self.assertEqual(str(Perm(8)), '8')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(~Perm.R), 'W|X')
|
|
|
|
self.assertEqual(str(~Perm.W), 'R|X')
|
|
|
|
self.assertEqual(str(~Perm.X), 'R|W')
|
|
|
|
self.assertEqual(str(~(Perm.R | Perm.W)), 'X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(~(Perm.R | Perm.W | Perm.X)), 'Perm(0)')
|
|
|
|
self.assertEqual(str(~(Perm.R | 8)), '-13')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Perm(~0)), 'R|W|X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(Perm(~8)), '-9')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
|
|
|
Open = self.Open
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Open.RO), 'RO')
|
|
|
|
self.assertEqual(str(Open.WO), 'WO')
|
|
|
|
self.assertEqual(str(Open.AC), 'AC')
|
|
|
|
self.assertEqual(str(Open.RO | Open.CE), 'CE')
|
|
|
|
self.assertEqual(str(Open.WO | Open.CE), 'WO|CE')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(Open(4)), '4')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(~Open.RO), 'WO|RW|CE')
|
|
|
|
self.assertEqual(str(~Open.WO), 'RW|CE')
|
|
|
|
self.assertEqual(str(~Open.AC), 'CE')
|
|
|
|
self.assertEqual(str(~(Open.RO | Open.CE)), 'AC')
|
|
|
|
self.assertEqual(str(~(Open.WO | Open.CE)), 'RW')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(str(Open(~4)), '-5')
|
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_repr(self):
|
|
|
|
Perm = self.Perm
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Perm.R), 'Perm.R')
|
|
|
|
self.assertEqual(repr(Perm.W), 'Perm.W')
|
|
|
|
self.assertEqual(repr(Perm.X), 'Perm.X')
|
|
|
|
self.assertEqual(repr(Perm.R | Perm.W), 'Perm.R|Perm.W')
|
|
|
|
self.assertEqual(repr(Perm.R | Perm.W | Perm.X), 'Perm.R|Perm.W|Perm.X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(Perm.R | 8), '12')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Perm(0)), '0x0')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(Perm(8)), '8')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(~Perm.R), 'Perm.W|Perm.X')
|
|
|
|
self.assertEqual(repr(~Perm.W), 'Perm.R|Perm.X')
|
|
|
|
self.assertEqual(repr(~Perm.X), 'Perm.R|Perm.W')
|
|
|
|
self.assertEqual(repr(~(Perm.R | Perm.W)), 'Perm.X')
|
|
|
|
self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '0x0')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(~(Perm.R | 8)), '-13')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Perm(~0)), 'Perm.R|Perm.W|Perm.X')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(Perm(~8)), '-9')
|
2016-08-31 04:12:15 -03:00
|
|
|
|
|
|
|
Open = self.Open
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(Open.RO), 'Open.RO')
|
|
|
|
self.assertEqual(repr(Open.WO), 'Open.WO')
|
|
|
|
self.assertEqual(repr(Open.AC), 'Open.AC')
|
|
|
|
self.assertEqual(repr(Open.RO | Open.CE), 'Open.CE')
|
|
|
|
self.assertEqual(repr(Open.WO | Open.CE), 'Open.WO|Open.CE')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(Open(4)), '4')
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(repr(~Open.RO), 'Open.WO|Open.RW|Open.CE')
|
|
|
|
self.assertEqual(repr(~Open.WO), 'Open.RW|Open.CE')
|
|
|
|
self.assertEqual(repr(~Open.AC), 'Open.CE')
|
|
|
|
self.assertEqual(repr(~(Open.RO | Open.CE)), 'Open.AC')
|
|
|
|
self.assertEqual(repr(~(Open.WO | Open.CE)), 'Open.RW')
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(repr(Open(~4)), '-5')
|
|
|
|
|
2020-12-08 15:14:10 -04:00
|
|
|
def test_format(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
self.assertEqual(format(Perm.R, ''), '4')
|
|
|
|
self.assertEqual(format(Perm.R | Perm.X, ''), '5')
|
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_or(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for i in Perm:
|
|
|
|
for j in Perm:
|
|
|
|
self.assertEqual(i | j, i.value | j.value)
|
|
|
|
self.assertEqual((i | j).value, i.value | j.value)
|
|
|
|
self.assertIs(type(i | j), Perm)
|
|
|
|
for j in range(8):
|
|
|
|
self.assertEqual(i | j, i.value | j)
|
|
|
|
self.assertEqual((i | j).value, i.value | j)
|
|
|
|
self.assertIs(type(i | j), Perm)
|
|
|
|
self.assertEqual(j | i, j | i.value)
|
|
|
|
self.assertEqual((j | i).value, j | i.value)
|
|
|
|
self.assertIs(type(j | i), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i | i, i)
|
|
|
|
self.assertIs(i | 0, i)
|
|
|
|
self.assertIs(0 | i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO | Open.CE, Open.CE)
|
|
|
|
|
|
|
|
def test_and(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
RW = Perm.R | Perm.W
|
|
|
|
RX = Perm.R | Perm.X
|
|
|
|
WX = Perm.W | Perm.X
|
|
|
|
RWX = Perm.R | Perm.W | Perm.X
|
|
|
|
values = list(Perm) + [RW, RX, WX, RWX, Perm(0)]
|
|
|
|
for i in values:
|
|
|
|
for j in values:
|
|
|
|
self.assertEqual(i & j, i.value & j.value, 'i is %r, j is %r' % (i, j))
|
|
|
|
self.assertEqual((i & j).value, i.value & j.value, 'i is %r, j is %r' % (i, j))
|
|
|
|
self.assertIs(type(i & j), Perm, 'i is %r, j is %r' % (i, j))
|
|
|
|
for j in range(8):
|
|
|
|
self.assertEqual(i & j, i.value & j)
|
|
|
|
self.assertEqual((i & j).value, i.value & j)
|
|
|
|
self.assertIs(type(i & j), Perm)
|
|
|
|
self.assertEqual(j & i, j & i.value)
|
|
|
|
self.assertEqual((j & i).value, j & i.value)
|
|
|
|
self.assertIs(type(j & i), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i & i, i)
|
|
|
|
self.assertIs(i & 7, i)
|
|
|
|
self.assertIs(7 & i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO & Open.CE, Open.RO)
|
|
|
|
|
|
|
|
def test_xor(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for i in Perm:
|
|
|
|
for j in Perm:
|
|
|
|
self.assertEqual(i ^ j, i.value ^ j.value)
|
|
|
|
self.assertEqual((i ^ j).value, i.value ^ j.value)
|
|
|
|
self.assertIs(type(i ^ j), Perm)
|
|
|
|
for j in range(8):
|
|
|
|
self.assertEqual(i ^ j, i.value ^ j)
|
|
|
|
self.assertEqual((i ^ j).value, i.value ^ j)
|
|
|
|
self.assertIs(type(i ^ j), Perm)
|
|
|
|
self.assertEqual(j ^ i, j ^ i.value)
|
|
|
|
self.assertEqual((j ^ i).value, j ^ i.value)
|
|
|
|
self.assertIs(type(j ^ i), Perm)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(i ^ 0, i)
|
|
|
|
self.assertIs(0 ^ i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.RO ^ Open.CE, Open.CE)
|
|
|
|
self.assertIs(Open.CE ^ Open.CE, Open.RO)
|
|
|
|
|
|
|
|
def test_invert(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
RW = Perm.R | Perm.W
|
|
|
|
RX = Perm.R | Perm.X
|
|
|
|
WX = Perm.W | Perm.X
|
|
|
|
RWX = Perm.R | Perm.W | Perm.X
|
|
|
|
values = list(Perm) + [RW, RX, WX, RWX, Perm(0)]
|
|
|
|
for i in values:
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(~i, (~i).value)
|
2016-08-31 04:12:15 -03:00
|
|
|
self.assertIs(type(~i), Perm)
|
|
|
|
self.assertEqual(~~i, i)
|
|
|
|
for i in Perm:
|
|
|
|
self.assertIs(~~i, i)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open.WO & ~Open.WO, Open.RO)
|
|
|
|
self.assertIs((Open.WO|Open.CE) & ~Open.WO, Open.CE)
|
|
|
|
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
def test_boundary(self):
|
|
|
|
self.assertIs(enum.IntFlag._boundary_, EJECT)
|
|
|
|
class Iron(IntFlag, boundary=STRICT):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Iron._boundary_, STRICT)
|
|
|
|
#
|
|
|
|
class Water(IntFlag, boundary=CONFORM):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Water._boundary_, CONFORM)
|
|
|
|
#
|
|
|
|
class Space(IntFlag, boundary=EJECT):
|
|
|
|
ONE = 1
|
|
|
|
TWO = 2
|
|
|
|
EIGHT = 8
|
|
|
|
self.assertIs(Space._boundary_, EJECT)
|
|
|
|
#
|
|
|
|
class Bizarre(IntFlag, boundary=KEEP):
|
|
|
|
b = 3
|
|
|
|
c = 4
|
|
|
|
d = 6
|
|
|
|
#
|
|
|
|
self.assertRaisesRegex(ValueError, 'invalid value: 5', Iron, 5)
|
|
|
|
self.assertIs(Water(7), Water.ONE|Water.TWO)
|
|
|
|
self.assertIs(Water(~9), Water.TWO)
|
|
|
|
self.assertEqual(Space(7), 7)
|
|
|
|
self.assertTrue(type(Space(7)) is int)
|
|
|
|
self.assertEqual(list(Bizarre), [Bizarre.c])
|
|
|
|
self.assertIs(Bizarre(3), Bizarre.b)
|
|
|
|
self.assertIs(Bizarre(6), Bizarre.d)
|
|
|
|
|
|
|
|
def test_iter(self):
|
|
|
|
Color = self.Color
|
|
|
|
Open = self.Open
|
|
|
|
self.assertEqual(list(Color), [Color.RED, Color.GREEN, Color.BLUE])
|
|
|
|
self.assertEqual(list(Open), [Open.WO, Open.RW, Open.CE])
|
|
|
|
|
2016-08-31 04:12:15 -03:00
|
|
|
def test_programatic_function_string(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = IntFlag('Perm', 'R W X')
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e, v)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_string_with_start(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = IntFlag('Perm', 'R W X', start=8)
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 8<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e, v)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_string_list(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = IntFlag('Perm', ['R', 'W', 'X'])
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<i
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e, v)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_iterable(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = IntFlag('Perm', (('R', 2), ('W', 8), ('X', 32)))
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<(2*i+1)
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e, v)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
def test_programatic_function_from_dict(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = IntFlag('Perm', OrderedDict((('R', 2), ('W', 8), ('X', 32))))
|
2016-08-31 04:12:15 -03:00
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 3, Perm)
|
|
|
|
self.assertEqual(lst, [Perm.R, Perm.W, Perm.X])
|
|
|
|
for i, n in enumerate('R W X'.split()):
|
|
|
|
v = 1<<(2*i+1)
|
|
|
|
e = Perm(v)
|
|
|
|
self.assertEqual(e.value, v)
|
|
|
|
self.assertEqual(type(e.value), int)
|
|
|
|
self.assertEqual(e, v)
|
|
|
|
self.assertEqual(e.name, n)
|
|
|
|
self.assertIn(e, Perm)
|
|
|
|
self.assertIs(type(e), Perm)
|
|
|
|
|
|
|
|
|
2017-06-21 13:52:32 -03:00
|
|
|
def test_programatic_function_from_empty_list(self):
|
|
|
|
Perm = enum.IntFlag('Perm', [])
|
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 0, Perm)
|
|
|
|
Thing = enum.Enum('Thing', [])
|
|
|
|
lst = list(Thing)
|
|
|
|
self.assertEqual(len(lst), len(Thing))
|
|
|
|
self.assertEqual(len(Thing), 0, Thing)
|
|
|
|
|
|
|
|
|
|
|
|
def test_programatic_function_from_empty_tuple(self):
|
|
|
|
Perm = enum.IntFlag('Perm', ())
|
|
|
|
lst = list(Perm)
|
|
|
|
self.assertEqual(len(lst), len(Perm))
|
|
|
|
self.assertEqual(len(Perm), 0, Perm)
|
|
|
|
Thing = enum.Enum('Thing', ())
|
|
|
|
self.assertEqual(len(lst), len(Thing))
|
|
|
|
self.assertEqual(len(Thing), 0, Thing)
|
|
|
|
|
2018-09-10 15:21:04 -03:00
|
|
|
def test_contains(self):
|
|
|
|
Open = self.Open
|
|
|
|
Color = self.Color
|
|
|
|
self.assertTrue(Color.GREEN in Color)
|
|
|
|
self.assertTrue(Open.RW in Open)
|
|
|
|
self.assertFalse(Color.GREEN in Open)
|
|
|
|
self.assertFalse(Open.RW in Color)
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
'GREEN' in Color
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
'RW' in Open
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
2 in Color
|
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
2 in Open
|
|
|
|
|
|
|
|
def test_member_contains(self):
|
2016-09-02 03:55:19 -03:00
|
|
|
Perm = self.Perm
|
|
|
|
R, W, X = Perm
|
|
|
|
RW = R | W
|
|
|
|
RX = R | X
|
|
|
|
WX = W | X
|
|
|
|
RWX = R | W | X
|
|
|
|
self.assertTrue(R in RW)
|
|
|
|
self.assertTrue(R in RX)
|
|
|
|
self.assertTrue(R in RWX)
|
|
|
|
self.assertTrue(W in RW)
|
|
|
|
self.assertTrue(W in WX)
|
|
|
|
self.assertTrue(W in RWX)
|
|
|
|
self.assertTrue(X in RX)
|
|
|
|
self.assertTrue(X in WX)
|
|
|
|
self.assertTrue(X in RWX)
|
|
|
|
self.assertFalse(R in WX)
|
|
|
|
self.assertFalse(W in RX)
|
|
|
|
self.assertFalse(X in RW)
|
2018-09-10 15:21:04 -03:00
|
|
|
with self.assertRaises(TypeError):
|
|
|
|
self.assertFalse('test' in RW)
|
2016-09-02 03:55:19 -03:00
|
|
|
|
2020-09-16 17:01:00 -03:00
|
|
|
def test_member_iter(self):
|
|
|
|
Color = self.Color
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(list(Color.BLACK), [])
|
|
|
|
self.assertEqual(list(Color.PURPLE), [Color.RED, Color.BLUE])
|
2020-09-16 17:01:00 -03:00
|
|
|
self.assertEqual(list(Color.BLUE), [Color.BLUE])
|
|
|
|
self.assertEqual(list(Color.GREEN), [Color.GREEN])
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
self.assertEqual(list(Color.WHITE), [Color.RED, Color.GREEN, Color.BLUE])
|
|
|
|
|
|
|
|
def test_member_length(self):
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.BLACK), 0)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.GREEN), 1)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.PURPLE), 2)
|
|
|
|
self.assertEqual(self.Color.__len__(self.Color.BLANCO), 3)
|
|
|
|
|
|
|
|
def test_aliases(self):
|
|
|
|
Color = self.Color
|
|
|
|
self.assertEqual(Color(1).name, 'RED')
|
|
|
|
self.assertEqual(Color['ROJO'].name, 'RED')
|
|
|
|
self.assertEqual(Color(7).name, 'WHITE')
|
|
|
|
self.assertEqual(Color['BLANCO'].name, 'WHITE')
|
|
|
|
self.assertIs(Color.BLANCO, Color.WHITE)
|
|
|
|
Open = self.Open
|
|
|
|
self.assertIs(Open['AC'], Open.AC)
|
2020-09-16 17:01:00 -03:00
|
|
|
|
2016-09-02 20:32:32 -03:00
|
|
|
def test_bool(self):
|
|
|
|
Perm = self.Perm
|
|
|
|
for f in Perm:
|
|
|
|
self.assertTrue(f)
|
|
|
|
Open = self.Open
|
|
|
|
for f in Open:
|
|
|
|
self.assertEqual(bool(f.value), bool(f))
|
|
|
|
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
def test_bizarre(self):
|
|
|
|
with self.assertRaisesRegex(TypeError, "invalid Flag 'Bizarre' -- missing values: 1, 2"):
|
|
|
|
class Bizarre(IntFlag):
|
|
|
|
b = 3
|
|
|
|
c = 4
|
|
|
|
d = 6
|
|
|
|
|
2018-09-21 23:03:09 -03:00
|
|
|
def test_multiple_mixin(self):
|
|
|
|
class AllMixin:
|
|
|
|
@classproperty
|
|
|
|
def ALL(cls):
|
|
|
|
members = list(cls)
|
|
|
|
all_value = None
|
|
|
|
if members:
|
|
|
|
all_value = members[0]
|
|
|
|
for member in members[1:]:
|
|
|
|
all_value |= member
|
|
|
|
cls.ALL = all_value
|
|
|
|
return all_value
|
|
|
|
class StrMixin:
|
|
|
|
def __str__(self):
|
|
|
|
return self._name_.lower()
|
|
|
|
class Color(AllMixin, IntFlag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
2021-03-31 01:17:26 -03:00
|
|
|
self.assertEqual(str(Color.BLUE), 'BLUE')
|
2018-09-21 23:03:09 -03:00
|
|
|
class Color(AllMixin, StrMixin, IntFlag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
class Color(StrMixin, AllMixin, IntFlag):
|
|
|
|
RED = auto()
|
|
|
|
GREEN = auto()
|
|
|
|
BLUE = auto()
|
|
|
|
self.assertEqual(Color.RED.value, 1)
|
|
|
|
self.assertEqual(Color.GREEN.value, 2)
|
|
|
|
self.assertEqual(Color.BLUE.value, 4)
|
|
|
|
self.assertEqual(Color.ALL.value, 7)
|
|
|
|
self.assertEqual(str(Color.BLUE), 'blue')
|
|
|
|
|
2020-05-27 19:10:27 -03:00
|
|
|
@threading_helper.reap_threads
|
2017-01-24 16:12:06 -04:00
|
|
|
def test_unique_composite(self):
|
|
|
|
# override __eq__ to be identity only
|
|
|
|
class TestFlag(IntFlag):
|
|
|
|
one = auto()
|
|
|
|
two = auto()
|
|
|
|
three = auto()
|
|
|
|
four = auto()
|
|
|
|
five = auto()
|
|
|
|
six = auto()
|
|
|
|
seven = auto()
|
|
|
|
eight = auto()
|
|
|
|
def __eq__(self, other):
|
|
|
|
return self is other
|
|
|
|
def __hash__(self):
|
|
|
|
return hash(self._value_)
|
|
|
|
# have multiple threads competing to complete the composite members
|
|
|
|
seen = set()
|
|
|
|
failed = False
|
|
|
|
def cycle_enum():
|
|
|
|
nonlocal failed
|
|
|
|
try:
|
|
|
|
for i in range(256):
|
|
|
|
seen.add(TestFlag(i))
|
|
|
|
except Exception:
|
|
|
|
failed = True
|
|
|
|
threads = [
|
|
|
|
threading.Thread(target=cycle_enum)
|
|
|
|
for _ in range(8)
|
|
|
|
]
|
2020-05-27 19:10:27 -03:00
|
|
|
with threading_helper.start_threads(threads):
|
2017-01-24 16:12:06 -04:00
|
|
|
pass
|
|
|
|
# check that only 248 members were created
|
|
|
|
self.assertFalse(
|
|
|
|
failed,
|
|
|
|
'at least one thread failed while creating composite members')
|
|
|
|
self.assertEqual(256, len(seen), 'too many composite members created')
|
|
|
|
|
|
|
|
|
2019-03-03 18:09:11 -04:00
|
|
|
class TestEmptyAndNonLatinStrings(unittest.TestCase):
|
|
|
|
|
|
|
|
def test_empty_string(self):
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
empty_abc = Enum('empty_abc', ('', 'B', 'C'))
|
|
|
|
|
|
|
|
def test_non_latin_character_string(self):
|
|
|
|
greek_abc = Enum('greek_abc', ('\u03B1', 'B', 'C'))
|
|
|
|
item = getattr(greek_abc, '\u03B1')
|
|
|
|
self.assertEqual(item.value, 1)
|
|
|
|
|
|
|
|
def test_non_latin_number_string(self):
|
|
|
|
hebrew_123 = Enum('hebrew_123', ('\u05D0', '2', '3'))
|
|
|
|
item = getattr(hebrew_123, '\u05D0')
|
|
|
|
self.assertEqual(item.value, 1)
|
|
|
|
|
|
|
|
|
2013-07-18 21:05:39 -03:00
|
|
|
class TestUnique(unittest.TestCase):
|
|
|
|
|
|
|
|
def test_unique_clean(self):
|
|
|
|
@unique
|
|
|
|
class Clean(Enum):
|
|
|
|
one = 1
|
|
|
|
two = 'dos'
|
|
|
|
tres = 4.0
|
|
|
|
@unique
|
|
|
|
class Cleaner(IntEnum):
|
|
|
|
single = 1
|
|
|
|
double = 2
|
|
|
|
triple = 3
|
|
|
|
|
|
|
|
def test_unique_dirty(self):
|
|
|
|
with self.assertRaisesRegex(ValueError, 'tres.*one'):
|
|
|
|
@unique
|
|
|
|
class Dirty(Enum):
|
|
|
|
one = 1
|
|
|
|
two = 'dos'
|
|
|
|
tres = 1
|
|
|
|
with self.assertRaisesRegex(
|
|
|
|
ValueError,
|
|
|
|
'double.*single.*turkey.*triple',
|
|
|
|
):
|
|
|
|
@unique
|
|
|
|
class Dirtier(IntEnum):
|
|
|
|
single = 1
|
|
|
|
double = 1
|
|
|
|
triple = 3
|
|
|
|
turkey = 3
|
|
|
|
|
2016-05-01 14:03:53 -03:00
|
|
|
def test_unique_with_name(self):
|
|
|
|
@unique
|
|
|
|
class Silly(Enum):
|
|
|
|
one = 1
|
|
|
|
two = 'dos'
|
|
|
|
name = 3
|
|
|
|
@unique
|
|
|
|
class Sillier(IntEnum):
|
|
|
|
single = 1
|
|
|
|
name = 2
|
|
|
|
triple = 3
|
|
|
|
value = 4
|
|
|
|
|
2013-07-18 21:05:39 -03:00
|
|
|
|
2021-03-31 01:17:26 -03:00
|
|
|
class TestEnumTypeSubclassing(unittest.TestCase):
|
|
|
|
pass
|
2018-09-21 23:03:09 -03:00
|
|
|
|
2015-04-11 13:39:59 -03:00
|
|
|
expected_help_output_with_docs = """\
|
2013-10-22 00:45:55 -03:00
|
|
|
Help on class Color in module %s:
|
|
|
|
|
|
|
|
class Color(enum.Enum)
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
| Color(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)
|
2017-01-24 03:06:22 -04:00
|
|
|
|\x20\x20
|
2015-04-12 03:23:06 -03:00
|
|
|
| An enumeration.
|
2015-04-04 06:48:04 -03:00
|
|
|
|\x20\x20
|
2013-10-22 00:45:55 -03:00
|
|
|
| Method resolution order:
|
|
|
|
| Color
|
|
|
|
| enum.Enum
|
|
|
|
| builtins.object
|
|
|
|
|\x20\x20
|
|
|
|
| Data and other attributes defined here:
|
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| blue = Color.blue
|
2013-10-22 00:45:55 -03:00
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| green = Color.green
|
2013-10-22 00:45:55 -03:00
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| red = Color.red
|
2013-10-22 00:45:55 -03:00
|
|
|
|\x20\x20
|
|
|
|
| ----------------------------------------------------------------------
|
|
|
|
| Data descriptors inherited from enum.Enum:
|
|
|
|
|\x20\x20
|
|
|
|
| name
|
|
|
|
| The name of the Enum member.
|
|
|
|
|\x20\x20
|
|
|
|
| value
|
|
|
|
| The value of the Enum member.
|
|
|
|
|\x20\x20
|
|
|
|
| ----------------------------------------------------------------------
|
2021-03-31 01:17:26 -03:00
|
|
|
| Readonly properties inherited from enum.EnumType:
|
2013-10-22 00:45:55 -03:00
|
|
|
|\x20\x20
|
|
|
|
| __members__
|
|
|
|
| Returns a mapping of member name->value.
|
|
|
|
|\x20\x20\x20\x20\x20\x20
|
|
|
|
| This mapping lists all enum members, including aliases. Note that this
|
2015-04-11 13:39:59 -03:00
|
|
|
| is a read-only view of the internal mapping."""
|
|
|
|
|
|
|
|
expected_help_output_without_docs = """\
|
|
|
|
Help on class Color in module %s:
|
|
|
|
|
|
|
|
class Color(enum.Enum)
|
2017-01-24 03:06:22 -04:00
|
|
|
| Color(value, names=None, *, module=None, qualname=None, type=None, start=1)
|
|
|
|
|\x20\x20
|
2015-04-11 13:39:59 -03:00
|
|
|
| Method resolution order:
|
|
|
|
| Color
|
|
|
|
| enum.Enum
|
|
|
|
| builtins.object
|
|
|
|
|\x20\x20
|
|
|
|
| Data and other attributes defined here:
|
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| blue = Color.blue
|
2015-04-11 13:39:59 -03:00
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| green = Color.green
|
2015-04-11 13:39:59 -03:00
|
|
|
|\x20\x20
|
2021-03-31 01:17:26 -03:00
|
|
|
| red = Color.red
|
2015-04-11 13:39:59 -03:00
|
|
|
|\x20\x20
|
|
|
|
| ----------------------------------------------------------------------
|
|
|
|
| Data descriptors inherited from enum.Enum:
|
|
|
|
|\x20\x20
|
|
|
|
| name
|
|
|
|
|\x20\x20
|
|
|
|
| value
|
|
|
|
|\x20\x20
|
|
|
|
| ----------------------------------------------------------------------
|
2021-03-31 01:17:26 -03:00
|
|
|
| Data descriptors inherited from enum.EnumType:
|
2015-04-11 13:39:59 -03:00
|
|
|
|\x20\x20
|
|
|
|
| __members__"""
|
2013-10-22 00:45:55 -03:00
|
|
|
|
|
|
|
class TestStdLib(unittest.TestCase):
|
|
|
|
|
2015-04-12 03:23:06 -03:00
|
|
|
maxDiff = None
|
|
|
|
|
2013-10-22 00:45:55 -03:00
|
|
|
class Color(Enum):
|
|
|
|
red = 1
|
|
|
|
green = 2
|
|
|
|
blue = 3
|
|
|
|
|
|
|
|
def test_pydoc(self):
|
|
|
|
# indirectly test __objclass__
|
2015-04-11 13:39:59 -03:00
|
|
|
if StrEnum.__doc__ is None:
|
|
|
|
expected_text = expected_help_output_without_docs % __name__
|
|
|
|
else:
|
|
|
|
expected_text = expected_help_output_with_docs % __name__
|
2013-10-22 00:45:55 -03:00
|
|
|
output = StringIO()
|
|
|
|
helper = pydoc.Helper(output=output)
|
|
|
|
helper(self.Color)
|
|
|
|
result = output.getvalue().strip()
|
2014-06-16 17:48:43 -03:00
|
|
|
self.assertEqual(result, expected_text)
|
2013-10-22 00:45:55 -03:00
|
|
|
|
|
|
|
def test_inspect_getmembers(self):
|
|
|
|
values = dict((
|
2021-03-31 01:17:26 -03:00
|
|
|
('__class__', EnumType),
|
2015-04-12 03:23:06 -03:00
|
|
|
('__doc__', 'An enumeration.'),
|
2013-10-22 00:45:55 -03:00
|
|
|
('__members__', self.Color.__members__),
|
|
|
|
('__module__', __name__),
|
|
|
|
('blue', self.Color.blue),
|
|
|
|
('green', self.Color.green),
|
|
|
|
('name', Enum.__dict__['name']),
|
|
|
|
('red', self.Color.red),
|
|
|
|
('value', Enum.__dict__['value']),
|
|
|
|
))
|
|
|
|
result = dict(inspect.getmembers(self.Color))
|
2021-01-13 03:47:57 -04:00
|
|
|
self.assertEqual(set(values.keys()), set(result.keys()))
|
2013-10-22 00:45:55 -03:00
|
|
|
failed = False
|
|
|
|
for k in values.keys():
|
|
|
|
if result[k] != values[k]:
|
|
|
|
print()
|
|
|
|
print('\n%s\n key: %s\n result: %s\nexpected: %s\n%s\n' %
|
|
|
|
('=' * 75, k, result[k], values[k], '=' * 75), sep='')
|
|
|
|
failed = True
|
|
|
|
if failed:
|
|
|
|
self.fail("result does not equal expected, see print above")
|
|
|
|
|
|
|
|
def test_inspect_classify_class_attrs(self):
|
|
|
|
# indirectly test __objclass__
|
|
|
|
from inspect import Attribute
|
|
|
|
values = [
|
|
|
|
Attribute(name='__class__', kind='data',
|
2021-03-31 01:17:26 -03:00
|
|
|
defining_class=object, object=EnumType),
|
2013-10-22 00:45:55 -03:00
|
|
|
Attribute(name='__doc__', kind='data',
|
2015-04-12 03:23:06 -03:00
|
|
|
defining_class=self.Color, object='An enumeration.'),
|
2013-10-22 00:45:55 -03:00
|
|
|
Attribute(name='__members__', kind='property',
|
2021-03-31 01:17:26 -03:00
|
|
|
defining_class=EnumType, object=EnumType.__members__),
|
2013-10-22 00:45:55 -03:00
|
|
|
Attribute(name='__module__', kind='data',
|
|
|
|
defining_class=self.Color, object=__name__),
|
|
|
|
Attribute(name='blue', kind='data',
|
|
|
|
defining_class=self.Color, object=self.Color.blue),
|
|
|
|
Attribute(name='green', kind='data',
|
|
|
|
defining_class=self.Color, object=self.Color.green),
|
|
|
|
Attribute(name='red', kind='data',
|
|
|
|
defining_class=self.Color, object=self.Color.red),
|
|
|
|
Attribute(name='name', kind='data',
|
|
|
|
defining_class=Enum, object=Enum.__dict__['name']),
|
|
|
|
Attribute(name='value', kind='data',
|
|
|
|
defining_class=Enum, object=Enum.__dict__['value']),
|
|
|
|
]
|
|
|
|
values.sort(key=lambda item: item.name)
|
|
|
|
result = list(inspect.classify_class_attrs(self.Color))
|
|
|
|
result.sort(key=lambda item: item.name)
|
2021-01-13 03:47:57 -04:00
|
|
|
self.assertEqual(
|
|
|
|
len(values), len(result),
|
|
|
|
"%s != %s" % ([a.name for a in values], [a.name for a in result])
|
|
|
|
)
|
2013-10-22 00:45:55 -03:00
|
|
|
failed = False
|
|
|
|
for v, r in zip(values, result):
|
|
|
|
if r != v:
|
|
|
|
print('\n%s\n%s\n%s\n%s\n' % ('=' * 75, r, v, '=' * 75), sep='')
|
|
|
|
failed = True
|
|
|
|
if failed:
|
|
|
|
self.fail("result does not equal expected, see print above")
|
|
|
|
|
2015-11-14 08:46:42 -04:00
|
|
|
|
|
|
|
class MiscTestCase(unittest.TestCase):
|
|
|
|
def test__all__(self):
|
bpo-38250: [Enum] single-bit flags are canonical (GH-24215)
Flag members are now divided by one-bit verses multi-bit, with multi-bit being treated as aliases. Iterating over a flag only returns the contained single-bit flags.
Iterating, repr(), and str() show members in definition order.
When constructing combined-member flags, any extra integer values are either discarded (CONFORM), turned into ints (EJECT) or treated as errors (STRICT). Flag classes can specify which of those three behaviors is desired:
>>> class Test(Flag, boundary=CONFORM):
... ONE = 1
... TWO = 2
...
>>> Test(5)
<Test.ONE: 1>
Besides the three above behaviors, there is also KEEP, which should not be used unless necessary -- for example, _convert_ specifies KEEP as there are flag sets in the stdlib that are incomplete and/or inconsistent (e.g. ssl.Options). KEEP will, as the name suggests, keep all bits; however, iterating over a flag with extra bits will only return the canonical flags contained, not the extra bits.
Iteration is now in member definition order. If member definition order
matches increasing value order, then a more efficient method of flag
decomposition is used; otherwise, sort() is called on the results of
that method to get definition order.
``re`` module:
repr() has been modified to support as closely as possible its previous
output; the big difference is that inverted flags cannot be output as
before because the inversion operation now always returns the comparable
positive result; i.e.
re.A|re.I|re.M|re.S is ~(re.L|re.U|re.S|re.T|re.DEBUG)
in both of the above terms, the ``value`` is 282.
re's tests have been updated to reflect the modifications to repr().
2021-01-25 18:26:19 -04:00
|
|
|
support.check__all__(self, enum, not_exported={'bin'})
|
2015-11-14 08:46:42 -04:00
|
|
|
|
|
|
|
|
2016-06-03 16:14:52 -03:00
|
|
|
# These are unordered here on purpose to ensure that declaration order
|
|
|
|
# makes no difference.
|
|
|
|
CONVERT_TEST_NAME_D = 5
|
|
|
|
CONVERT_TEST_NAME_C = 5
|
|
|
|
CONVERT_TEST_NAME_B = 5
|
|
|
|
CONVERT_TEST_NAME_A = 5 # This one should sort first.
|
|
|
|
CONVERT_TEST_NAME_E = 5
|
|
|
|
CONVERT_TEST_NAME_F = 5
|
|
|
|
|
|
|
|
class TestIntEnumConvert(unittest.TestCase):
|
|
|
|
def test_convert_value_lookup_priority(self):
|
2018-09-12 14:28:53 -03:00
|
|
|
test_type = enum.IntEnum._convert_(
|
2017-01-24 16:12:06 -04:00
|
|
|
'UnittestConvert',
|
|
|
|
('test.test_enum', '__main__')[__name__=='__main__'],
|
2016-06-03 16:14:52 -03:00
|
|
|
filter=lambda x: x.startswith('CONVERT_TEST_'))
|
|
|
|
# We don't want the reverse lookup value to vary when there are
|
|
|
|
# multiple possible names for a given value. It should always
|
|
|
|
# report the first lexigraphical name in that case.
|
|
|
|
self.assertEqual(test_type(5).name, 'CONVERT_TEST_NAME_A')
|
|
|
|
|
|
|
|
def test_convert(self):
|
2018-09-12 14:28:53 -03:00
|
|
|
test_type = enum.IntEnum._convert_(
|
2017-01-24 16:12:06 -04:00
|
|
|
'UnittestConvert',
|
|
|
|
('test.test_enum', '__main__')[__name__=='__main__'],
|
2016-06-03 16:14:52 -03:00
|
|
|
filter=lambda x: x.startswith('CONVERT_TEST_'))
|
|
|
|
# Ensure that test_type has all of the desired names and values.
|
|
|
|
self.assertEqual(test_type.CONVERT_TEST_NAME_F,
|
|
|
|
test_type.CONVERT_TEST_NAME_A)
|
|
|
|
self.assertEqual(test_type.CONVERT_TEST_NAME_B, 5)
|
|
|
|
self.assertEqual(test_type.CONVERT_TEST_NAME_C, 5)
|
|
|
|
self.assertEqual(test_type.CONVERT_TEST_NAME_D, 5)
|
|
|
|
self.assertEqual(test_type.CONVERT_TEST_NAME_E, 5)
|
|
|
|
# Ensure that test_type only picked up names matching the filter.
|
|
|
|
self.assertEqual([name for name in dir(test_type)
|
|
|
|
if name[0:2] not in ('CO', '__')],
|
|
|
|
[], msg='Names other than CONVERT_TEST_* found.')
|
|
|
|
|
2018-09-12 14:28:53 -03:00
|
|
|
@unittest.skipUnless(sys.version_info[:2] == (3, 8),
|
|
|
|
'_convert was deprecated in 3.8')
|
|
|
|
def test_convert_warn(self):
|
|
|
|
with self.assertWarns(DeprecationWarning):
|
|
|
|
enum.IntEnum._convert(
|
|
|
|
'UnittestConvert',
|
|
|
|
('test.test_enum', '__main__')[__name__=='__main__'],
|
|
|
|
filter=lambda x: x.startswith('CONVERT_TEST_'))
|
|
|
|
|
|
|
|
@unittest.skipUnless(sys.version_info >= (3, 9),
|
|
|
|
'_convert was removed in 3.9')
|
|
|
|
def test_convert_raise(self):
|
|
|
|
with self.assertRaises(AttributeError):
|
|
|
|
enum.IntEnum._convert(
|
|
|
|
'UnittestConvert',
|
|
|
|
('test.test_enum', '__main__')[__name__=='__main__'],
|
|
|
|
filter=lambda x: x.startswith('CONVERT_TEST_'))
|
|
|
|
|
2021-03-31 01:17:26 -03:00
|
|
|
def test_convert_repr_and_str(self):
|
|
|
|
module = ('test.test_enum', '__main__')[__name__=='__main__']
|
|
|
|
test_type = enum.IntEnum._convert_(
|
|
|
|
'UnittestConvert',
|
|
|
|
module,
|
|
|
|
filter=lambda x: x.startswith('CONVERT_TEST_'))
|
|
|
|
self.assertEqual(repr(test_type.CONVERT_TEST_NAME_A), '%s.CONVERT_TEST_NAME_A' % module)
|
|
|
|
self.assertEqual(str(test_type.CONVERT_TEST_NAME_A), 'CONVERT_TEST_NAME_A')
|
|
|
|
self.assertEqual(format(test_type.CONVERT_TEST_NAME_A), '5')
|
|
|
|
|
|
|
|
# global names for StrEnum._convert_ test
|
|
|
|
CONVERT_STR_TEST_2 = 'goodbye'
|
|
|
|
CONVERT_STR_TEST_1 = 'hello'
|
|
|
|
|
|
|
|
class TestStrEnumConvert(unittest.TestCase):
|
|
|
|
|
|
|
|
def test_convert(self):
|
|
|
|
test_type = enum.StrEnum._convert_(
|
|
|
|
'UnittestConvert',
|
|
|
|
('test.test_enum', '__main__')[__name__=='__main__'],
|
|
|
|
filter=lambda x: x.startswith('CONVERT_STR_'))
|
|
|
|
# Ensure that test_type has all of the desired names and values.
|
|
|
|
self.assertEqual(test_type.CONVERT_STR_TEST_1, 'hello')
|
|
|
|
self.assertEqual(test_type.CONVERT_STR_TEST_2, 'goodbye')
|
|
|
|
# Ensure that test_type only picked up names matching the filter.
|
|
|
|
self.assertEqual([name for name in dir(test_type)
|
|
|
|
if name[0:2] not in ('CO', '__')],
|
|
|
|
[], msg='Names other than CONVERT_STR_* found.')
|
|
|
|
|
|
|
|
def test_convert_repr_and_str(self):
|
|
|
|
module = ('test.test_enum', '__main__')[__name__=='__main__']
|
|
|
|
test_type = enum.StrEnum._convert_(
|
|
|
|
'UnittestConvert',
|
|
|
|
module,
|
|
|
|
filter=lambda x: x.startswith('CONVERT_STR_'))
|
|
|
|
self.assertEqual(repr(test_type.CONVERT_STR_TEST_1), '%s.CONVERT_STR_TEST_1' % module)
|
|
|
|
self.assertEqual(str(test_type.CONVERT_STR_TEST_2), 'goodbye')
|
|
|
|
self.assertEqual(format(test_type.CONVERT_STR_TEST_1), 'hello')
|
|
|
|
|
2016-06-03 16:14:52 -03:00
|
|
|
|
2013-06-14 20:55:46 -03:00
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|