관리-도구
편집 파일: test.py
# -*- coding: utf-8 -*- from __future__ import division, print_function import sys import aenum import doctest import os import shutil import tempfile import textwrap import unittest import uuid import warnings from aenum import EnumType, EnumMeta, Enum, IntEnum, StrEnum, LowerStrEnum, UpperStrEnum, ReprEnum from aenum import AutoNumberEnum, MultiValueEnum, OrderedEnum, UniqueEnum, AddValueEnum, Flag, IntFlag from aenum import NamedTuple, TupleSize, NamedConstant, constant, NoAlias, AddValue, Unique from aenum import STRICT, CONFORM, EJECT, KEEP from aenum import _reduce_ex_by_name, unique, skip, extend_enum, auto, enum, MultiValue, member, nonmember, no_arg from aenum import basestring, baseinteger, unicode, enum_property from aenum import pyver, PY2, PY3, PY2_6, PY3_3, PY3_4, PY3_5, PY3_6, PY3_7, PY3_11 from aenum._enum import _high_bit from collections import OrderedDict from datetime import timedelta from pickle import dumps, loads, PicklingError, HIGHEST_PROTOCOL from operator import or_ as _or_, and_ as _and_, xor as _xor_, inv as _inv_ from operator import abs as _abs_, add as _add_, floordiv as _floordiv_ from operator import lshift as _lshift_, rshift as _rshift_, mod as _mod_ from operator import mul as _mul_, neg as _neg_, pos as _pos_, pow as _pow_ from operator import truediv as _truediv_, sub as _sub_ if PY2: from operator import div as _div_ try: import threading except ImportError: threading = None try: any except NameError: from aenum import any MODULE = __name__ SHORT_MODULE = MODULE.split('.')[-1] def load_tests(loader, tests, ignore): tests.addTests(doctest.DocTestSuite(aenum)) tests.addTests(doctest.DocFileSuite( 'doc/aenum.rst', package=aenum, optionflags=doctest.ELLIPSIS|doctest.NORMALIZE_WHITESPACE, )) return tests class TestCase(unittest.TestCase): def __init__(self, *args, **kwds): regex = getattr(self, 'assertRaisesRegex', None) if regex is None: self.assertRaisesRegex = getattr(self, 'assertRaisesRegexp') super(TestCase, self).__init__(*args, **kwds) # for pickle tests try: class Stooges(Enum): LARRY = 1 CURLY = 2 MOE = 3 except Exception: Stooges = sys.exc_info()[1] try: class IntStooges(int, Enum): LARRY = 1 CURLY = 2 MOE = 3 except Exception: IntStooges = sys.exc_info()[1] try: class FloatStooges(float, Enum): LARRY = 1.39 CURLY = 2.72 MOE = 3.142596 except Exception: FloatStooges = sys.exc_info()[1] try: class FlagStooges(Flag): LARRY = 1 CURLY = 2 MOE = 4 PMASK = 5 CMASK = 15 except Exception as exc: FlagStooges = exc try: class FlagStoogesWithZero(Flag): NOFLAG = 0 LARRY = 1 CURLY = 2 MOE = 4 PMASK = 5 CMASK = 15 except Exception as exc: FlagStoogesWithZero = exc try: class IntFlagStooges(IntFlag): LARRY = 1 CURLY = 2 MOE = 4 PMASK = 5 CMASK = 15 except Exception: IntFlagStooges = exc try: class IntFlagStoogesWithZero(IntFlag): NOFLAG = 0 LARRY = 1 CURLY = 2 MOE = 4 PMASK = 5 CMASK = 15 except Exception as exc: IntFlagStoogesWithZero = exc try: LifeForm = NamedTuple('LifeForm', 'branch genus species', module=__name__) except Exception: LifeForm = sys.exc_info()[1] try: class DeathForm(NamedTuple): color = 0 rigidity = 1 odor = 2 except Exception: DeathForm = sys.exc_info()[1] # for pickle test and subclass tests try: class Name(StrEnum): BDFL = 'Guido van Rossum' FLUFL = 'Barry Warsaw' except Exception: Name = sys.exc_info()[1] try: Question = Enum('Question', 'who what when where why', module=__name__) except Exception: Question = sys.exc_info()[1] try: Answer = Enum('Answer', 'him this then there because') except Exception: Answer = sys.exc_info()[1] try: class WhatsIt(NamedTuple): def what(self): return self[0] class ThatsIt(WhatsIt): blah = 0 bleh = 1 except Exception: ThatsIt = sys.exc_info()[1] # for doctests try: class Fruit(Enum): tomato = 1 banana = 2 cherry = 3 except Exception: pass def test_pickle_dump_load(assertion, source, target=None, protocol=(0, HIGHEST_PROTOCOL)): start, stop = protocol failures = [] for protocol in range(start, stop+1): try: if target is None: assertion(loads(dumps(source, protocol=protocol)), source) else: assertion(loads(dumps(source, protocol=protocol)), target) except Exception: exc, tb = sys.exc_info()[1:] failures.append('%2d: %s' %(protocol, exc)) if failures: raise ValueError('Failed with protocols: %s' % ', '.join(failures)) def test_pickle_exception(assertion, exception, obj, protocol=(0, HIGHEST_PROTOCOL)): start, stop = protocol failures = [] for protocol in range(start, stop+1): try: assertion(exception, dumps, obj, protocol=protocol) except Exception: exc = sys.exc_info()[1] failures.append('%d: %s %s' % (protocol, exc.__class__.__name__, exc)) if failures: raise ValueError('Failed with protocols: %s' % ', '.join(failures)) if PY3: from .test_v3 import TestEnumV3, TestOrderV3, TestNamedTupleV3, TestStackoverflowAnswersV3, TestIssuesV3, TestExtendEnumV3 from . import test_v3 test_v3.IntStooges = IntStooges test_v3.test_pickle_exception = test_pickle_exception test_v3.test_pickle_dump_load = test_pickle_dump_load if pyver >= PY3_7: from .test_v37 import TestEnumV37 # for subclassing tests class classproperty(object): 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) # tests class TestOrder(TestCase): """ Test _order_ extra/missing members. """ 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_order_has_extra_members(self): with self.assertRaisesRegex(TypeError, r'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, r'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, r'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, r'member order does not match _order_'): class Color(Enum): _order_ = 'red green blue' red = 1 green = 2 blue = 3 purple = 4 verde = green def test_same_members_flag(self): class Color(Flag): _order_ = 'red green blue' red = 1 green = 2 blue = 4 def test_same_members_with_aliases_flag(self): class Color(Flag): _order_ = 'red green blue' red = 1 green = 2 blue = 4 verde = green def test_order_has_extra_members_flag(self): with self.assertRaisesRegex(TypeError, r'member order does not match _order_'): class Color(Flag): _order_ = 'red green blue purple' red = 1 green = 2 blue = 4 def test_order_has_extra_members_with_aliases_flag(self): with self.assertRaisesRegex(TypeError, r'member order does not match _order_'): class Color(Flag): _order_ = 'red green blue purple' red = 1 green = 2 blue = 4 verde = green def test_enum_has_extra_members_flag(self): with self.assertRaisesRegex(TypeError, r'member order does not match _order_'): class Color(Flag): _order_ = 'red green blue' red = 1 green = 2 blue = 4 purple = 8 def test_enum_has_extra_members_with_aliases_flag(self): with self.assertRaisesRegex(TypeError, r'member order does not match _order_'): class Color(Flag): _order_ = 'red green blue' red = 1 green = 2 blue = 4 purple = 8 verde = green class TestAutoValue(TestCase): def test_bare(self): # class BareEnum(Enum): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(BareEnum.THREE.value, 3) # class BareIntEnum(IntEnum): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(BareIntEnum.THREE, 3) # class BareFlag(Flag): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(BareFlag.THREE.value, 4) # class BareIntFlag(IntFlag): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(BareIntFlag.THREE, 4) def test_init_only_final(self): # class InitEnumValue(Enum): _init_ = 'value description' _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitEnumValue.THREE.value, 3) self.assertEqual(InitEnumValue.THREE.description, 'a triangle') # class InitEnum(Enum): _init_ = 'value description' _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitEnum.THREE.value, 3) self.assertEqual(InitEnum.THREE.description, 'a triangle') # class InitIntEnum(IntEnum): _init_ = 'value description' _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitIntEnum.THREE, 3) self.assertEqual(InitIntEnum.THREE.description, 'a triangle') # class InitFlag(Flag): _init_ = 'value description' _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitFlag.THREE.value, 4) self.assertEqual(InitFlag.THREE.description, 'a triangle') # class InitIntFlag(IntFlag): _init_ = 'value description' _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitIntFlag.THREE, 4) self.assertEqual(InitIntFlag.THREE.description, 'a triangle') def test_init_only_inherit(self): # class InitInheritEnum(Enum): _init_ = 'value description' # class InitEnum(InitInheritEnum): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitEnum.THREE.value, 3) self.assertEqual(InitEnum.THREE.description, 'a triangle') # # class InitInheritValueEnum(Enum): _init_ = 'value description' # class InitEnum(InitInheritValueEnum): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitEnum.THREE.value, 3) self.assertEqual(InitEnum.THREE.description, 'a triangle') # class InitIntEnum(int, InitInheritValueEnum): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitIntEnum.THREE, 3) self.assertEqual(InitIntEnum.THREE.description, 'a triangle') # class InitInheritValueFlag(Flag): _init_ = 'value description' # class InitFlag(InitInheritValueFlag): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitFlag.THREE.value, 4) self.assertEqual(InitFlag.THREE.description, 'a triangle') # class InitIntFlag(int, InitInheritValueFlag): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitIntFlag.THREE, 4) self.assertEqual(InitIntFlag.THREE.description, 'a triangle') def test_new_only_final(self): # class NewFinalEnum(Enum): _order_ = 'ONE TWO THREE' def __new__(cls, value): member = object.__new__(cls) member._value_ = value member.proof = 'NFE1' return member ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalEnum.THREE.value, 3) self.assertEqual(NewFinalEnum.TWO.proof, 'NFE1') # class NewFinalIntEnum(IntEnum): _order_ = 'ONE TWO THREE' def __new__(cls, value): member = int.__new__(cls, value) member._value_ = value member.proof = 'NFE2' return member ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalIntEnum.THREE, 3) self.assertEqual(NewFinalIntEnum.TWO.proof, 'NFE2') # class NewFinalFlag(Flag): _order_ = 'ONE TWO THREE' def __new__(cls, value): member = object.__new__(cls) member._value_ = value member.proof = 'NFE3' return member ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalFlag.THREE.value, 4) self.assertEqual(NewFinalFlag.TWO.proof, 'NFE3') # class NewFinalIntFlag(IntFlag): _order_ = 'ONE TWO THREE' def __new__(cls, value): member = int.__new__(cls, value) member._value_ = value member.proof = 'NFE4' return member ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalIntFlag.THREE, 4) self.assertEqual(NewFinalIntFlag.TWO.proof, 'NFE4') # class NewFinalStrEnum(str, Enum): # _order_ = "AllReset Bright FG_Cyan BG_Black" # def __new__(cls, value, code, description): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code obj.description = description return obj # __str__ = str.__str__ # AllReset = '0', 'reset all (colors and brightness)' Bright = '1', 'bright lights!' FG_Cyan = '36', 'cyan' BG_Black = '40', 'black' self.assertEqual(NewFinalStrEnum.FG_Cyan.value, 3) self.assertEqual(NewFinalStrEnum.BG_Black.value, 4) self.assertEqual(NewFinalStrEnum.AllReset.code, '0') self.assertEqual(NewFinalStrEnum.Bright.description, 'bright lights!') # class NewFinalStrFlag(str, Flag): # _order_ = "AllReset Bright FG_Cyan BG_Black" # def __new__(cls, value, code, description): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code obj.description = description return obj # __str__ = str.__str__ # AllReset = '0', 'reset all (colors and brightness)' Bright = '1', 'bright lights!' FG_Cyan = '36', 'cyan' BG_Black = '40', 'black' self.assertEqual(NewFinalStrFlag.FG_Cyan.value, 4) self.assertEqual(NewFinalStrFlag.BG_Black.value, 8) self.assertEqual(NewFinalStrFlag.AllReset.code, '0') self.assertEqual(NewFinalStrFlag.Bright.description, 'bright lights!') def test_new_only_inherited(self): # class NewInheritEnum(Enum): def __new__(cls, value): if cls._member_type_ is int: member = int.__new__(cls, value*2) else: member = object.__new__(cls) member._value_ = value * 2 member.proof = 'NIE' return member # class NewFinalEnum(NewInheritEnum): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalEnum.THREE.value, 6) self.assertEqual(NewFinalEnum.TWO.proof, 'NIE') # class NewFinalIntEnum(int, NewInheritEnum): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalIntEnum.THREE, 6) self.assertEqual(NewFinalIntEnum.TWO.proof, 'NIE') # class NewInheritFlag(Flag): def __new__(cls, value): if cls._member_type_ is int: member = int.__new__(cls, value*2) else: member = object.__new__(cls) member._value_ = value * 2 member.proof = 'NIE' return member # class NewFinalFlag(NewInheritFlag): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalFlag.THREE.value, 8) self.assertEqual(NewFinalFlag.TWO.proof, 'NIE') # class NewFinalIntFlag(int, NewInheritFlag): _order_ = 'ONE TWO THREE' ONE = auto() TWO = auto() THREE = auto() self.assertEqual(NewFinalIntFlag.THREE, 8) self.assertEqual(NewFinalIntFlag.TWO.proof, 'NIE') def test_init_new_only(self): # class InitNewEnum(Enum): _init_ = "value description" _order_ = 'ONE TWO THREE' def __new__(cls, value, *args): member = object.__new__(cls) member._value_ = value member.proof = 'INE1' return member ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewEnum.THREE.value, 3) self.assertEqual(InitNewEnum.THREE.description, 'a triangle') self.assertEqual(InitNewEnum.TWO.proof, 'INE1') # class InitNewIntEnum(IntEnum): _init_ = "value description" _order_ = 'ONE TWO THREE' def __new__(cls, value, *args): member = int.__new__(cls, value) member._value_ = value member.proof = 'INE2' return member ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewIntEnum.THREE, 3) self.assertEqual(InitNewIntEnum.THREE.description, 'a triangle') self.assertEqual(InitNewIntEnum.TWO.proof, 'INE2') # class InitNewFlag(Flag): _init_ = "value description" _order_ = 'ONE TWO THREE' def __new__(cls, value, *args): member = object.__new__(cls) member._value_ = value member.proof = 'INE3' return member ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewFlag.THREE.value, 4) self.assertEqual(InitNewFlag.THREE.description, 'a triangle') self.assertEqual(InitNewFlag.TWO.proof, 'INE3') # class InitNewIntFlag(IntFlag): _init_ = "value description" _order_ = 'ONE TWO THREE' def __new__(cls, value, *args): member = int.__new__(cls, value) member._value_ = value member.proof = 'INE4' return member ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewIntFlag.THREE, 4) self.assertEqual(InitNewIntFlag.THREE.description, 'a triangle') self.assertEqual(InitNewIntFlag.TWO.proof, 'INE4') def test_init_new_inherit(self): # class InitNew(Enum): _init_ = "value description" def __new__(cls, value, *args): member = object.__new__(cls) member._value_ = value member.proof = 'IN' return member # class InitNewEnum(InitNew): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewEnum.THREE.value, 3) self.assertEqual(InitNewEnum.THREE.description, 'a triangle') self.assertEqual(InitNewEnum.TWO.proof, 'IN') # class InitNewInt(Enum): _init_ = "value description" def __new__(cls, value, *args): member = int.__new__(cls, value) member._value_ = value member.proof = 'IN' return member # class InitNewIntEnum(int, InitNewInt): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewIntEnum.THREE, 3) self.assertEqual(InitNewIntEnum.THREE.description, 'a triangle') self.assertEqual(InitNewIntEnum.TWO.proof, 'IN') # class InitNewFlagBase(Flag): _init_ = "value description" def __new__(cls, value, *args): member = object.__new__(cls) member._value_ = value member.proof = 'IN' return member # class InitNewFlag(InitNewFlagBase): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewFlag.THREE.value, 4) self.assertEqual(InitNewFlag.THREE.description, 'a triangle') self.assertEqual(InitNewFlag.TWO.proof, 'IN') # class InitNewIntFlagBase(int, Flag): _init_ = "value description" def __new__(cls, value, *args): member = int.__new__(cls, value) member._value_ = value member.proof = 'IN' return member # class InitNewIntFlag(InitNewIntFlagBase): _order_ = 'ONE TWO THREE' ONE = 'the loneliest number' TWO = 'the number with you' THREE = 'a triangle' self.assertEqual(InitNewIntFlag.THREE, 4) self.assertEqual(InitNewIntFlag.THREE.description, 'a triangle') self.assertEqual(InitNewIntFlag.TWO.proof, 'IN') class TestHelpers(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(aenum.is_descriptor(obj)) setattr(obj, attr, 1) self.assertTrue(aenum.is_descriptor(obj)) def test_is_sunder(self): for s in ('_a_', '_aa_'): self.assertTrue(aenum.is_sunder(s)) for s in ('a', 'a_', '_a', '__a', 'a__', '__a__', '_a__', '__a_', '_', '__', '___', '____', '_____',): self.assertFalse(aenum.is_sunder(s)) def test_is_dunder(self): for s in ('__a__', '__aa__'): self.assertTrue(aenum.is_dunder(s)) for s in ('a', 'a_', '_a', '__a', 'a__', '_a_', '_a__', '__a_', '_', '__', '___', '____', '_____',): self.assertFalse(aenum.is_dunder(s)) def test_auto(self): def tester(first, op, final, second=None): if second is None: left = auto() value = op(left) left.value = first self.assertEqual(value.value, final, "%s %r -> %r != %r" % (op.__name__, first, value, final)) else: left = first right = auto() value = op(left, right) right.value = second self.assertEqual(value.value, final, "forward: %r %s %r -> %r != %r" % (first, op.__name__, second, value.value, final)) left = auto() right = second value = op(left, right) left.value = first self.assertEqual(value.value, final, "reversed: %r %s %r -> %r != %r" % (second, op.__name__, first, value.value, final)) for args in ( (1, _abs_, abs(1)), (-3, _abs_, abs(-3)), (1, _add_, 1+2, 2), (25, _floordiv_, 25 // 5, 5), (49, _truediv_, 49 / 9, 9), (6, _mod_, 6 % 9, 9), (5, _lshift_, 5 << 2, 2), (5, _rshift_, 5 >> 2, 2), (3, _mul_, 3 * 6, 6), (5, _neg_, -5), (-4, _pos_, +(-4)), (2, _pow_, 2**5, 5), (7, _sub_, 7 - 10, 10), (1, _or_, 1 | 2, 2), (3, _xor_, 3 ^ 6, 6), (3, _and_, 3 & 6, 6), (7, _inv_, ~7), ('a', _add_, 'a'+'b', 'b'), ('a', _mul_, 'a' * 3, 3), ): tester(*args) # operator.div is gone in 3 if PY2: tester(12, _div_, 12 // 5, 5) # strings are a pain left = auto() right = 'eggs' value = _mod_(left, right) left.value = 'I see 17 %s!' self.assertEqual(value.value, 'I see 17 %s!' % 'eggs') def test_constant(self): errors = [] def tester(first, op, final, second=None): if second is None: primary = constant(first) secondary = constant(op(primary)) if secondary.value != final: errors.append( "%s %r -> %r != %r" % (op.__name__, first, secondary.value, final), ) else: left = constant(first) right = second value = op(left, right) if value != final: errors.append( "forward: %r %s %r -> %r != %r" % (first, op.__name__, second, value, final), ) left = first right = constant(second) value = op(left, right) if value != final: errors.append( "reversed: %r %s %r -> %r != %r" % (second, op.__name__, first, value, final), ) for args in ( (1, _abs_, abs(1)), (-3, _abs_, abs(-3)), (1, _add_, 1+2, 2), (25, _floordiv_, 25 // 5, 5), (49, _truediv_, 49 / 9, 9), (6, _mod_, 6 % 9, 9), (5, _lshift_, 5 << 2, 2), (5, _rshift_, 5 >> 2, 2), (3, _mul_, 3 * 6, 6), (5, _neg_, -5), (-4, _pos_, +(-4)), (2, _pow_, 2**5, 5), (7, _sub_, 7 - 10, 10), (1, _or_, 1 | 2, 2), (3, _xor_, 3 ^ 6, 6), (3, _and_, 3 & 6, 6), (7, _inv_, ~7), ('a', _add_, 'a'+'b', 'b'), ('a', _mul_, 'a' * 3, 3), ): tester(*args) # operator.div is gone in 3 if PY2: tester(12, _div_, 12 // 5, 5) # strings are a pain left = constant('I see 17 %s!') right = 'eggs' value = _mod_(left, right) if value != 'I see 17 %s!' % 'eggs': errors.append("'I see 17 eggs!' != %r" % value) if errors: print() for error in errors: print(error) self.assertTrue(False) class TestEnumType(TestCase): def test_immutability(self): class Hah(object): @classproperty def all_values(cls): return [m.value for m in cls] class Huh(Hah, Enum): one = 1 two = 2 self.assertRaisesRegex(AttributeError, 'cannot rebind property', setattr, Huh, 'value', 'boom') self.assertRaisesRegex(AttributeError, 'cannot delete property', delattr, Huh, 'value') self.assertRaisesRegex(AttributeError, 'cannot set attribute', setattr, Huh.one, 'value', 'boom') self.assertRaisesRegex(AttributeError, 'cannot delete attribute', delattr, Huh.two, 'value') self.assertEqual(Huh.one.value, 1) self.assertEqual(Huh.two.value, 2) self.assertEqual(Huh.all_values, [1, 2]) setattr(Huh, 'all_values', 99) self.assertEqual(Huh.all_values, 99) def test_enum_shadow_base(self): class hohum(object): def cyan(self): "cyanize a color" return self.value * 'cyan' @property def azure(self): return 'azure ' + self.name class Color(hohum, Enum): red = 1 green = 2 blue = 3 cyan = 4 azure = 5 self.assertEqual(len(Color), 5) self.assertEqual(list(Color), [Color.red, Color.green, Color.blue, Color.cyan, Color.azure]) self.assertRaisesRegex(AttributeError, 'no attribute .cyan.', lambda: Color.blue.cyan) self.assertEqual(Color.red.azure, 'azure red') class TestEnum(TestCase): def setUp(self): class Season(Enum): SPRING = 1 SUMMER = 2 AUTUMN = 3 WINTER = 4 self.Season = Season class ReprKonstants(float, ReprEnum): E = 2.7182818 PI = 3.1415926 TAU = 2 * PI self.ReprKonstants = ReprKonstants class ReprGrades(IntEnum): A = 5 B = 4 C = 3 D = 2 F = 0 self.ReprGrades = ReprGrades class ReprDirectional(str, ReprEnum): EAST = 'east' WEST = 'west' NORTH = 'north' SOUTH = 'south' self.ReprDirectional = ReprDirectional from datetime import date class ReprHoliday(date, ReprEnum): NEW_YEAR = 2013, 1, 1 IDES_OF_MARCH = 2013, 3, 15 self.ReprHoliday = ReprHoliday class Konstants(float, Enum): E = 2.7182818 PI = 3.1415926 TAU = 2 * PI self.Konstants = Konstants class Grades(int, Enum): 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 def test_set_name(self): class Descriptor(object): name = None def __get__(self, instance, owner_class=None): if instance is None: return self else: return instance.__dict__[self.name] def __set__(self, instance, value): instance.__dict__[self.name] = value def __set_name__(self, owner, name): self.name = name # class AnEnum(Enum): ONE = 'one' two = Descriptor() # self.assertEqual(list(AnEnum), [AnEnum.ONE]) self.assertEqual(AnEnum.two.name, 'two') AnEnum.ONE.two = 'three' self.assertEqual(AnEnum.ONE.two, 'three') self.assertEqual(AnEnum.ONE.__dict__['two'], 'three') def test_private_names(self): class Private(Enum): __corporal = 'Radar' __major_ = 'Hoolihan' self.assertEqual(len(Private), 0) self.assertEqual(Private._Private__corporal, 'Radar') self.assertFalse(isinstance(Private._Private__corporal, Enum)) self.assertEqual(Private._Private__major_, 'Hoolihan') self.assertFalse(isinstance(Private._Private__major_, Enum)) def test_new_with_keywords(self): class Huh(IntEnum): __order__ = 'PLAIN BOLD_ITALIC HIGHLIGHT' def __new__(cls, docstring, open=None, close=None): value = len(cls.__members__) member = int.__new__(cls, value) if open and close is None: close = open member.open = open member.close = close member.__doc__ = docstring member._value_ = value return member PLAIN = 'normal' BOLD_ITALIC = '***really super important***', '***' HIGHLIGHT = 'please ==take notice==', '==', '==' p = Huh.PLAIN self.assertTrue(type(p) is Huh, type(p)) self.assertEqual( (p.value, p.__doc__, p.open, p.close), (0, 'normal', None, None), ) bi = Huh.BOLD_ITALIC self.assertEqual( (bi.value, bi.__doc__, bi.open, bi.close), (1, '***really super important***', '***', '***'), ) h = Huh.HIGHLIGHT self.assertEqual( (h.value, h.__doc__, h.open, h.close), (2, 'please ==take notice==', '==', '=='), ) def test_members_is_ordereddict_if_ordered(self): class Ordered(Enum): __order__ = 'first second third' first = 'bippity' second = 'boppity' third = 'boo' self.assertTrue(type(Ordered.__members__) is OrderedDict) def test_members_is_ordereddict_if_not_ordered(self): class Unordered(Enum): this = 'that' these = 'those' self.assertTrue(type(Unordered.__members__) is OrderedDict) def test_enum_in_enum_out(self): Season = self.Season self.assertTrue(Season(Season.WINTER) is 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()): i += 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.assertTrue(e in Season) self.assertTrue(type(e) is Season) self.assertTrue(isinstance(e, Season)) self.assertEqual(str(e), 'Season.' + season) self.assertEqual( repr(e), '<Season.%s: %s>' % (season, i), ) def test_enum_helper(self): e1 = enum(1, 2, three=9) e2 = enum(1, 2, three=9) e3 = enum(1, 2, 9) self.assertTrue(e1 is not e2) self.assertEqual(e1, e2) self.assertNotEqual(e1, e3) self.assertNotEqual(e2, e3) def test_enum_in_enum(self): # class Level(Enum): _order_ = 'DATA_CHECK DESIGN_CHECK ALERT' # def __new__(cls, *args, **kwds): member = object.__new__(cls) member._value_ = len(cls) + 1 # members are 1-based return member # def __init__(self, prereq=None, dependent=None): # create priority level lists self.lower_priority_levels = list(self.__class__._member_map_.values()) self.greater_priority_levels = [] # update previous members' greater priority list for member in self.lower_priority_levels: member.greater_priority_levels.append(self) # and save prereq and dependent self.prerequisite = prereq and self.__class__[prereq.name] or None self.dependent = dependent and self.__class__[dependent.name] or None # DATA_CHECK = enum() DESIGN_CHECK = enum(DATA_CHECK) ALERT = enum(None, DATA_CHECK) # self.assertEqual(Level.DATA_CHECK.value, 1) self.assertEqual(Level.DATA_CHECK.prerequisite, None) self.assertEqual(Level.DATA_CHECK.dependent, None) self.assertEqual(Level.DESIGN_CHECK.prerequisite, Level.DATA_CHECK) self.assertEqual(Level.DESIGN_CHECK.dependent, None) self.assertEqual(Level.ALERT.prerequisite, None) self.assertEqual(Level.ALERT.dependent, Level.DATA_CHECK) def test_value_name(self): Season = self.Season self.assertEqual(Season.SPRING.name, 'SPRING') self.assertEqual(Season.SPRING.value, 1) def set_name(obj, new_value): obj.name = new_value def set_value(obj, new_value): obj.value = new_value self.assertRaises(AttributeError, set_name, Season.SPRING, 'invierno', ) self.assertRaises(AttributeError, set_value, Season.SPRING, 2) 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')) self.assertRaises(AttributeError, delattr, Season, 'SPRING') self.assertRaises(AttributeError, delattr, Season, 'DRY') self.assertRaises(AttributeError, delattr, Season.SPRING, 'name') 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)) def test_invalid_names(self): def create_bad_class_1(): class Wrong(Enum): mro = 9 def create_bad_class_2(): class Wrong(Enum): _reserved_ = 3 self.assertRaises(ValueError, create_bad_class_1) self.assertRaises(ValueError, create_bad_class_2) def test_bool(self): class Logic(Enum): true = True false = False def __bool__(self): return bool(self.value) __nonzero__ = __bool__ self.assertTrue(Logic.true) self.assertFalse(Logic.false) def test_contains(self): Season = self.Season self.assertFalse('AUTUMN' in Season) self.assertTrue(Season.AUTUMN in Season) self.assertTrue(3 in Season) val = Season(3) self.assertTrue(val in Season) self.assertFalse([] in Season) # class OtherEnum(Enum): one = 1; two = 2 self.assertTrue(OtherEnum.two not in Season) # class NumericEnum(IntEnum): one = 1; two = 2 self.assertTrue(NumericEnum.two in Season) self.assertTrue(Season(NumericEnum.two) is Season.SUMMER) # class Wierd(Enum): this = [1, 2, 3] that = (1, 2, 3) those = {1: 1, 2: 2, 3: 3} self.assertTrue(Wierd.this in Wierd) self.assertTrue((1, 2, 3) in Wierd) self.assertTrue({1: 1, 2: 2, 3: 3} in Wierd) self.assertFalse({1:1} in Wierd) def test_member_contains(self): self.assertRaises(TypeError, lambda: 'test' in self.Season.AUTUMN) def test_format_enum(self): Season = self.Season self.assertEqual('{0}'.format(Season.SPRING), '{0}'.format(str(Season.SPRING))) self.assertEqual( '{0:}'.format(Season.SPRING), '{0:}'.format(str(Season.SPRING))) self.assertEqual('{0:20}'.format(Season.SPRING), '{0:20}'.format(str(Season.SPRING))) self.assertEqual('{0:^20}'.format(Season.SPRING), '{0:^20}'.format(str(Season.SPRING))) self.assertEqual('{0:>20}'.format(Season.SPRING), '{0:>20}'.format(str(Season.SPRING))) self.assertEqual('{0:<20}'.format(Season.SPRING), '{0:<20}'.format(str(Season.SPRING))) def test_custom_format(self): class TestFloat(float, Enum): one = 1.0 two = 2.0 def __format__(self, spec): return 'TestFloat success!' self.assertEqual(str(TestFloat.one), 'TestFloat.one') self.assertEqual('{0}'.format(TestFloat.one), 'TestFloat success!') def test_format_with_custom_str(self): class TestInt(int, Enum): one = 1 two = 2 def __str__(self): return self.name * 3 self.assertEqual(str(TestInt.two), 'twotwotwo') self.assertEqual('{0}'.format(TestInt.two), 'twotwotwo') def assertFormatIsName(self, spec, member): self.assertEqual(spec.format(member), spec.format(str(member))) def test_format_enum_date(self): Holiday = self.Holiday self.assertFormatIsName('{0}', Holiday.IDES_OF_MARCH) self.assertFormatIsName('{0:}', Holiday.IDES_OF_MARCH) self.assertFormatIsName('{0:20}', Holiday.IDES_OF_MARCH) self.assertFormatIsName('{0:^20}', Holiday.IDES_OF_MARCH) self.assertFormatIsName('{0:>20}', Holiday.IDES_OF_MARCH) self.assertFormatIsName('{0:<20}', Holiday.IDES_OF_MARCH) def test_format_enum_float(self): Konstants = self.Konstants self.assertFormatIsName('{0}', Konstants.TAU) self.assertFormatIsName('{0:}', Konstants.TAU) self.assertFormatIsName('{0:20}', Konstants.TAU) self.assertFormatIsName('{0:^20}', Konstants.TAU) self.assertFormatIsName('{0:>20}', Konstants.TAU) self.assertFormatIsName('{0:<20}', Konstants.TAU) def test_format_enum_int(self): Grades = self.Grades self.assertFormatIsName('{0}', Grades.C) self.assertFormatIsName('{0:}', Grades.C) self.assertFormatIsName('{0:20}', Grades.C) self.assertFormatIsName('{0:^20}', Grades.C) self.assertFormatIsName('{0:>20}', Grades.C) self.assertFormatIsName('{0:<20}', Grades.C) def test_format_enum_str(self): Directional = self.Directional self.assertFormatIsName('{0}', Directional.WEST) self.assertFormatIsName('{0:}', Directional.WEST) self.assertFormatIsName('{0:20}', Directional.WEST) self.assertFormatIsName('{0:^20}', Directional.WEST) self.assertFormatIsName('{0:>20}', Directional.WEST) self.assertFormatIsName('{0:<20}', Directional.WEST) def assertFormatIsValue(self, spec, member): self.assertEqual(spec.format(member), spec.format(member.value)) def test_format_reprenum_date(self): ReprHoliday = self.ReprHoliday self.assertFormatIsValue('{0}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:20}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:^20}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:>20}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:<20}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:%Y %m}', ReprHoliday.IDES_OF_MARCH) self.assertFormatIsValue('{0:%Y %m %M:00}', ReprHoliday.IDES_OF_MARCH) def test_format_reprenum_float(self): ReprKonstants = self.ReprKonstants self.assertFormatIsValue('{0}', ReprKonstants.TAU) self.assertFormatIsValue('{0:}', ReprKonstants.TAU) self.assertFormatIsValue('{0:20}', ReprKonstants.TAU) self.assertFormatIsValue('{0:^20}', ReprKonstants.TAU) self.assertFormatIsValue('{0:>20}', ReprKonstants.TAU) self.assertFormatIsValue('{0:<20}', ReprKonstants.TAU) self.assertFormatIsValue('{0:n}', ReprKonstants.TAU) self.assertFormatIsValue('{0:5.2}', ReprKonstants.TAU) self.assertFormatIsValue('{0:f}', ReprKonstants.TAU) def test_format_reprenum_int(self): ReprGrades = self.ReprGrades self.assertFormatIsValue('{0}', ReprGrades.C) self.assertFormatIsValue('{0:}', ReprGrades.C) self.assertFormatIsValue('{0:20}', ReprGrades.C) self.assertFormatIsValue('{0:^20}', ReprGrades.C) self.assertFormatIsValue('{0:>20}', ReprGrades.C) self.assertFormatIsValue('{0:<20}', ReprGrades.C) self.assertFormatIsValue('{0:+}', ReprGrades.C) self.assertFormatIsValue('{0:08X}', ReprGrades.C) self.assertFormatIsValue('{0:b}', ReprGrades.C) def test_format_reprenum_str(self): ReprDirectional = self.ReprDirectional self.assertFormatIsValue('{0}', ReprDirectional.WEST) self.assertFormatIsValue('{0:}', ReprDirectional.WEST) self.assertFormatIsValue('{0:20}', ReprDirectional.WEST) self.assertFormatIsValue('{0:^20}', ReprDirectional.WEST) self.assertFormatIsValue('{0:>20}', ReprDirectional.WEST) self.assertFormatIsValue('{0:<20}', ReprDirectional.WEST) 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_enum_duplicates(self): class Season(Enum): __order__ = "SPRING SUMMER AUTUMN WINTER" 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.assertTrue(Season.FALL is Season.AUTUMN) self.assertEqual(Season.FALL.value, 3) self.assertEqual(Season.AUTUMN.value, 3) self.assertTrue(Season(3) is Season.AUTUMN) self.assertTrue(Season(1) is Season.SPRING) self.assertEqual(Season.FALL.name, 'AUTUMN') self.assertEqual( set([k for k,v in Season.__members__.items() if v.name != k]), set(['FALL', 'ANOTHER_SPRING']), ) def test_enum_with_value_name(self): class Huh(Enum): _order_ = 'name value' name = 1 value = 2 self.assertEqual( list(Huh), [Huh.name, Huh.value], ) self.assertTrue(type(Huh.name) is Huh) self.assertEqual(Huh.name.name, 'name') self.assertEqual(Huh.name.value, 1) 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): pi = 3.1415926 tau = 2 * pi self.assertTrue(phy.pi < phy.tau) def test_floatenum_inherited(self): class FloatEnum(float, Enum): pass class phy(FloatEnum): pi = 3.1415926 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) 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): i += 1 e = WeekDay(i) self.assertEqual(e, i) self.assertEqual(int(e), i) self.assertEqual(e.name, weekday) self.assertTrue(e in WeekDay) self.assertEqual(lst.index(e)+1, i) self.assertTrue(0 < e < 8) self.assertTrue(type(e) is WeekDay) self.assertTrue(isinstance(e, int)) self.assertTrue(isinstance(e, Enum)) def test_intenum_duplicates(self): class WeekDay(IntEnum): __order__ = 'SUNDAY MONDAY TUESDAY WEDNESDAY THURSDAY FRIDAY SATURDAY' SUNDAY = 1 MONDAY = 2 TUESDAY = TEUSDAY = 3 WEDNESDAY = 4 THURSDAY = 5 FRIDAY = 6 SATURDAY = 7 self.assertTrue(WeekDay.TEUSDAY is WeekDay.TUESDAY) self.assertEqual(WeekDay(3).name, 'TUESDAY') self.assertEqual([k for k,v in WeekDay.__members__.items() if v.name != k], ['TEUSDAY', ]) 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) def test_pickle_enum(self): if isinstance(Stooges, Exception): raise Stooges test_pickle_dump_load(self.assertTrue, Stooges.CURLY) test_pickle_dump_load(self.assertTrue, Stooges) def test_pickle_int(self): if isinstance(IntStooges, Exception): raise IntStooges test_pickle_dump_load(self.assertTrue, IntStooges.CURLY) test_pickle_dump_load(self.assertTrue, IntStooges) def test_pickle_float(self): if isinstance(FloatStooges, Exception): raise FloatStooges test_pickle_dump_load(self.assertTrue, FloatStooges.CURLY) test_pickle_dump_load(self.assertTrue, FloatStooges) def test_pickle_enum_function(self): if isinstance(Answer, Exception): raise Answer test_pickle_dump_load(self.assertTrue, Answer.him) test_pickle_dump_load(self.assertTrue, Answer) def test_pickle_enum_function_with_module(self): if isinstance(Question, Exception): raise Question test_pickle_dump_load(self.assertTrue, Question.who) test_pickle_dump_load(self.assertTrue, Question) def test_pickle_by_name(self): class ReplaceGlobalInt(IntEnum): ONE = 1 TWO = 2 ReplaceGlobalInt.__reduce_ex__ = _reduce_ex_by_name for proto in range(HIGHEST_PROTOCOL): self.assertEqual(ReplaceGlobalInt.TWO.__reduce_ex__(proto), 'TWO') def test_exploding_pickle(self): BadPickle = Enum('BadPickle', 'dill sweet bread-n-butter') aenum.make_class_unpicklable(BadPickle) globals()['BadPickle'] = BadPickle test_pickle_exception(self.assertRaises, TypeError, BadPickle.dill) test_pickle_exception(self.assertRaises, PicklingError, BadPickle) 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.assertTrue(Period(2) is Period.noon) self.assertTrue(getattr(Period, 'night') is Period.night) self.assertTrue(Period['morning'] is Period.morning) def test_getattr_dunder(self): Season = self.Season self.assertTrue(getattr(Season, '__hash__')) def test_iteration_order(self): class Season(Enum): __order__ = 'SUMMER WINTER AUTUMN SPRING' SUMMER = 2 WINTER = 4 AUTUMN = 3 SPRING = 1 self.assertEqual( list(Season), [Season.SUMMER, Season.WINTER, Season.AUTUMN, Season.SPRING], ) def test_iteration_order_reversed(self): self.assertEqual( list(reversed(self.Season)), [self.Season.WINTER, self.Season.AUTUMN, self.Season.SUMMER, self.Season.SPRING] ) def test_iteration_order_with_unorderable_values(self): class Complex(Enum): a = complex(7, 9) b = complex(3.14, 2) c = complex(1, -1) d = complex(-77, 32) self.assertEqual( list(Complex), [Complex.a, Complex.b, Complex.c, Complex.d], ) def test_programatic_function_string(self): 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()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_string_with_start(self): 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.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_string_list(self): 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()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_string_list_with_start(self): 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.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_iterable(self): 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()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_from_dict(self): SummerMonth = Enum( 'SummerMonth', dict((('june', 1), ('july', 2), ('august', 3))) ) lst = list(SummerMonth) self.assertEqual(len(lst), len(SummerMonth)) self.assertEqual(len(SummerMonth), 3, SummerMonth) if PY2: self.assertEqual( [SummerMonth.june, SummerMonth.july, SummerMonth.august], lst, ) for i, month in enumerate('june july august'.split()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_type(self): 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()): i += 1 e = SummerMonth(i) self.assertEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_type_with_start(self): 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.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_type_from_subclass(self): 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()): i += 1 e = SummerMonth(i) self.assertEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_type_from_subclass_with_start(self): 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.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_unicode(self): SummerMonth = Enum('SummerMonth', unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_unicode_list(self): SummerMonth = Enum('SummerMonth', [unicode('june'), unicode('july'), unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_unicode_iterable(self): SummerMonth = Enum( 'SummerMonth', ((unicode('june'), 1), (unicode('july'), 2), (unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_from_unicode_dict(self): SummerMonth = Enum( 'SummerMonth', dict(((unicode('june'), 1), (unicode('july'), 2), (unicode('august'), 3))) ) lst = list(SummerMonth) self.assertEqual(len(lst), len(SummerMonth)) self.assertEqual(len(SummerMonth), 3, SummerMonth) if PY2: self.assertEqual( [SummerMonth.june, SummerMonth.july, SummerMonth.august], lst, ) for i, month in enumerate(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(int(e.value), i) self.assertNotEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_unicode_type(self): SummerMonth = Enum('SummerMonth', unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programatic_function_unicode_type_from_subclass(self): SummerMonth = IntEnum('SummerMonth', unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(e, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) def test_programmatic_function_unicode_class(self): if PY2: class_names = unicode('SummerMonth'), 'S\xfcmm\xe9rM\xf6nth'.decode('latin1') else: class_names = 'SummerMonth', 'S\xfcmm\xe9rM\xf6nth' for i, class_name in enumerate(class_names): if PY2 and i == 1: self.assertRaises(TypeError, Enum, class_name, unicode('june july august')) else: SummerMonth = Enum(class_name, unicode('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(unicode('june july august').split()): i += 1 e = SummerMonth(i) self.assertEqual(e.value, i) self.assertEqual(e.name, month) self.assertTrue(e in SummerMonth) self.assertTrue(type(e) is SummerMonth) 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.assertTrue(Name.BDFL is getattr(Name, 'BDFL')) test_pickle_dump_load(self.assertTrue, Name.BDFL) def test_extending(self): def bad_extension(): class Color(Enum): red = 1 green = 2 blue = 3 class MoreColor(Color): cyan = 4 magenta = 5 yellow = 6 self.assertRaises(TypeError, bad_extension) def test_exclude_methods(self): class whatever(Enum): this = 'that' these = 'those' def really(self): return 'no, not %s' % self.value self.assertFalse(type(whatever.really) is whatever) self.assertEqual(whatever.this.really(), 'no, not that') def test_wrong_inheritance_order(self): def wrong_inherit(): class Wrong(Enum, str): NotHere = 'error before this point' self.assertRaises(TypeError, wrong_inherit) 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_introspection(self): class Number(IntEnum): one = 100 two = 200 self.assertTrue(Number.one._member_type_ is int) self.assertTrue(Number._member_type_ is int) class String(str, Enum): yarn = 'soft' rope = 'rough' wire = 'hard' self.assertTrue(String.yarn._member_type_ is str) self.assertTrue(String._member_type_ is str) class Plain(Enum): vanilla = 'white' one = 1 self.assertTrue(Plain.vanilla._member_type_ is object) self.assertTrue(Plain._member_type_ is object) 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.assertTrue(Monochrome(Gender.female) is 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.assertTrue(Monochrome(Gender.male) is 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) def test_no_such_enum_member(self): class Color(Enum): red = 1 green = 2 blue = 3 self.assertRaises(ValueError, Color, 4) self.assertRaises(KeyError, Color.__getitem__, '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(EnumMeta): def __new__(metacls, cls, bases, classdict): original_dict = classdict temp_dict = metacls.__prepare__(cls, bases, {}) if hasattr(original_dict, '_member_names'): for k in original_dict._member_names: temp_dict[k] = original_dict[k] sunders = [k for k in original_dict.keys() if aenum.is_sunder(k)] else: sunders = [] for k, v in original_dict.items(): if aenum.is_sunder(k): sunders.append(k) temp_dict[k] = v classdict = metacls.__prepare__(cls, bases, {}) i = 0 for k in sunders: classdict[k] = original_dict[k] for k in temp_dict._member_names: v = original_dict[k] if v == (): v = i else: i = v i += 1 classdict[k] = v for k, v in original_dict.items(): if k not in temp_dict._member_names and k not in sunders: classdict[k] = v return super(auto_enum, metacls).__new__( metacls, cls, bases, classdict) AutoNumberedEnum = auto_enum('AutoNumberedEnum', (Enum,), {}) AutoIntEnum = auto_enum('AutoIntEnum', (IntEnum,), {}) # class TestAutoNumber(AutoNumberedEnum): # a = () # b = 3 # c = () # self.assertEqual(TestAutoNumber.b.value, 3) # # if pyver >= 3.0: # self.assertEqual( # [TestAutoNumber.a.value, TestAutoNumber.b.value, TestAutoNumber.c.value], # [0, 3, 4], # ) # # class TestAutoInt(AutoIntEnum): # a = () # b = 3 # c = () # self.assertEqual(TestAutoInt.b, 3) # # if pyver >= 3.0: # self.assertEqual( # [TestAutoInt.a.value, TestAutoInt.b.value, TestAutoInt.c.value], # [0, 3, 4], # ) def test_meta_reconfigure(self): def identity(*args): if len(args) == 1: return args[0] return args JSONEnum = None class JSONEnumMeta(EnumMeta): @classmethod def __prepare__(metacls, cls, bases, init=None, start=None, settings=()): return {} def __init__(cls, *args , **kwds): super(JSONEnumMeta, cls).__init__(*args) def __new__(metacls, cls, bases, clsdict, init=None, start=None, settings=()): import json members = [] if JSONEnum is not None: if '_file' not in clsdict: raise TypeError('_file is required') if '_name' not in clsdict: raise TypeError('_name is required') if '_value' not in clsdict: raise TypeError('_value is required') name_spec = clsdict.pop('_name') if not isinstance(name_spec, (tuple, list)): name_spec = (name_spec, ) value_spec = clsdict.pop('_value') file = clsdict.pop('_file') with open(file) as f: json_data = json.load(f) for data in json_data: values = [] name = data[name_spec[0]] for piece in name_spec[1:]: name = name[piece] for order, (value_path, func) in sorted(value_spec.items()): if not isinstance(value_path, (list, tuple)): value_path = (value_path, ) value = data[value_path[0]] for piece in value_path[1:]: value = value[piece] if func is not None: value = func(value) values.append(value) values = tuple(values) members.append( (name, identity(*values)) ) # get the real EnumDict enum_dict = super(JSONEnumMeta, metacls).__prepare__(cls, bases, init, start, settings) # transfer the original dict content, _items first items = list(clsdict.items()) items.sort(key=lambda p: (0 if p[0][0] == '_' else 1, p)) for name, value in items: enum_dict[name] = value # add the members for name, value in members: enum_dict[name] = value return super(JSONEnumMeta, metacls).__new__(metacls, cls, bases, enum_dict, init, start, settings) # for use with both Python 2/3 JSONEnum = JSONEnumMeta('JsonEnum', (Enum, ), {}) test_file = os.path.join(tempdir, 'test_json.json') with open(test_file, 'w') as f: f.write( '[{"name":"Afghanistan","alpha-2":"AF","country-code":"004","notes":{"description":"pretty"}},' '{"name":"Åland Islands","alpha-2":"AX","country-code":"248","notes":{"description":"serene"}},' '{"name":"Albania","alpha-2":"AL","country-code":"008","notes":{"description":"exciting"}},' '{"name":"Algeria","alpha-2":"DZ","country-code":"012","notes":{"description":"scarce"}}]') class Country(JSONEnum): _init_ = 'abbr code country_name description' _file = test_file _name = 'alpha-2' _value = { 1: ('alpha-2', None), 2: ('country-code', lambda c: int(c)), 3: ('name', None), 4: (('notes','description'), lambda s: s.title()), } self.assertEqual([Country.AF, Country.AX, Country.AL, Country.DZ], list(Country)) self.assertEqual(Country.AF.abbr, 'AF') self.assertEqual(Country.AX.code, 248) self.assertEqual(Country.AL.country_name, 'Albania') self.assertEqual(Country.DZ.description, 'Scarce') def test_subclasses_with_getnewargs(self): class NamedInt(int): __qualname__ = 'NamedInt' # needed for pickle protocol 4 def __new__(cls, *args): _args = args if len(args) < 1: raise TypeError("name and value must be specified") name, args = args[0], args[1:] self = int.__new__(cls, *args) self._intname = name self._args = _args return self def __getnewargs__(self): return self._args @property def __name__(self): return self._intname def __repr__(self): # repr() is updated to include the name and type info return "%s(%r, %s)" % (type(self).__name__, self.__name__, int.__repr__(self)) 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( '(%s + %s)' % (self.__name__, other.__name__), temp ) else: return temp class NEI(NamedInt, Enum): __qualname__ = 'NEI' # needed for pickle protocol 4 x = ('the-x', 1) y = ('the-y', 2) self.assertTrue(NEI.__new__ is 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.assertTrue, NI5, 5) self.assertEqual(NEI.y.value, 2) test_pickle_dump_load(self.assertTrue, NEI.y) def test_subclasses_with_reduce(self): class NamedInt(int): __qualname__ = 'NamedInt' # needed for pickle protocol 4 def __new__(cls, *args): _args = args if len(args) < 1: raise TypeError("name and value must be specified") name, args = args[0], args[1:] 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 return "%s(%r, %s)" % (type(self).__name__, self.__name__, int.__repr__(self)) 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( '(%s + %s)' % (self.__name__, other.__name__), temp ) else: return temp class NEI(NamedInt, Enum): __qualname__ = 'NEI' # needed for pickle protocol 4 x = ('the-x', 1) y = ('the-y', 2) self.assertTrue(NEI.__new__ is 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.assertTrue, NEI.y) def test_subclasses_with_reduce_ex(self): class NamedInt(int): __qualname__ = 'NamedInt' # needed for pickle protocol 4 def __new__(cls, *args): _args = args if len(args) < 1: raise TypeError("name and value must be specified") name, args = args[0], args[1:] self = int.__new__(cls, *args) self._intname = name self._args = _args return self def __reduce_ex__(self, proto): 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 return "%s(%r, %s)" % (type(self).__name__, self.__name__, int.__repr__(self)) 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( '(%s + %s)' % (self.__name__, other.__name__), temp ) else: return temp class NEI(NamedInt, Enum): __qualname__ = 'NEI' # needed for pickle protocol 4 x = ('the-x', 1) y = ('the-y', 2) self.assertTrue(NEI.__new__ is 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.assertTrue, NEI.y) def test_subclasses_without_direct_pickle_support(self): class NamedInt(int): __qualname__ = 'NamedInt' def __new__(cls, *args): _args = args name, args = args[0], args[1:] 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 return "%s(%r, %s)" % (type(self).__name__, self.__name__, int.__repr__(self)) 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( '(%s + %s)' % (self.__name__, other.__name__), temp ) else: return temp class NEI(NamedInt, Enum): __qualname__ = 'NEI' x = ('the-x', 1) y = ('the-y', 2) self.assertTrue(NEI.__new__ is 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_exception(self.assertRaises, TypeError, NEI.x) test_pickle_exception(self.assertRaises, PicklingError, NEI) def test_subclasses_without_direct_pickle_support_using_name(self): class NamedInt(int): __qualname__ = 'NamedInt' def __new__(cls, *args): _args = args name, args = args[0], args[1:] 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 return "%s(%r, %s)" % (type(self).__name__, self.__name__, int.__repr__(self)) 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( '(%s + %s)' % (self.__name__, other.__name__), temp ) 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.assertTrue(NEI.__new__ is 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.assertTrue, NEI.y) test_pickle_dump_load(self.assertTrue, NEI) def test_tuple_subclass(self): class SomeTuple(tuple, Enum): __qualname__ = 'SomeTuple' first = (1, 'for the money') second = (2, 'for the show') third = (3, 'for the music') self.assertTrue(type(SomeTuple.first) is SomeTuple) self.assertTrue(isinstance(SomeTuple.second, tuple)) self.assertEqual(SomeTuple.third, (3, 'for the music')) globals()['SomeTuple'] = SomeTuple test_pickle_dump_load(self.assertTrue, SomeTuple.first) # def test_duplicate_values_give_unique_enum_items(self): # class NumericEnum(AutoNumberEnum): # __order__ = 'enum_m enum_d enum_y' # enum_m = () # enum_d = () # enum_y = () # def __int__(self): # return int(self._value_) # self.assertEqual(int(NumericEnum.enum_d), 2) # self.assertEqual(NumericEnum.enum_y.value, 3) # self.assertTrue(NumericEnum(1) is NumericEnum.enum_m) # self.assertEqual( # list(NumericEnum), # [NumericEnum.enum_m, NumericEnum.enum_d, NumericEnum.enum_y], # ) def test_inherited_new_from_enhanced_enum(self): class AutoNumber2(Enum): def __new__(cls): value = len(cls.__members__) + 1 obj = object.__new__(cls) obj._value_ = value return obj def __int__(self): return int(self._value_) class Color(AutoNumber2): __order__ = 'red green blue' red = () green = () blue = () self.assertEqual(len(Color), 3, "wrong number of elements: %d (should be %d)" % (len(Color), 3)) self.assertEqual(list(Color), [Color.red, Color.green, Color.blue]) if PY3: self.assertEqual(list(map(int, Color)), [1, 2, 3]) def test_inherited_new_from_mixed_enum(self): class AutoNumber3(IntEnum): def __new__(cls): value = len(cls.__members__) + 11 obj = int.__new__(cls, value) obj._value_ = value return obj class Color(AutoNumber3): __order__ = 'red green blue' red = () green = () blue = () self.assertEqual(len(Color), 3, "wrong number of elements: %d (should be %d)" % (len(Color), 3)) Color.red Color.green Color.blue self.assertEqual(Color.blue, 13) def test_equality(self): class AlwaysEqual: def __eq__(self, other): return True class OrdinaryEnum(Enum): a = 1 self.assertEqual(AlwaysEqual(), OrdinaryEnum.a) self.assertEqual(OrdinaryEnum.a, AlwaysEqual()) def test_ordered_mixin(self): class Grade(OrderedEnum): __order__ = 'A B C D F' A = 5 B = 4 C = 3 D = 2 F = 1 self.assertEqual(list(Grade), [Grade.A, Grade.B, Grade.C, Grade.D, Grade.F]) self.assertTrue(Grade.A > Grade.B) self.assertTrue(Grade.F <= Grade.C) self.assertTrue(Grade.D < Grade.A) self.assertTrue(Grade.B >= Grade.B) def test_missing_deprecated(self): class Label(Enum): AnyApple = 0 RedApple = 1 GreenApple = 2 @classmethod def _missing_(cls, name): return cls.AnyApple self.assertEqual(Label.AnyApple, Label(4)) with self.assertRaises(AttributeError): Label.redapple with self.assertRaises(KeyError): Label['redapple'] def test_missing(self): class Label(Enum): AnyApple = 0 RedApple = 1 GreenApple = 2 @classmethod def _missing_value_(cls, value): return cls.AnyApple self.assertEqual(Label.AnyApple, Label(4)) with self.assertRaises(AttributeError): Label.redapple with self.assertRaises(KeyError): Label['redapple'] def test_missing_name(self): class Label(Enum): RedApple = 1 GreenApple = 2 @classmethod def _missing_name_(cls, name): for member in cls: if member.name.lower() == name.lower(): return member Label['redapple'] with self.assertRaises(AttributeError): Label.redapple with self.assertRaises(ValueError): Label('redapple') def test_missing_value_bad_input(self): class Label(Enum): AnyApple = 0 RedApple = 1 GreenApple = 2 @classmethod def _missing_value_(cls, value): return cls.AnyApple self.assertEqual(Label.AnyApple, Label(4)) with self.assertRaises(KeyError): Label[True] def test_missing_name_bad_return(self): class Label(Enum): RedApple = 1 GreenApple = 2 @classmethod def _missing_name_(cls, name): return None with self.assertRaises(AttributeError): Label.redapple with self.assertRaises(ValueError): Label('redapple') with self.assertRaises(KeyError): Label['redapple'] def test_extending2(self): def bad_extension(): class Shade(Enum): def shade(self): print(self.name) class Color(Shade): red = 1 green = 2 blue = 3 class MoreColor(Color): cyan = 4 magenta = 5 yellow = 6 self.assertRaises(TypeError, bad_extension) 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!') def test_extending5(self): class Color(Enum): _order_ = 'red green blue value' red = 1 green = 2 blue = 3 value = 4 self.assertEqual(list(Color), [Color.red, Color.green, Color.blue, Color.value]) self.assertEqual(Color.value.name, 'value') self.assertEqual(Color.value.value, 4) self.assertTrue(Color.value in Color) self.assertEqual(Color(4), Color.value) self.assertEqual(Color['value'], Color.value) self.assertEqual(Color.red.value, 1) CONTINUE = 100, 'Continue', 'Request received, please continue' SWITCHING_PROTOCOLS = (101, 'Switching Protocols', 'Switching to new protocol; obey Upgrade header') PROCESSING = 102, 'Processing' def test_no_duplicates(self): def bad_duplicates(): class Color1(UniqueEnum): red = 1 green = 2 blue = 3 class Color2(UniqueEnum): red = 1 green = 2 blue = 3 grene = 2 self.assertRaises(ValueError, bad_duplicates) def test_no_duplicates_kinda(self): class Silly(UniqueEnum): one = 1 two = 'dos' name = 3 class Sillier(IntEnum, UniqueEnum): single = 1 name = 2 triple = 3 value = 4 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)) def test_init_and_shadowing_attribute(self): class SelectionEnum(str, Enum): _init_ = 'db user' def __new__(cls, *args, **kwds): count = len(cls.__members__) obj = str.__new__(cls, args[0]) obj._count = count obj._value_ = args return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (name, ) + args class DeviceTypeSource(SelectionEnum): _order_ = 'user system' user = "User controlled" system = "System controlled" self.assertEqual(DeviceTypeSource.system.db, 'system') self.assertEqual(DeviceTypeSource.system.user, 'System controlled') self.assertEqual(DeviceTypeSource.user.db, 'user') self.assertEqual(DeviceTypeSource.user.user, 'User controlled') def test_nonhash_value(self): class AutoNumberInAList(Enum): def __new__(cls): value = [len(cls.__members__) + 1] obj = object.__new__(cls) obj._value_ = value return obj class ColorInAList(AutoNumberInAList): __order__ = 'red green blue' red = () green = () blue = () self.assertEqual(list(ColorInAList), [ColorInAList.red, ColorInAList.green, ColorInAList.blue]) self.assertEqual(ColorInAList.red.value, [1]) self.assertEqual(ColorInAList([1]), ColorInAList.red) def test_number_reset_and_order_cleanup(self): class Confused(Enum): _order_ = 'ONE TWO THREE UNO DOS TRES FOUR' ONE = auto() TWO = auto() THREE = auto() UNO = 1 DOS = auto() TRES = auto() FOUR = auto() self.assertEqual(list(Confused), [Confused.ONE, Confused.TWO, Confused.THREE, Confused.FOUR]) self.assertIs(Confused.TWO, Confused.DOS) self.assertEqual(Confused.DOS._value_, 2) self.assertEqual(Confused.TRES._value_, 3) self.assertEqual(Confused.FOUR._value_, 4) 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(LabelledList.unprocessed, 1) self.assertEqual(LabelledList(1), LabelledList.unprocessed) self.assertEqual(list(LabelledList), [LabelledList.unprocessed, LabelledList.payment_complete]) def test_auto_number(self): class Color(Enum): _order_ = 'red blue green' 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): _order_ = 'red blue green' 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): _order_ = 'red blue green' 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): _order_ = 'red blue' red = 'red' blue = auto() self.assertEqual(Color.blue.value, 1) def test_auto_garbage_corrected(self): class Color(Enum): _order_ = 'red blue green' 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) def test_duplicate_auto(self): # class MoreDupes(Enum): _order_ = 'A B C' A = auto() B = A, C = auto() self.assertEqual(list(MoreDupes), [MoreDupes.A, MoreDupes.B, MoreDupes.C]) self.assertEqual([m.value for m in MoreDupes], [1, (1, ), 2]) # class Dupes(Enum): _order_ = 'first second third' first = primero = auto() second = auto() third = auto() self.assertEqual([Dupes.first, Dupes.second, Dupes.third], list(Dupes)) def test_auto_value_with_auto(self): class SelectionEnum(Enum): _init_ = 'db user' def __new__(cls, *args, **kwds): count = len(cls.__members__) obj = object.__new__(cls) obj._count = count obj._value_ = args obj.db, obj.user = args return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (name, ) + args class Test(SelectionEnum): _order_ = 'this that' this = auto('these') that = auto('those') self.assertEqual(list(Test), [Test.this, Test.that]) self.assertEqual(Test.this.name, 'this') self.assertEqual(Test.this.value, ('this', 'these')) self.assertEqual(Test.this.db, 'this') self.assertEqual(Test.this.user, 'these') self.assertEqual(Test.that.name, 'that') self.assertEqual(Test.that.value, ('that', 'those')) self.assertEqual(Test.that.db, 'that') self.assertEqual(Test.that.user, 'those') def test_auto_value_with_autovalue(self): class SelectionEnum(Enum): _init_ = 'db user' def __new__(cls, *args, **kwds): count = len(cls.__members__) obj = object.__new__(cls) obj._count = count obj._value_ = args return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (name, ) + args class Test(SelectionEnum): _order_ = 'this that' this = 'these' that = 'those' self.assertEqual(list(Test), [Test.this, Test.that]) self.assertEqual(Test.this.name, 'this') self.assertEqual(Test.this.value, ('this', 'these')) self.assertEqual(Test.this.db, 'this') self.assertEqual(Test.this.user, 'these') self.assertEqual(Test.that.name, 'that') self.assertEqual(Test.that.value, ('that', 'those')) self.assertEqual(Test.that.db, 'that') self.assertEqual(Test.that.user, 'those') def test_auto_and_kwds(self): class Item(Enum): _order_ = 'A B' A = auto(size=100, req={'red': True}) B = auto(size=200, req={'red': False}) # def __new__(cls, value, size, req): obj = object.__new__(cls) obj._value_ = value obj.size = size obj.req= req return obj self.assertEqual((Item.A.value, Item.A.size, Item.A.req), (1, 100, {'red': True})) self.assertEqual((Item.B.value, Item.B.size, Item.B.req), (2, 200, {'red': False})) def test_empty_with_functional_api(self): empty = aenum.IntEnum('Foo', {}) self.assertEqual(len(empty), 0) def test_auto_init(self): class Planet(Enum): _init_ = 'mass radius' 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) @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)) def test_auto_init_with_value(self): class Color(Enum): _init_='value, rgb' RED = 1, (1, 0, 0) BLUE = 2, (0, 1, 0) GREEN = 3, (0, 0, 1) self.assertEqual(Color.RED.value, 1) self.assertEqual(Color.BLUE.value, 2) self.assertEqual(Color.GREEN.value, 3) self.assertEqual(Color.RED.rgb, (1, 0, 0)) self.assertEqual(Color.BLUE.rgb, (0, 1, 0)) self.assertEqual(Color.GREEN.rgb, (0, 0, 1)) def test_noalias(self): class Settings(Enum): _settings_ = NoAlias red = 1 rojo = 1 self.assertFalse(Settings.red is Settings.rojo) self.assertRaises(TypeError, Settings, 1) def test_auto_and_init(self): class Field(int, Enum): _order_ = 'TYPE START' _init_ = 'value __doc__' TYPE = "Char, Date, Logical, etc." START = "Field offset in record" self.assertEqual(Field.TYPE, 1) self.assertEqual(Field.START, 2) self.assertEqual(Field.TYPE.__doc__, 'Char, Date, Logical, etc.') self.assertEqual(Field.START.__doc__, 'Field offset in record') def test_auto_and_start(self): class Field(IntEnum): _order_ = 'TYPE START' _start_ = 0 _init_ = 'value __doc__' TYPE = "Char, Date, Logical, etc." START = "Field offset in record" self.assertEqual(Field.TYPE, 0) self.assertEqual(Field.START, 1) self.assertEqual(Field.TYPE.__doc__, 'Char, Date, Logical, etc.') self.assertEqual(Field.START.__doc__, 'Field offset in record') def test_auto_and_init_and_some_values(self): class Field(int, Enum): _order_ = 'TYPE START BLAH BELCH' _init_ = 'value __doc__' TYPE = "Char, Date, Logical, etc." START = "Field offset in record" BLAH = 5, "test blah" BELCH = 'test belch' self.assertEqual(Field.TYPE, 1) self.assertEqual(Field.START, 2) self.assertEqual(Field.BLAH, 5) self.assertEqual(Field.BELCH, 6) self.assertEqual(Field.TYPE.__doc__, 'Char, Date, Logical, etc.') self.assertEqual(Field.START.__doc__, 'Field offset in record') self.assertEqual(Field.BLAH.__doc__, 'test blah') self.assertEqual(Field.BELCH.__doc__, 'test belch') def test_auto_and_init_w_value_and_too_many_values(self): with self.assertRaisesRegex(TypeError, r'Field\.BLAH: number of fields provided do not match init'): class Field(int, Enum): _order_ = 'TYPE START BLAH BELCH' _init_ = 'value __doc__' TYPE = 1, "Char, Date, Logical, etc." START = 2, "Field offset in record" BLAH = 5, 6, "test blah" BELCH = 7, 'test belch' def test_auto_and_init_and_some_complex_values(self): class Field(int, Enum): _order_ = 'TYPE START BLAH BELCH' _init_ = 'value __doc__ help' TYPE = "Char, Date, Logical, etc.", "fields composed of character data" START = "Field offset in record", "where the data starts in the record" BLAH = 5, "test blah", "some help" BELCH = 'test belch', "some more help" self.assertEqual(Field.TYPE, 1) self.assertEqual(Field.START, 2) self.assertEqual(Field.BLAH, 5) self.assertEqual(Field.BELCH, 6) self.assertEqual(Field.TYPE.__doc__, 'Char, Date, Logical, etc.') self.assertEqual(Field.START.__doc__, 'Field offset in record') self.assertEqual(Field.BLAH.__doc__, 'test blah') self.assertEqual(Field.BELCH.__doc__, 'test belch') self.assertEqual(Field.TYPE.help, "fields composed of character data") self.assertEqual(Field.START.help, "where the data starts in the record") self.assertEqual(Field.BLAH.help, "some help") self.assertEqual(Field.BELCH.help, "some more help") def test_auto_and_init_inherited(self): class AutoEnum(IntEnum): _start_ = 0 _init_ = 'value __doc__' class Field(AutoEnum): _order_ = 'TYPE START BLAH BELCH' TYPE = "Char, Date, Logical, etc." START = "Field offset in record" BLAH = 5, "test blah" BELCH = 'test belch' self.assertEqual(Field.TYPE, 0) self.assertEqual(Field.START, 1) self.assertEqual(Field.BLAH, 5) self.assertEqual(Field.BELCH, 6) self.assertEqual(Field.TYPE.__doc__, 'Char, Date, Logical, etc.') self.assertEqual(Field.START.__doc__, 'Field offset in record') self.assertEqual(Field.BLAH.__doc__, 'test blah') self.assertEqual(Field.BELCH.__doc__, 'test belch') def test_missing_value_error(self): with self.assertRaisesRegex(TypeError, r"_value_ not set in __new__"): class Combined(str, Enum): # _init_ = 'value sequence' _order_ = lambda m: m.sequence # def __new__(cls, value, *args): 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 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 def test_auto_and_enum(self): class Foo(aenum.Flag): _order_ = 'a b c' a = aenum.auto() b = a | aenum.auto() c = 2 self.assertEqual([Foo.a, Foo.c], list(Foo)) self.assertEqual(Foo.a.value, 1) self.assertEqual(Foo.b.value, 3) def test_multiple_arg_auto(self): class AutoName(Enum): def _generate_next_value_(name, start, count, last, *args, **kwds): return (name, ) + args # class Planet(AutoName): _init_ = 'value mass radius' MERCURY = auto(3.303e+23, 2.4397e6) VENUS = auto(4.869e+24, 6.0518e6) self.assertEqual(Planet.MERCURY.value, 'MERCURY') def test_auto_w_multiple_arg(self): class AutoName(Enum): def _generate_next_value_(name, start, count, last, *args, **kwds): if args: return (name, ) + args else: return name # class Planet(AutoName): _init_ = 'value mass radius' MERCURY = auto(), 3.303e+23, 2.4397e6 # doesn't work VENUS = auto(), 4.869e+24, 6.0518e6 # doesn't work self.assertEqual(Planet.MERCURY.value, 'MERCURY') def test_auto_gnv_and_init(self): class AutoName(Enum): def _generate_next_value_(name, start, count, last, *args, **kwds): return (name, ) + args # class Planet(AutoName): _init_ = 'value mass radius' MERCURY = 3.303e+23, 2.4397e6 # doesn't work VENUS = 4.869e+24, 6.0518e6 # doesn't work self.assertEqual(Planet.MERCURY.value, 'MERCURY') # def test_AutoNumberEnum_and_property(self): # class Color(aenum.AutoNumberEnum): # red = () # green = () # blue = () # @property # def cap_name(self): # return self.name.title() # self.assertEqual(Color.blue.cap_name, 'Blue') # def test_AutoNumberEnum(self): # class Color(aenum.AutoNumberEnum): # _order_ = 'red green blue' # red = () # green = () # blue = () # self.assertEqual(list(Color), [Color.red, Color.green, Color.blue]) # self.assertEqual(Color.red.value, 1) # self.assertEqual(Color.green.value, 2) # self.assertEqual(Color.blue.value, 3) def test_MultiValue_with_init_wo_value(self): class Color(Enum): _init_ = 'color r g b' _order_ = 'red green blue' _settings_ = MultiValue red = 'red', 1, 2, 3 green = 'green', 4, 5, 6 blue = 'blue', 7, 8, 9 self.assertEqual(Color.red.value, 'red') self.assertEqual(Color.red.color, 'red') self.assertEqual(Color.red.r, 1) self.assertEqual(Color.red.g, 2) self.assertEqual(Color.red.b, 3) self.assertEqual(Color.green.value, 'green') self.assertEqual(Color.green.color, 'green') self.assertEqual(Color.green.r, 4) self.assertEqual(Color.green.g, 5) self.assertEqual(Color.green.b, 6) self.assertEqual(Color.blue.value, 'blue') self.assertEqual(Color.blue.color, 'blue') self.assertEqual(Color.blue.r, 7) self.assertEqual(Color.blue.g, 8) self.assertEqual(Color.blue.b, 9) self.assertIs(Color('red'), Color.red) self.assertIs(Color(1), Color.red) self.assertIs(Color(2), Color.red) self.assertIs(Color(3), Color.red) self.assertIs(Color('green'), Color.green) self.assertIs(Color(4), Color.green) self.assertIs(Color(5), Color.green) self.assertIs(Color(6), Color.green) self.assertIs(Color('blue'), Color.blue) self.assertIs(Color(7), Color.blue) self.assertIs(Color(8), Color.blue) self.assertIs(Color(9), Color.blue) def test_MultiValue_with_init_w_value(self): class Color(Enum): _init_ = 'value r g b' _order_ = 'red green blue' _settings_ = MultiValue red = 'red', 1, 2, 3 green = 'green', 4, 5, 6 blue = 'blue', 7, 8, 9 self.assertEqual(Color.red.value, 'red') self.assertEqual(Color.red.r, 1) self.assertEqual(Color.red.g, 2) self.assertEqual(Color.red.b, 3) self.assertEqual(Color.green.value, 'green') self.assertEqual(Color.green.r, 4) self.assertEqual(Color.green.g, 5) self.assertEqual(Color.green.b, 6) self.assertEqual(Color.blue.value, 'blue') self.assertEqual(Color.blue.r, 7) self.assertEqual(Color.blue.g, 8) self.assertEqual(Color.blue.b, 9) self.assertIs(Color('red'), Color.red) self.assertIs(Color(1), Color.red) self.assertIs(Color(2), Color.red) self.assertIs(Color(3), Color.red) self.assertIs(Color('green'), Color.green) self.assertIs(Color(4), Color.green) self.assertIs(Color(5), Color.green) self.assertIs(Color(6), Color.green) self.assertIs(Color('blue'), Color.blue) self.assertIs(Color(7), Color.blue) self.assertIs(Color(8), Color.blue) self.assertIs(Color(9), Color.blue) def test_MultiValue_with_init_wo_value_w_autonumber(self): class Color(AutoNumberEnum): _init_ = 'color r g b' _order_ = 'red green blue' _settings_ = MultiValue red = 'red', 10, 20, 30 green = 'green', 40, 50, 60 blue = 'blue', 70, 80, 90 self.assertEqual(Color.red.value, 1) self.assertEqual(Color.red.color, 'red') self.assertEqual(Color.red.r, 10) self.assertEqual(Color.red.g, 20) self.assertEqual(Color.red.b, 30) self.assertEqual(Color.green.value, 2) self.assertEqual(Color.green.color, 'green') self.assertEqual(Color.green.r, 40) self.assertEqual(Color.green.g, 50) self.assertEqual(Color.green.b, 60) self.assertEqual(Color.blue.value, 3) self.assertEqual(Color.blue.color, 'blue') self.assertEqual(Color.blue.r, 70) self.assertEqual(Color.blue.g, 80) self.assertEqual(Color.blue.b, 90) self.assertIs(Color(1), Color.red) self.assertIs(Color('red'), Color.red) self.assertIs(Color(10), Color.red) self.assertIs(Color(20), Color.red) self.assertIs(Color(30), Color.red) self.assertIs(Color(2), Color.green) self.assertIs(Color('green'), Color.green) self.assertIs(Color(40), Color.green) self.assertIs(Color(50), Color.green) self.assertIs(Color(60), Color.green) self.assertIs(Color(3), Color.blue) self.assertIs(Color('blue'), Color.blue) self.assertIs(Color(70), Color.blue) self.assertIs(Color(80), Color.blue) self.assertIs(Color(90), Color.blue) def test_multivalue_and_autonumber_wo_init_wo_value(self): class Day(Enum): _settings_ = MultiValue, AddValue _order_ = 'one two three' _start_ = 7 one = "21", "one" two = "22", "two" three = "23", "three" self.assertEqual(Day.one.value, 7) self.assertEqual(Day.two.value, 8) self.assertEqual(Day.three.value, 9) self.assertEqual(Day('21'), Day.one) self.assertEqual(Day('one'), Day.one) def test_multivalue_and_autonumber_wo_init_w_some_value(self): class Color(Enum): _settings_ = MultiValue, Unique _order_ = 'BLACK RED BLUE YELLOW GREEN MAGENTA' _init_ = "value description" BLACK = -1, "Text0" RED = -50, "Text1" BLUE = auto(), "Text2" YELLOW = auto(), "Text3" GREEN = -70, "Text4" MAGENTA = auto(), "Text5" self.assertEqual(Color.BLACK.value, -1) self.assertEqual(Color.RED.value, -50) self.assertEqual(Color.BLUE.value, -49) self.assertEqual(Color.YELLOW.value, -48) self.assertEqual(Color.GREEN.value, -70) self.assertEqual(Color.MAGENTA.value, -69) self.assertEqual(Color(-1), Color.BLACK) self.assertEqual(Color('Text2'), Color.BLUE) def test_combine_new_settings_with_old_settings(self): class Auto(Enum): _settings_ = Unique with self.assertRaises(ValueError): class AutoUnique(Auto): BLAH = auto() BLUH = auto() ICK = 1 def test_timedelta(self): class Period(timedelta, Enum): ''' different lengths of time ''' _init_ = 'value period' _settings_ = NoAlias _ignore_ = 'Period i' Period = vars() for i in range(31): Period['day_%d' % i] = i, 'day' for i in range(15): Period['week_%d' % i] = i*7, 'week' for i in range(12): Period['month_%d' % i] = i*30, 'month' OneDay = day_1 OneWeek = week_1 self.assertFalse(hasattr(Period, '_ignore_')) self.assertFalse(hasattr(Period, 'Period')) self.assertFalse(hasattr(Period, 'i')) self.assertTrue(isinstance(Period.day_1, timedelta)) def test_skip(self): class enumA(Enum): @skip class enumB(Enum): elementA = 'a' elementB = 'b' @skip class enumC(Enum): elementC = 'c' elementD = 'd' self.assertIs(enumA.enumB, enumA.__dict__['enumB']) def test_nonmember(self): class enumA(Enum): @nonmember class enumB(Enum): elementA = 'a' elementB = 'b' @nonmember class enumC(Enum): elementC = 'c' elementD = 'd' self.assertIs(enumA.enumB, enumA.__dict__['enumB']) def test_member_with_external_functions(self): class Func(Enum): _order_ = 'an_int a_str' an_int = member(int) a_str = member(str) @classproperty def types(cls): return [m.value for m in list(cls)] def __repr__(self): return "<%s.%s>" % (self.__class__.__name__, self.name, ) def __call__(self, *args, **kwds): return self.value(*args, **kwds) # self.assertEqual([Func.an_int, Func.a_str], list(Func)) self.assertEqual([int, str], Func.types) self.assertEqual(Func.an_int(7), 7) self.assertEqual(Func.a_str('BlahBlah'), 'BlahBlah') def test_member_with_internal_functions(self): class Func(Enum): _order_ = 'haha hehe' @member def haha(): return 'haha' @member def hehe(name): return 'hehe -- what a name! %s!' % name @classproperty def types(cls): return [m.value for m in list(cls)] def __repr__(self): return "<%s.%s>" % (self.__class__.__name__, self.name, ) def __call__(self, *args, **kwds): return self.value(*args, **kwds) # self.assertEqual([Func.haha, Func.hehe], list(Func)) self.assertEqual([Func.haha.value, Func.hehe.value], Func.types) self.assertEqual(Func.haha(), 'haha') self.assertEqual(Func.hehe('BlahBlah'), 'hehe -- what a name! BlahBlah!') def test_constantness_of_constants(self): class Universe(Enum): PI = constant(3.141596) G = constant(6.67300E-11) self.assertEqual(Universe.PI, 3.141596) self.assertRaisesRegex(AttributeError, r'cannot rebind constant', setattr, Universe, 'PI', 9) self.assertRaisesRegex(AttributeError, r'cannot delete constant', delattr, Universe, 'PI') def test_math_and_stuff_with_constants(self): class Universe(Enum): PI = constant(3.141596) TAU = constant(2 * PI) self.assertEqual(Universe.PI, 3.141596) self.assertEqual(Universe.TAU, 2 * Universe.PI) def test_constant_with_auto_is_updated(self): class Fruit(Flag): _order_ = 'apple banana lemon orange' apple = auto() banana = auto() lemon = auto() orange = auto() CitrusTypes = constant(lemon | orange) self.assertEqual(list(Fruit), [Fruit.apple, Fruit.banana, Fruit.lemon, Fruit.orange]) self.assertEqual(list(Fruit.CitrusTypes), [Fruit.lemon, Fruit.orange]) self.assertTrue(Fruit.orange in Fruit.CitrusTypes) def test_order_as_function(self): # first with _init_ class TestSequence(Enum): _init_ = 'value, sequence' _order_ = lambda member: member.sequence item_id = 'An$(1,6)', 0 # Item Code company_id = 'An$(7,2)', 1 # Company Code warehouse_no = 'An$(9,4)', 2 # Warehouse Number company = 'Hn$(13,6)', 3 # 4 SPACES + COMPANY key_type = 'Cn$(19,3)', 4 # Key Type = '1**' available = 'Zn$(1,1)', 5 # Available? contract_item = 'Bn(2,1)', 6 # Contract Item? sales_category = 'Fn', 7 # Sales Category gl_category = 'Rn$(5,1)', 8 # G/L Category warehouse_category = 'Sn$(6,1)', 9 # Warehouse Category inv_units = 'Qn$(7,2)', 10 # Inv Units for i, member in enumerate(TestSequence): self.assertEqual(i, member.sequence) ts = TestSequence self.assertEqual(ts.item_id.name, 'item_id') self.assertEqual(ts.item_id.value, 'An$(1,6)') self.assertEqual(ts.item_id.sequence, 0) self.assertEqual(ts.company_id.name, 'company_id') self.assertEqual(ts.company_id.value, 'An$(7,2)') self.assertEqual(ts.company_id.sequence, 1) self.assertEqual(ts.warehouse_no.name, 'warehouse_no') self.assertEqual(ts.warehouse_no.value, 'An$(9,4)') self.assertEqual(ts.warehouse_no.sequence, 2) self.assertEqual(ts.company.name, 'company') self.assertEqual(ts.company.value, 'Hn$(13,6)') self.assertEqual(ts.company.sequence, 3) self.assertEqual(ts.key_type.name, 'key_type') self.assertEqual(ts.key_type.value, 'Cn$(19,3)') self.assertEqual(ts.key_type.sequence, 4) self.assertEqual(ts.available.name, 'available') self.assertEqual(ts.available.value, 'Zn$(1,1)') self.assertEqual(ts.available.sequence, 5) self.assertEqual(ts.contract_item.name, 'contract_item') self.assertEqual(ts.contract_item.value, 'Bn(2,1)') self.assertEqual(ts.contract_item.sequence, 6) self.assertEqual(ts.sales_category.name, 'sales_category') self.assertEqual(ts.sales_category.value, 'Fn') self.assertEqual(ts.sales_category.sequence, 7) self.assertEqual(ts.gl_category.name, 'gl_category') self.assertEqual(ts.gl_category.value, 'Rn$(5,1)') self.assertEqual(ts.gl_category.sequence, 8) self.assertEqual(ts.warehouse_category.name, 'warehouse_category') self.assertEqual(ts.warehouse_category.value, 'Sn$(6,1)') self.assertEqual(ts.warehouse_category.sequence, 9) self.assertEqual(ts.inv_units.name, 'inv_units') self.assertEqual(ts.inv_units.value, 'Qn$(7,2)') self.assertEqual(ts.inv_units.sequence, 10) # and then without class TestSequence(Enum): _order_ = lambda member: member.value[1] item_id = 'An$(1,6)', 0 # Item Code company_id = 'An$(7,2)', 1 # Company Code warehouse_no = 'An$(9,4)', 2 # Warehouse Number company = 'Hn$(13,6)', 3 # 4 SPACES + COMPANY key_type = 'Cn$(19,3)', 4 # Key Type = '1**' available = 'Zn$(1,1)', 5 # Available? contract_item = 'Bn(2,1)', 6 # Contract Item? sales_category = 'Fn', 7 # Sales Category gl_category = 'Rn$(5,1)', 8 # G/L Category warehouse_category = 'Sn$(6,1)', 9 # Warehouse Category inv_units = 'Qn$(7,2)', 10 # Inv Units for i, member in enumerate(TestSequence): self.assertEqual(i, member.value[1]) ts = TestSequence self.assertEqual(ts.item_id.name, 'item_id') self.assertEqual(ts.item_id.value, ('An$(1,6)', 0)) self.assertEqual(ts.company_id.name, 'company_id') self.assertEqual(ts.company_id.value, ('An$(7,2)', 1)) self.assertEqual(ts.warehouse_no.name, 'warehouse_no') self.assertEqual(ts.warehouse_no.value, ('An$(9,4)', 2)) self.assertEqual(ts.company.name, 'company') self.assertEqual(ts.company.value, ('Hn$(13,6)', 3)) self.assertEqual(ts.key_type.name, 'key_type') self.assertEqual(ts.key_type.value, ('Cn$(19,3)', 4)) self.assertEqual(ts.available.name, 'available') self.assertEqual(ts.available.value, ('Zn$(1,1)', 5)) self.assertEqual(ts.contract_item.name, 'contract_item') self.assertEqual(ts.contract_item.value, ('Bn(2,1)', 6)) self.assertEqual(ts.sales_category.name, 'sales_category') self.assertEqual(ts.sales_category.value, ('Fn', 7)) self.assertEqual(ts.gl_category.name, 'gl_category') self.assertEqual(ts.gl_category.value, ('Rn$(5,1)', 8)) self.assertEqual(ts.warehouse_category.name, 'warehouse_category') self.assertEqual(ts.warehouse_category.value, ('Sn$(6,1)', 9)) self.assertEqual(ts.inv_units.name, 'inv_units') self.assertEqual(ts.inv_units.value, ('Qn$(7,2)', 10)) # then with _init_ but without value with self.assertRaises(TypeError): class TestSequence(Enum): _init_ = 'sequence' _order_ = lambda member: member.sequence item_id = 'An$(1,6)', 0 # Item Code company_id = 'An$(7,2)', 1 # Company Code warehouse_no = 'An$(9,4)', 2 # Warehouse Number company = 'Hn$(13,6)', 3 # 4 SPACES + COMPANY key_type = 'Cn$(19,3)', 4 # Key Type = '1**' available = 'Zn$(1,1)', 5 # Available? contract_item = 'Bn(2,1)', 6 # Contract Item? sales_category = 'Fn', 7 # Sales Category gl_category = 'Rn$(5,1)', 8 # G/L Category warehouse_category = 'Sn$(6,1)', 9 # Warehouse Category inv_units = 'Qn$(7,2)', 10 # Inv Units # finally, out of order so Python 3 barfs with self.assertRaises(TypeError): class TestSequence(Enum): _init_ = 'sequence' _order_ = lambda member: member.sequence item_id = 'An$(1,6)', 0 # Item Code warehouse_no = 'An$(9,4)', 2 # Warehouse Number company = 'Hn$(13,6)', 3 # 4 SPACES + COMPANY company_id = 'An$(7,2)', 1 # Company Code inv_units = 'Qn$(7,2)', 10 # Inv Units available = 'Zn$(1,1)', 5 # Available? contract_item = 'Bn(2,1)', 6 # Contract Item? sales_category = 'Fn', 7 # Sales Category key_type = 'Cn$(19,3)', 4 # Key Type = '1**' gl_category = 'Rn$(5,1)', 8 # G/L Category warehouse_category = 'Sn$(6,1)', 9 # Warehouse Category def test_order_as_function_in_subclass(self): # class Parent(Enum): _init_ = 'value sequence' _order_ = lambda m: m.sequence # class Child(Parent): item_id = 'An$(1,6)', 0 # Item Code company_id = 'An$(7,2)', 1 # Company Code warehouse_no = 'An$(9,4)', 2 # Warehouse Number company = 'Hn$(13,6)', 3 # 4 SPACES + COMPANY key_type = 'Cn$(19,3)', 4 # Key Type = '1**' available = 'Zn$(1,1)', 5 # Available? contract_item = 'Bn(2,1)', 6 # Contract Item? sales_category = 'Fn', 7 # Sales Category gl_category = 'Rn$(5,1)', 8 # G/L Category warehouse_category = 'Sn$(6,1)', 9 # Warehouse Category inv_units = 'Qn$(7,2)', 10 # Inv Units # for i, member in enumerate(Child): self.assertEqual(i, member.sequence) # ts = Child self.assertEqual(ts.item_id.name, 'item_id') self.assertEqual(ts.item_id.value, 'An$(1,6)') self.assertEqual(ts.item_id.sequence, 0) self.assertEqual(ts.company_id.name, 'company_id') self.assertEqual(ts.company_id.value, 'An$(7,2)') self.assertEqual(ts.company_id.sequence, 1) self.assertEqual(ts.warehouse_no.name, 'warehouse_no') self.assertEqual(ts.warehouse_no.value, 'An$(9,4)') self.assertEqual(ts.warehouse_no.sequence, 2) self.assertEqual(ts.company.name, 'company') self.assertEqual(ts.company.value, 'Hn$(13,6)') self.assertEqual(ts.company.sequence, 3) self.assertEqual(ts.key_type.name, 'key_type') self.assertEqual(ts.key_type.value, 'Cn$(19,3)') self.assertEqual(ts.key_type.sequence, 4) self.assertEqual(ts.available.name, 'available') self.assertEqual(ts.available.value, 'Zn$(1,1)') self.assertEqual(ts.available.sequence, 5) self.assertEqual(ts.contract_item.name, 'contract_item') self.assertEqual(ts.contract_item.value, 'Bn(2,1)') self.assertEqual(ts.contract_item.sequence, 6) self.assertEqual(ts.sales_category.name, 'sales_category') self.assertEqual(ts.sales_category.value, 'Fn') self.assertEqual(ts.sales_category.sequence, 7) self.assertEqual(ts.gl_category.name, 'gl_category') self.assertEqual(ts.gl_category.value, 'Rn$(5,1)') self.assertEqual(ts.gl_category.sequence, 8) self.assertEqual(ts.warehouse_category.name, 'warehouse_category') self.assertEqual(ts.warehouse_category.value, 'Sn$(6,1)') self.assertEqual(ts.warehouse_category.sequence, 9) self.assertEqual(ts.inv_units.name, 'inv_units') self.assertEqual(ts.inv_units.value, 'Qn$(7,2)') self.assertEqual(ts.inv_units.sequence, 10) pass def test_multiple_mixin(self): class MaxMixin(object): @classproperty def MAX(cls): max = len(cls) cls.MAX = max return max class StrMixin(object): 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): _order_ = 'RED GREEN BLUE' 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), 'Color.BLUE') class Color(MaxMixin, StrMixin, Enum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) class Color(StrMixin, MaxMixin, Enum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) class CoolColor(StrMixin, SomeEnum, Enum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) self.assertEqual(CoolColor.RED.behavior(), 'booyah') class CoolerColor(StrMixin, AnotherEnum, Enum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) self.assertEqual(CoolerColor.RED.behavior(), 'nuhuh!') self.assertEqual(CoolerColor.RED.social(), "what's up?") class CoolestColor(StrMixin, SomeEnum, AnotherEnum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) self.assertEqual(CoolestColor.RED.behavior(), 'booyah') self.assertEqual(CoolestColor.RED.social(), "what's up?") class ConfusedColor(StrMixin, AnotherEnum, SomeEnum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) self.assertEqual(ConfusedColor.RED.behavior(), 'nuhuh!') self.assertEqual(ConfusedColor.RED.social(), "what's up?") class ReformedColor(StrMixin, IntEnum, SomeEnum, AnotherEnum): _order_ = 'RED GREEN BLUE' 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', '%r is not %r' % (str(Color.BLUE), 'blue')) self.assertEqual(ReformedColor.RED.behavior(), 'booyah') self.assertEqual(ConfusedColor.RED.social(), "what's up?") self.assertTrue(issubclass(ReformedColor, int)) 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" def test_value_auto_assign(self): class Some(Enum): def __new__(cls, val): return object.__new__(cls) x = 1 y = 2 self.assertEqual(Some.x.value, 1) self.assertEqual(Some.y.value, 2) def test_enum_of_types(self): """Support using Enum to refer to types deliberately.""" class MyTypes(Enum): i = int f = float s = str self.assertEqual(MyTypes.i.value, int) self.assertEqual(MyTypes.f.value, float) self.assertEqual(MyTypes.s.value, str) class Foo: pass class Bar: pass class MyTypes2(Enum): a = Foo b = Bar self.assertEqual(MyTypes2.a.value, Foo) self.assertEqual(MyTypes2.b.value, Bar) class SpamEnumNotInner: pass class SpamEnum(Enum): spam = SpamEnumNotInner self.assertEqual(SpamEnum.spam.value, SpamEnumNotInner) if PY2: def test_nested_classes_in_enum_do_become_members(self): # manually set __qualname__ to remove testing framework noise class Outer(Enum): _order_ = 'a b Inner' __qualname__ = "Outer" a = 1 b = 2 class Inner(Enum): __qualname__ = "Outer.Inner" foo = 10 bar = 11 self.assertTrue(isinstance(Outer.Inner, Outer)) self.assertEqual(Outer.a.value, 1) self.assertEqual(Outer.Inner.value.foo.value, 10) self.assertEqual( list(Outer.Inner.value), [Outer.Inner.value.foo, Outer.Inner.value.bar], ) self.assertEqual( list(Outer), [Outer.a, Outer.b, Outer.Inner], ) def test_really_nested_classes_in_enum_do_become_members(self): class Outer(Enum): _order_ = 'a b Inner' a = 1 b = 2 class Inner(Enum): foo = 10 bar = 11 self.assertTrue(isinstance(Outer.Inner, Outer)) self.assertEqual(Outer.a.value, 1) self.assertEqual(Outer.Inner.value.foo.value, 10) self.assertEqual( list(Outer.Inner.value), [Outer.Inner.value.foo, Outer.Inner.value.bar], ) self.assertEqual( list(Outer), [Outer.a, Outer.b, Outer.Inner], ) def test_nested_classes_in_enum_are_skipped_with_skip(self): """Support locally-defined nested classes using @skip""" # manually set __qualname__ to remove testing framework noise class Outer(Enum): __qualname__ = "Outer" a = 1 b = 2 @skip class Inner(Enum): __qualname__ = "Outer.Inner" foo = 10 bar = 11 self.assertTrue(isinstance(Outer.Inner, type)) self.assertEqual(Outer.a.value, 1) self.assertEqual(Outer.Inner.foo.value, 10) self.assertEqual( list(Outer.Inner), [Outer.Inner.foo, Outer.Inner.bar], ) self.assertEqual( list(Outer), [Outer.a, Outer.b], ) def test_really_nested_classes_in_enum_are_skipped_with_skip(self): """Support locally-defined nested classes using @skip""" class Outer(Enum): a = 1 b = 2 @skip class Inner(Enum): foo = 10 bar = 11 self.assertTrue(isinstance(Outer.Inner, type)) self.assertEqual(Outer.a.value, 1) self.assertEqual(Outer.Inner.foo.value, 10) self.assertEqual( list(Outer.Inner), [Outer.Inner.foo, Outer.Inner.bar], ) self.assertEqual( list(Outer), [Outer.a, Outer.b], ) def test_enum_call_without_arg(self): class Color(Enum): black = 0 red = 1 green = 2 blue = 3 # @classmethod def _missing_value_(cls, value): if value is no_arg: return cls.black self.assertTrue(Color.red is Color(1)) self.assertTrue(Color.black is Color()) def test_init_subclass(self): class MyEnum(Enum): def __init_subclass__(cls, **kwds): super(MyEnum, cls).__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 = 'one' 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)) def test_init_exception(self): class Base(object): def __new__(cls, *args): return object.__new__(cls) def __init__(self, x): raise ValueError("I don't like", x) with self.assertRaises(TypeError): class MyEnum(Base, Enum): A = 'a' def __init__(self, y): self.y = y with self.assertRaises(ValueError): class MyEnum(Base, Enum): A = 'a' def __init__(self, y): self.y = y def __new__(cls, value): member = Base.__new__(cls) member._value_ = Base(value) return member def test_namedtuple_as_value(self): from collections import namedtuple TTuple = namedtuple('TTuple', 'id a blist') class NTEnum(Enum): NONE = TTuple(0, 0, []) A = TTuple(1, 2, [4]) B = TTuple(2, 4, [0, 1, 2]) self.assertEqual(repr(NTEnum.NONE), "<NTEnum.NONE: TTuple(id=0, a=0, blist=[])>") self.assertEqual(NTEnum.NONE.value, TTuple(id=0, a=0, blist=[])) self.assertEqual( [x.value for x in NTEnum], [TTuple(id=0, a=0, blist=[]), TTuple(id=1, a=2, blist=[4]), TTuple(id=2, a=4, blist=[0, 1, 2])], ) def test_gnv_is_static(self): class LazyGNV(Enum): def _generate_next_value_(name, start, last, values): pass class BusyGNV(Enum): @staticmethod def _generate_next_value_(name, start, last, values): pass self.assertTrue(type(LazyGNV.__dict__['_generate_next_value_']) is staticmethod) self.assertTrue(type(BusyGNV.__dict__['_generate_next_value_']) is staticmethod) def test_namedtuple_as_value(self): from collections import namedtuple TTuple = namedtuple('TTuple', 'id a blist') class NTEnum(Enum): NONE = TTuple(0, 0, []) A = TTuple(1, 2, [4]) B = TTuple(2, 4, [0, 1, 2]) self.assertEqual(repr(NTEnum.NONE), "<NTEnum.NONE: TTuple(id=0, a=0, blist=[])>") self.assertEqual(NTEnum.NONE.value, TTuple(id=0, a=0, blist=[])) self.assertEqual( [x.value for x in NTEnum], [TTuple(id=0, a=0, blist=[]), TTuple(id=1, a=2, blist=[4]), TTuple(id=2, a=4, blist=[0, 1, 2])], ) def test_multiple_auto_assignment(self): class Many(Enum): _order_ = 'A B C' A = auto(), 2, auto() B = 4, 5, auto() C = auto(), auto(), auto() self.assertEqual(len(Many), 3) self.assertEqual(Many.A.value, (1, 2, 2)) self.assertEqual(Many.B.value, (4, 5, 3)) self.assertEqual(Many.C.value, (4, 5, 6)) class TestStrEnum(TestCase): def test_set_name(self): class Descriptor(object): name = None def __get__(self, instance, owner_class=None): if instance is None: return self else: return instance.__dict__[self.name] def __set__(self, instance, value): instance.__dict__[self.name] = value def __set_name__(self, owner, name): self.name = name # class AnEnum(Enum): ONE = 'one' two = Descriptor() # self.assertEqual(list(AnEnum), [AnEnum.ONE]) self.assertEqual(AnEnum.two.name, 'two') AnEnum.ONE.two = 'three' self.assertEqual(AnEnum.ONE.two, 'three') self.assertEqual(AnEnum.ONE.__dict__['two'], 'three') def test_private_names(self): class Private(Enum): __corporal = 'Radar' __major_ = 'Hoolihan' self.assertEqual(len(Private), 0) self.assertEqual(Private._Private__corporal, 'Radar') self.assertFalse(isinstance(Private._Private__corporal, Enum)) self.assertEqual(Private._Private__major_, 'Hoolihan') self.assertFalse(isinstance(Private._Private__major_, Enum)) def test_strenum_inherited_methods(self): class phy(StrEnum): pi = 'Pi' tau = 'Tau' self.assertTrue(phy.pi < phy.tau) self.assertEqual(phy.pi.upper(), 'PI') self.assertEqual(phy.tau.count('a'), 1) def test_strict_strenum(self): for uhoh in (object, object(), [], Enum, 9): with self.assertRaisesRegex(TypeError, r'values must be str'): class Huh(StrEnum): huh = uhoh class Either(StrEnum): _order_ = 'this that Those lower upper' this = auto() that = 'That' Those = auto() lower = 'lower' upper = 'UPPER' self.assertEqual([m.value for m in Either], ['this', 'That', 'those', 'lower', 'UPPER']) # with self.assertRaisesRegex(ValueError, r' is not lower-case'): class Huh(LowerStrEnum): huh = 'What' # class Lower(LowerStrEnum): _order_ = 'this that Those lower upper' this = auto() that = 'that' Those = auto() lower = 'lower' upper = 'upper' self.assertEqual([m.value for m in Lower], ['this', 'that', 'those', 'lower', 'upper']) # with self.assertRaisesRegex(ValueError, r' is not upper-case'): class Huh(UpperStrEnum): huh = 'What' # class Upper(UpperStrEnum): _order_ = 'this that Those lower upper' this = auto() that = 'THAT' Those = auto() lower = 'LOWER' upper = 'UPPER' self.assertEqual([m.value for m in Upper], ['THIS', 'THAT', 'THOSE', 'LOWER', 'UPPER']) def test_init_subclass(self): class MyEnum(StrEnum): def __init_subclass__(cls, **kwds): super(MyEnum, cls).__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 = 'one' 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 = 'one' 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)) class TestFlag(TestCase): """Tests of the Flags.""" def setUp(self): class Perm(Flag): _order_ = 'R W X' R, W, X = 4, 2, 1 self.Perm = Perm # class Color(Flag): BLACK = 0 RED = 1 ROJO = 1 GREEN = 2 BLUE = 4 PURPLE = RED|BLUE WHITE = RED|GREEN|BLUE BLANCO = RED|GREEN|BLUE self.Color = Color # class Fun(Flag): _order_ = 'ONE TWO FOUR EIGHT' ONE = auto() TWO = auto() THREE = ONE | TWO FOUR = auto() FIVE = FOUR | ONE SIX = FOUR | TWO SEVEN = FOUR | TWO | ONE EIGHT = auto() self.Fun = Fun # class TermColor(str, Flag): def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj # @classmethod def _create_pseudo_member_values_(cls, members, *values): code = ';'.join(m.code for m in members) return values + (code, ) # AllReset = '0' # ESC [ 0 m # reset all (colors and brightness) Bright = '1' # ESC [ 1 m # bright Dim = '2' # ESC [ 2 m # dim (looks same as normal brightness) Underline = '4' Normal = '22' # ESC [ 22 m # normal brightness # # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Yellow = '33' # ESC [ 33 m # yellow FG_Blue = '34' # ESC [ 34 m # blue FG_Magenta = '35' # ESC [ 35 m # magenta FG_Cyan = '36' # ESC [ 36 m # cyan FG_White = '37' # ESC [ 37 m # white FG_Reset = '39' # ESC [ 39 m # reset # BG_Black = '40' # ESC [ 30 m # black BG_Red = '41' # ESC [ 31 m # red BG_Green = '42' # ESC [ 32 m # green BG_Yellow = '43' # ESC [ 33 m # yellow BG_Blue = '44' # ESC [ 34 m # blue BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white BG_Reset = '49' # ESC [ 39 m # reset # __str__ = str.__str__ # def __repr__(self): if self._name_ is not None: return '<%s.%s>' % (self.__class__.__name__, self._name_) else: return '<%s: %s>' % (self.__class__.__name__, '|'.join([m.name for m in Flag.__iter__(self)])) # def __enter__(self): print(self.AllReset, end='', verbose=0) return self # def __exit__(self, *args): print(self.AllReset, end='', verbose=0) self.TermColor = TermColor # class Open(Flag): RO = 0 WO = 1 RW = 2 AC = 3 CE = 1<<19 self.Open = Open def test_set_name(self): class Descriptor(object): name = None def __get__(self, instance, owner_class=None): if instance is None: return self else: return instance.__dict__[self.name] def __set__(self, instance, value): instance.__dict__[self.name] = value def __set_name__(self, owner, name): self.name = name # class AnEnum(Enum): ONE = 1 two = Descriptor() # self.assertEqual(list(AnEnum), [AnEnum.ONE]) self.assertEqual(AnEnum.two.name, 'two') AnEnum.ONE.two = 'three' self.assertEqual(AnEnum.ONE.two, 'three') self.assertEqual(AnEnum.ONE.__dict__['two'], 'three') def test_new_with_keywords(self): class Huh(IntFlag): __order__ = 'PLAIN BOLD_ITALIC HIGHLIGHT' def __new__(cls, docstring, open=None, close=None): if cls.__members__: value = 2 ** (len(cls.__members__)-1) else: value = 0 member = int.__new__(cls, value) if open and close is None: close = open member.open = open member.close = close member.__doc__ = docstring member._value_ = value return member PLAIN = 'normal' BOLD_ITALIC = '***really super important***', '***' HIGHLIGHT = 'please ==take notice==', '==', '==' p = Huh.PLAIN self.assertTrue(type(p) is Huh, type(p)) self.assertEqual( (p.value, p.__doc__, p.open, p.close), (0, 'normal', None, None), ) bi = Huh.BOLD_ITALIC self.assertEqual( (bi.value, bi.__doc__, bi.open, bi.close), (1, '***really super important***', '***', '***'), ) h = Huh.HIGHLIGHT self.assertEqual( (h.value, h.__doc__, h.open, h.close), (2, 'please ==take notice==', '==', '=='), ) def test_private_names(self): class Private(Enum): __corporal = 'Radar' __major_ = 'Hoolihan' self.assertEqual(len(Private), 0) self.assertEqual(Private._Private__corporal, 'Radar') self.assertFalse(isinstance(Private._Private__corporal, Enum)) self.assertEqual(Private._Private__major_, 'Hoolihan') self.assertFalse(isinstance(Private._Private__major_, Enum)) def test_auto_alias(self): Fun = self.Fun self.assertEqual( list(Fun), [Fun.ONE, Fun.TWO, Fun.FOUR, Fun.EIGHT], ) self.assertEqual(Fun.THREE._value_, 3) self.assertEqual(repr(Fun.SEVEN), '<Fun.SEVEN: 7>') self.assertEqual(list(Fun.SEVEN), [Fun.ONE, Fun.TWO, Fun.FOUR]) def test_str_is_str_str(self): red, white = self.TermColor.FG_Red, self.TermColor.BG_White barber = red | white self.assertEqual(barber, '\x1b[31;47m') self.assertEqual(barber.value, red.value | white.value) self.assertEqual(barber.code, ';'.join([red.code, white.code])) self.assertEqual(repr(barber), '<TermColor.FG_Red|BG_White>') self.assertEqual(str(barber), '\x1b[31;47m') def test_membership(self): Color = self.Color Open = self.Open self.assertFalse('BLACK' in Color) self.assertFalse('RO' in Open) self.assertTrue(Color.BLACK in Color) self.assertTrue(Open.RO in Open) self.assertFalse(Color.BLACK in Open) self.assertFalse(Open.RO in Color) self.assertTrue(0 in Color) self.assertTrue(0 in Open) def test_member_contains(self): Color = self.Color self.assertRaises(TypeError, lambda: 'test' in Color.BLUE) self.assertRaises(TypeError, lambda: 2 in Color.BLUE) self.assertTrue(Color.BLUE in Color.BLUE) self.assertTrue(Color.BLUE in Color['RED|GREEN|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_str(self): Perm = self.Perm self.assertEqual(str(Perm.R), 'Perm.R') self.assertEqual(str(Perm.W), 'Perm.W') self.assertEqual(str(Perm.X), 'Perm.X') self.assertEqual(str(Perm.R | Perm.W), 'Perm.R|W') self.assertEqual(str(Perm.R | Perm.W | Perm.X), 'Perm.R|W|X') self.assertEqual(str(Perm(0)), 'Perm(0)') self.assertEqual(str(~Perm.R), 'Perm.W|X') self.assertEqual(str(~Perm.W), 'Perm.R|X') self.assertEqual(str(~Perm.X), 'Perm.R|W') self.assertEqual(str(~(Perm.R | Perm.W)), 'Perm.X') self.assertEqual(str(~(Perm.R | Perm.W | Perm.X)), 'Perm(0)') self.assertEqual(str(Perm(-1)), 'Perm.R|W|X') self.assertEqual(str(Perm(~0)), 'Perm.R|W|X') Open = self.Open self.assertEqual(str(Open.RO), 'Open.RO') self.assertEqual(str(Open.WO), 'Open.WO') self.assertEqual(str(Open.AC), 'Open.AC') self.assertEqual(str(Open.RO | Open.CE), 'Open.CE') self.assertEqual(str(Open.WO | Open.CE), 'Open.WO|CE') self.assertEqual(str(~Open.RO), 'Open.WO|RW|CE') self.assertEqual(str(~Open.WO), 'Open.RW|CE') self.assertEqual(str(~Open.AC), 'Open.CE') self.assertEqual(str(~(Open.RO | Open.CE)), 'Open.AC') self.assertEqual(str(~(Open.WO | Open.CE)), 'Open.RW') def test_repr(self): Perm = self.Perm self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm(~0)), '<Perm.R|W|X: 7>') Open = self.Open self.assertEqual(repr(Open.RO), '<Open.RO: 0>') self.assertEqual(repr(Open.WO), '<Open.WO: 1>') self.assertEqual(repr(Open.AC), '<Open.AC: 3>') self.assertEqual(repr(Open.RO | Open.CE), '<Open.CE: 524288>') self.assertEqual(repr(Open.WO | Open.CE), '<Open.WO|CE: 524289>') self.assertEqual(repr(~Open.RO), '<Open.WO|RW|CE: 524291>') self.assertEqual(repr(~Open.WO), '<Open.RW|CE: 524290>') self.assertEqual(repr(~Open.AC), '<Open.CE: 524288>') self.assertEqual(repr(~(Open.RO | Open.CE)), '<Open.AC: 3>') self.assertEqual(repr(~(Open.WO | Open.CE)), '<Open.RW: 2>') def test_name_lookup(self): Color = self.Color self.assertTrue(Color.RED is Color['RED']) self.assertTrue(Color.RED|Color.GREEN is Color['RED|GREEN']) self.assertTrue(Color.PURPLE is Color['RED|BLUE']) 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) 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)) def test_doc_flag(self): class DocFlag(Flag): _init_ = 'value __doc__' _start_ = 0 # def __new__(cls, value, doc=None): # # if doc is None and isinstance(value, basestring): # # value, doc = doc, value # # if value is None: # # if not len(cls): # # value = 0 # # else: # # value = 2 ** (len(cls) -1) # # if not isinstance(value, baseinteger): # # raise TypeError("%r is not a valid %s value" % (value, cls.__name__)) # obj = object.__new__(cls) # # if doc is None, don't mess with the value # if doc: # value = value >> 1 # obj._value_ = value # obj.__doc__ = doc # return obj # class AddressSegment(DocFlag): _order_ = 'UNKNOWN PO PO_TYPE NUMBER PREORD NAME STREET POSTORD SECONDARY_TYPE SECONDARY_NUMBER AND' UNKNOWN = "unable to determine address element type" PO = "post office delivery" PO_TYPE = "box or drawer" NUMBER = "main unit designator" PREORD = "N S E W etc" NAME = "street name" STREET = "st ave blvd etc" POSTORD = "N S E W etc" SECONDARY_TYPE = "apt bldg floor etc" SECONDARY_NUMBER = "secondary unit designator" AND = "& indicates a corner address" AS = AddressSegment self.assertEqual(AS.NAME._value_, 16) self.assertEqual(AS.STREET._value_, 32) self.assertEqual(AS.SECONDARY_TYPE._value_, 128) self.assertEqual((AS.NAME | AS.STREET)._value_, 48, "%r is not 48" % (AS.NAME | AS.STREET)) def test_iteration(self): C = self.Color self.assertEqual(list(C), [C.RED, C.GREEN, C.BLUE]) self.assertEqual(list(C.PURPLE), [C.RED, C.BLUE]) def test_member_iteration(self): C = self.Color self.assertEqual(list(C.BLACK), []) self.assertEqual(list(C.RED), [C.RED]) self.assertEqual(list(C.PURPLE), [C.RED, C.BLUE]) def test_programatic_function_string(self): Perm = Flag('Perm', 'R W X') 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): Perm = Flag('Perm', 'R W X', start=8) 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): Perm = Flag('Perm', ['R', 'W', 'X']) 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): Perm = Flag('Perm', (('R', 2), ('W', 8), ('X', 32))) 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): Perm = Flag('Perm', OrderedDict((('R', 2), ('W', 8), ('X', 32)))) 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_empty_list(self): Perm = IntFlag('Perm', []) self.assertEqual(len(list(Perm)), len(Perm)) self.assertEqual(len(Perm), 0) Thing = Enum('Thing', []) self.assertEqual(len(list(Thing)), len(Thing)) self.assertEqual(len(Thing), 0) def test_programatic_function_empty_tuple(self): Perm = IntFlag('Perm', ()) self.assertEqual(len(list(Perm)), len(Perm)) self.assertEqual(len(Perm), 0) Thing = Enum('Thing', ()) self.assertEqual(len(list(Thing)), len(Thing)) self.assertEqual(len(Thing), 0) 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, ) test_pickle_dump_load( self.assertEqual, FlagStooges.CURLY & ~FlagStooges.CURLY, ) test_pickle_dump_load( self.assertIs, FlagStooges, ) test_pickle_dump_load( self.assertEqual, FlagStooges.CMASK, ) test_pickle_dump_load( self.assertEqual, FlagStooges.CURLY | FlagStooges.CMASK, ) test_pickle_dump_load( self.assertEqual, FlagStooges.PMASK, ) test_pickle_dump_load( self.assertEqual, FlagStooges.CURLY | FlagStooges.PMASK, ) test_pickle_dump_load( self.assertIs, FlagStoogesWithZero.CURLY, ) test_pickle_dump_load( self.assertEqual, FlagStoogesWithZero.CURLY | FlagStoogesWithZero.MOE, ) test_pickle_dump_load( self.assertIs, FlagStoogesWithZero.NOFLAG, ) test_pickle_dump_load( self.assertEqual, FlagStoogesWithZero.CMASK, ) test_pickle_dump_load( self.assertEqual, FlagStoogesWithZero.CURLY | FlagStoogesWithZero.CMASK, ) test_pickle_dump_load( self.assertEqual, FlagStoogesWithZero.PMASK, ) test_pickle_dump_load( self.assertEqual, FlagStoogesWithZero.CURLY | FlagStoogesWithZero.PMASK, ) test_pickle_dump_load( self.assertIs, IntFlagStooges.CURLY, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CURLY | IntFlagStooges.MOE, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CURLY | IntFlagStooges.MOE|0x30, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges(0), ) test_pickle_dump_load( self.assertEqual, IntFlagStooges(0x30), ) test_pickle_dump_load( self.assertIs, IntFlagStooges, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CMASK | 1, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CURLY | IntFlagStooges.CMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.PMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.PMASK | 1, ) test_pickle_dump_load( self.assertEqual, IntFlagStooges.CURLY | IntFlagStooges.PMASK, ) test_pickle_dump_load( self.assertIs, IntFlagStoogesWithZero.CURLY, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.CURLY | IntFlagStoogesWithZero.MOE, ) test_pickle_dump_load( self.assertIs, IntFlagStoogesWithZero.NOFLAG, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.CMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.CMASK | 1, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.CURLY | IntFlagStoogesWithZero.CMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.PMASK, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.PMASK | 1, ) test_pickle_dump_load( self.assertEqual, IntFlagStoogesWithZero.CURLY | IntFlagStoogesWithZero.PMASK, ) def test_containment(self): 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) def test_auto_number(self): class Color(Flag): _order_ = 'red blue green' 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, r'invalid Flag value: .not an int.'): class Color(Flag): _order_ = 'red blue' red = 'not an int' blue = auto() def test_auto_w_pending(self): class Required(Flag): _order_ = 'TO_S FROM_S' NONE = 0 TO_S = auto() FROM_S = auto() BOTH = TO_S | FROM_S self.assertEqual(Required.TO_S.value, 1) self.assertEqual(Required.FROM_S.value, 2) self.assertEqual(Required.BOTH.value, 3) def test_duplicate_auto(self): class Dupes(Enum): _order_ = 'first second third' first = primero = auto() second = auto() third = auto() self.assertEqual([Dupes.first, Dupes.second, Dupes.third], list(Dupes)) def test_bizarre(self): class Bizarre(Flag): b = 3 c = 4 d = 6 # no error means the bizarre flag was created def test_multiple_mixin(self): class AllMixin(object): @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(object): def __str__(self): return self._name_.lower() class Color(AllMixin, Flag): _order_ = 'RED GREEN BLUE' 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), 'Color.BLUE') class Color(AllMixin, StrMixin, Flag): _order_ = 'RED GREEN BLUE' 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): _order_ = 'RED GREEN BLUE' 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') @unittest.skipUnless(threading, 'Threading required for this test.') def test_unique_composite(self): # override __eq__ to be identity only class TestFlag(Flag): _order_ = 'one two three four five six seven eight' 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[0] = True threads = [ threading.Thread(target=cycle_enum) for _ in range(8) ] for t in threads: t.start() for t in threads: t.join() # check that only 248 members were created (8 were created originally) self.assertFalse( failed[0], 'at least one thread failed while creating composite members') self.assertEqual(256, len(seen), 'too many composite members created') def test_init_with_autovalue_and_generate_next_value(self): class Color(Flag): _init_ = 'value code' def _generate_next_value_(name, start, count, last_values, *args, **kwds): if not count: return ((1, start)[start is not None], ) + args error = False for last_value in reversed(last_values): try: high_bit = _high_bit(last_value) break except Exception as e: exc = e error = True break if error: raise exc return (2 ** (high_bit+1), ) + args # TODO: actually test _create_pseudo_member @classmethod def _create_pseudo_member_(cls, value): members = list(cls._iter_member_(value)) pseudo_member = super(Color, cls)._create_pseudo_member_(value) pseudo_member.code = ';'.join(m.code for m in members) return pseudo_member AllReset = '0' # ESC [ 0 m # reset all (colors and brightness) Bright = '1' # ESC [ 1 m # bright Dim = '2' # ESC [ 2 m # dim (looks same as normal brightness) Underline = '4' Normal = '22' # ESC [ 22 m # normal brightness # if we got here, we're good def test_autovalue_and_generate_next_value(self): class Color(str, Flag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (2 ** count, ) + args # TODO: actually test _create_pseudo_member @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member # # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white # if we got here, we're good def test_subclass(self): class Color(str, Flag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (2 ** count, ) + args @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member # # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, '\x1b[30m') self.assertEqual(Color.FG_Black.code, '30') def test_sub_subclass_1(self): class StrFlag(str, Flag): def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member class Color(StrFlag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, '\x1b[30m') self.assertEqual(Color.FG_Black.code, '30') def test_sub_subclass_2(self): class StrFlag(str, Flag): @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (2 ** count, ) + args @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member # class Color(StrFlag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, '\x1b[30m') self.assertEqual(Color.FG_Black.code, '30') def test_sub_subclass_3(self): class StrFlag(str, Flag): def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member # class Color(StrFlag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, '\x1b[30m') self.assertEqual(Color.FG_Black.code, '30') def test_sub_subclass_4(self): class StrFlag(str, Flag): def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @classmethod def _create_pseudo_member_values_(cls, members, *values): code = ';'.join(m.code for m in members) return values + (code, ) # class Color(StrFlag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white # def __repr__(self): return '<%s.%s>' % (self.__class__.__name__, self._name_) self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, '\x1b[30m') self.assertEqual(Color.FG_Black.code, '30') colors = Color.BG_Magenta | Color.FG_Black self.assertTrue(isinstance(colors, Color)) self.assertTrue(isinstance(colors, str)) self.assertEqual(colors, '\x1b[30;45m') self.assertEqual(colors.code, '30;45') self.assertEqual(repr(colors), '<Color.FG_Black|BG_Magenta>') def test_sub_subclass_with_new_new(self): class StrFlag(str, Flag): def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(StrFlag, cls)._create_pseudo_member_(value, code) return pseudo_member # class Color(StrFlag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' def __new__(cls, value, string, abbr): str_value = (abbr or '').title() obj = str.__new__(cls, str_value) obj._value_ = value obj.code = string obj.abbr = abbr return obj # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30', 'blk' # ESC [ 30 m # black FG_Red = '31', 'red' # ESC [ 31 m # red FG_Green = '32', 'grn' # ESC [ 32 m # green FG_Blue = '34', 'blu' # ESC [ 34 m # blue # BG_Yellow = '43', 'ylw' # ESC [ 33 m # yellow BG_Magenta = '45', 'mag' # ESC [ 35 m # magenta BG_Cyan = '46', 'cyn' # ESC [ 36 m # cyan BG_White = '47', 'wht' # ESC [ 37 m # white # def __repr__(self): if self._name_ is not None: return '<%s.%s>' % (self.__class__.__name__, self._name_) else: return '<%s: %s>' % (self.__class__.__name__, '|'.join([m.name for m in self])) self.assertTrue(isinstance(Color.FG_Black, Color)) self.assertTrue(isinstance(Color.FG_Black, str)) self.assertEqual(Color.FG_Black, 'Blk', str.__repr__(Color.FG_Black)) self.assertEqual(Color.FG_Black.abbr, 'blk') def test_subclass_with_default_new(self): class MyFlag(str, Flag): _order_ = 'this these theother' this = 'that' these = 'those' theother = 'thingimibobs' self.assertEqual(MyFlag.this, 'that') self.assertEqual(MyFlag.this.value, 1) self.assertEqual(MyFlag.these, 'those') self.assertEqual(MyFlag.these.value, 2) self.assertEqual(MyFlag.theother, 'thingimibobs') self.assertEqual(MyFlag.theother.value, 4) def test_subclass_a_bunch(self): class Color(str, Flag): _order_ = 'FG_Black FG_Red FG_Green FG_Blue BG_Yellow BG_Magenta BG_Cyan BG_White' def __new__(cls, value, code): str_value = '\x1b[%sm' % code obj = str.__new__(cls, str_value) obj._value_ = value obj.code = code return obj @staticmethod def _generate_next_value_(name, start, count, values, *args, **kwds): return (2 ** count, ) + args @classmethod def _create_pseudo_member_(cls, value): # calculate the code members = list(cls._iter_member_(value)) code = ';'.join(m.code for m in members) pseudo_member = super(Color, cls)._create_pseudo_member_(value, code) return pseudo_member # # # FOREGROUND - 30s BACKGROUND - 40s: FG_Black = '30' # ESC [ 30 m # black FG_Red = '31' # ESC [ 31 m # red FG_Green = '32' # ESC [ 32 m # green FG_Blue = '34' # ESC [ 34 m # blue # BG_Yellow = '43' # ESC [ 33 m # yellow BG_Magenta = '45' # ESC [ 35 m # magenta BG_Cyan = '46' # ESC [ 36 m # cyan BG_White = '47' # ESC [ 37 m # white # def __repr__(self): if self._name_ is not None: return '<%s.%s>' % (self.__class__.__name__, self._name_) else: return '<%s: %s>' % (self.__class__.__name__, '|'.join([m.name for m in self])) # Purple = Color.BG_Magenta | Color.FG_Blue self.assertTrue(isinstance(Purple, Color)) self.assertTrue(isinstance(Purple, str)) self.assertIs(Purple, Color.BG_Magenta | Color.FG_Blue) self.assertEqual(Purple, '\x1b[34;45m') self.assertEqual(Purple.code, '34;45') self.assertEqual(Purple.name, 'FG_Blue|BG_Magenta') def test_init_subclass(self): class MyEnum(Flag): def __init_subclass__(cls, **kwds): super(MyEnum, cls).__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)) def test_int_long_conversion(self): class Perm(Flag): EXEC = 1 << 0 WRITE = 1 << 1 READ = 1 << 2 MSB32 = 1 << 31 MSB64 = 1 << 63 # 32-bit system test self.assertEqual(Perm.MSB32, Perm(0x80000000)) self.assertEqual(Perm.WRITE|Perm.MSB32, Perm(0x80000002)) # 64-bit system test self.assertEqual(Perm.MSB64, Perm(0x8000000000000000)) self.assertEqual(Perm.MSB64|Perm.WRITE, Perm(0x8000000000000002)) class TestIntFlag(TestCase): """Tests of the IntFlags.""" def setUp(self): # class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 # class Color(IntFlag): BLACK = 0 RED = 1 GREEN = 2 BLUE = 4 PURPLE = RED|BLUE # class Open(IntFlag): "not a good flag candidate" RO = 0 WO = 1 RW = 2 AC = 3 CE = 1<<19 # self.Perm = Perm self.Color = Color self.Open = Open def test_set_name(self): class Descriptor(object): name = None def __get__(self, instance, owner_class=None): if instance is None: return self else: return instance.__dict__[self.name] def __set__(self, instance, value): instance.__dict__[self.name] = value def __set_name__(self, owner, name): self.name = name # class AnEnum(Enum): ONE = 1 two = Descriptor() # self.assertEqual(list(AnEnum), [AnEnum.ONE]) self.assertEqual(AnEnum.two.name, 'two') AnEnum.ONE.two = 'three' self.assertEqual(AnEnum.ONE.two, 'three') self.assertEqual(AnEnum.ONE.__dict__['two'], 'three') def test_private_names(self): class Private(Enum): __corporal = 'Radar' __major_ = 'Hoolihan' self.assertEqual(len(Private), 0) self.assertEqual(Private._Private__corporal, 'Radar') self.assertFalse(isinstance(Private._Private__corporal, Enum)) self.assertEqual(Private._Private__major_, 'Hoolihan') self.assertFalse(isinstance(Private._Private__major_, Enum)) def test_membership(self): Color = self.Color Open = self.Open self.assertFalse('GREEN' in Color) self.assertFalse('RW' in Open) self.assertTrue(Color.GREEN in Color) self.assertTrue(Open.RW in Open) self.assertTrue(Color.GREEN in Open) # True because Open(2) is valid self.assertTrue(Open.RW in Color) # True because Color(2) is valid self.assertTrue(2 in Color) self.assertTrue(2 in Open) def test_member_contains(self): Color = self.Color self.assertRaises(TypeError, lambda: 'test' in Color.RED) self.assertTrue(1 in Color.RED) self.assertTrue(Color.RED in Color.RED) self.assertTrue(Color.RED in Color.PURPLE) def test_name_lookup(self): Color = self.Color self.assertTrue(Color.RED is Color['RED']) self.assertTrue(Color.RED|Color.GREEN is Color['RED|GREEN']) self.assertTrue(Color.PURPLE is Color['RED|BLUE']) def test_type(self): Perm = self.Perm 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) def test_str(self): Perm = self.Perm self.assertEqual(str(Perm.R), '4') self.assertEqual(str(Perm.W), '2') self.assertEqual(str(Perm.X), '1') self.assertEqual(str(Perm.R | Perm.W), '6') self.assertEqual(str(Perm.R | Perm.W | Perm.X), '7') self.assertEqual(str(Perm(0)), '0') self.assertEqual(str(~Perm.R), '3') self.assertEqual(str(~Perm.W), '5') self.assertEqual(str(~Perm.X), '6') self.assertEqual(str(~(Perm.R | Perm.W)), '1') self.assertEqual(str(~(Perm.R | Perm.W | Perm.X)), '0') self.assertEqual(str(Perm(~0)), '7') Open = self.Open self.assertEqual(str(Open.RO), '0') self.assertEqual(str(Open.WO), '1') self.assertEqual(str(Open.AC), '3') self.assertEqual(str(Open.RO | Open.CE), '524288') self.assertEqual(str(Open.WO | Open.CE), '524289') self.assertEqual(str(~Open.RO), '524291') self.assertEqual(str(~Open.WO), '524290') self.assertEqual(str(~Open.AC), '524288') self.assertEqual(str(~(Open.RO | Open.CE)), '3') self.assertEqual(str(~(Open.WO | Open.CE)), '2') def test_repr_strict(self): # test with complete flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 Perm._boundary_ = aenum.STRICT self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') # with self.assertRaisesRegex(ValueError, r'12 is not a valid Perm'): repr(Perm.R | 8) with self.assertRaisesRegex(ValueError, r'12 is not a valid Perm'): repr(~(Perm.R | 8)) with self.assertRaisesRegex(ValueError, r'-9 is not a valid Perm'): repr(Perm(~8)) # # test with open flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 FUTURE = 31 Perm._boundary_ = aenum.STRICT self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') # with self.assertRaisesRegex(ValueError, r'12 is not a valid Perm'): repr(Perm.R | 8) with self.assertRaisesRegex(ValueError, r'12 is not a valid Perm'): repr(~(Perm.R | 8)) with self.assertRaisesRegex(ValueError, r'-9 is not a valid Perm'): repr(Perm(~8)) def test_repr_conform(self): # test with complete flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 Perm._boundary_ = aenum.CONFORM self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '<Perm.R: 4>') self.assertEqual(repr(Perm(8)), '<Perm: 0>') self.assertEqual(repr(~(Perm.R | 8)), '<Perm.W|X: 3>') self.assertEqual(repr(Perm(~8)), '<Perm.R|W|X: 7>') # # test with open flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 FUTURE = 31 Perm._boundary_ = aenum.CONFORM self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '<Perm.R: 4>') self.assertEqual(repr(Perm(8)), '<Perm: 0>') self.assertEqual(repr(~(Perm.R | 8)), '<Perm.W|X: 3>') self.assertEqual(repr(Perm(~8)), '<Perm.R|W|X: 7>') def test_repr_eject(self): # test with complete flag class Perm(IntFlag): _order_ = 'R W X' _boundary_ = EJECT R = 1 << 2 W = 1 << 1 X = 1 << 0 self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '12') self.assertEqual(repr(Perm(8)), '8') self.assertEqual(repr(~(Perm.R | 8)), '-13') self.assertEqual(repr(Perm(~8)), '-9') # # test with open flag class Perm(IntFlag): _order_ = 'R W X' _boundary_ = EJECT R = 1 << 2 W = 1 << 1 X = 1 << 0 FUTURE = 31 self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '12') self.assertEqual(repr(Perm(8)), '8') self.assertEqual(repr(~(Perm.R | 8)), '-13') self.assertEqual(repr(Perm(~8)), '-9') def test_repr_keep(self): # test with complete flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 Perm._boundary_ = aenum.CONFORM self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '<Perm.R: 4>') self.assertEqual(repr(Perm(8)), '<Perm: 0>') self.assertEqual(repr(~(Perm.R | 8)), '<Perm.W|X: 3>') self.assertEqual(repr(Perm(~8)), '<Perm.R|W|X: 7>') # # test with open flag class Perm(IntFlag): _order_ = 'R W X' R = 1 << 2 W = 1 << 1 X = 1 << 0 FUTURE = 31 Perm._boundary_ = aenum.CONFORM self.assertEqual(repr(Perm.R), '<Perm.R: 4>') self.assertEqual(repr(Perm.W), '<Perm.W: 2>') self.assertEqual(repr(Perm.X), '<Perm.X: 1>') self.assertEqual(repr(Perm.R | Perm.W), '<Perm.R|W: 6>') self.assertEqual(repr(Perm.R | Perm.W | Perm.X), '<Perm.R|W|X: 7>') self.assertEqual(repr(Perm(0)), '<Perm: 0>') self.assertEqual(repr(~Perm.R), '<Perm.W|X: 3>') self.assertEqual(repr(~Perm.W), '<Perm.R|X: 5>') self.assertEqual(repr(~Perm.X), '<Perm.R|W: 6>') self.assertEqual(repr(~(Perm.R | Perm.W)), '<Perm.X: 1>') self.assertEqual(repr(~(Perm.R | Perm.W | Perm.X)), '<Perm: 0>') self.assertEqual(repr(Perm.R | 8), '<Perm.R: 4>') self.assertEqual(repr(Perm(8)), '<Perm: 0>') self.assertEqual(repr(~(Perm.R | 8)), '<Perm.W|X: 3>') self.assertEqual(repr(Perm(~8)), '<Perm.R|W|X: 7>') def test_repr_open(self): class Open(IntFlag): "not a good flag candidate" RO = 0 WO = 1 RW = 2 AC = 3 CE = 1<<19 Open._boundary_ = aenum.STRICT self.assertEqual(repr(Open.RO), '<Open.RO: 0>') self.assertEqual(repr(Open.WO), '<Open.WO: 1>') self.assertEqual(repr(Open.AC), '<Open.AC: 3>') self.assertEqual(repr(Open.RO | Open.CE), '<Open.CE: 524288>') self.assertEqual(repr(Open.WO | Open.CE), '<Open.WO|CE: 524289>') self.assertEqual(repr(~Open.RO), '<Open.WO|RW|CE: 524291>') self.assertEqual(repr(~Open.WO), '<Open.RW|CE: 524290>') self.assertEqual(repr(~Open.AC), '<Open.CE: 524288>') self.assertEqual(repr(~(Open.RO | Open.CE)), '<Open.AC: 3>') self.assertEqual(repr(~(Open.WO | Open.CE)), '<Open.RW: 2>') with self.assertRaisesRegex(ValueError, r'-5 is not a valid Open'): repr(Open(~4)) with self.assertRaisesRegex(ValueError, r'4 is not a valid Open'): repr(Open(4)) # class Open(IntFlag): "not a good flag candidate" RO = 0 WO = 1 RW = 2 AC = 3 CE = 1<<19 Open._boundary_ = aenum.CONFORM self.assertEqual(repr(Open.RO), '<Open.RO: 0>') self.assertEqual(repr(Open.WO), '<Open.WO: 1>') self.assertEqual(repr(Open.AC), '<Open.AC: 3>') self.assertEqual(repr(Open.RO | Open.CE), '<Open.CE: 524288>') self.assertEqual(repr(Open.WO | Open.CE), '<Open.WO|CE: 524289>') self.assertEqual(repr(~Open.RO), '<Open.WO|RW|CE: 524291>') self.assertEqual(repr(~Open.WO), '<Open.RW|CE: 524290>') self.assertEqual(repr(~Open.AC), '<Open.CE: 524288>') self.assertEqual(repr(~(Open.RO | Open.CE)), '<Open.AC: 3>') self.assertEqual(repr(~(Open.WO | Open.CE)), '<Open.RW: 2>') self.assertEqual(repr(Open(~4)), '<Open.WO|RW|CE: 524291>') self.assertEqual(repr(Open(4)), '<Open.RO: 0>') # class Open(IntFlag): "not a good flag candidate" RO = 0 WO = 1 RW = 2 AC = 3 CE = 1<<19 Open._boundary_ = aenum.EJECT self.assertEqual(repr(Open.RO), '<Open.RO: 0>') self.assertEqual(repr(Open.WO), '<Open.WO: 1>') self.assertEqual(repr(Open.AC), '<Open.AC: 3>') self.assertEqual(repr(Open.RO | Open.CE), '<Open.CE: 524288>') self.assertEqual(repr(Open.WO | Open.CE), '<Open.WO|CE: 524289>') self.assertEqual(repr(~Open.RO), '<Open.WO|RW|CE: 524291>') self.assertEqual(repr(~Open.WO), '<Open.RW|CE: 524290>') self.assertEqual(repr(~Open.AC), '<Open.CE: 524288>') self.assertEqual(repr(~(Open.RO | Open.CE)), '<Open.AC: 3>') self.assertEqual(repr(~(Open.WO | Open.CE)), '<Open.RW: 2>') self.assertEqual(repr(Open(~4)), '-5') self.assertEqual(repr(Open(4)), '4') 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: self.assertEqual(~i, (~i).value) 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) def test_iter(self): Perm = self.Perm NoPerm = Perm.R ^ Perm.R RWX = Perm.R | Perm.W | Perm.X self.assertEqual(list(NoPerm), []) self.assertEqual(list(Perm.R), [Perm.R]) self.assertEqual(list(RWX), [Perm.R, Perm.W, Perm.X]) def test_programatic_function_string(self): Perm = IntFlag('Perm', 'R W X') 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): Perm = IntFlag('Perm', 'R W X', start=8) 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): Perm = IntFlag('Perm', ['R', 'W', 'X']) 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): Perm = IntFlag('Perm', (('R', 2), ('W', 8), ('X', 32))) 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): Perm = IntFlag('Perm', OrderedDict((('R', 2), ('W', 8), ('X', 32)))) 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_containment(self): 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) 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)) def test_multiple_mixin(self): class AllMixin(object): @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(object): def __str__(self): return self._name_.lower() class Color(AllMixin, IntFlag): _order_ = 'RED GREEN BLUE' 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), '4') class Color(AllMixin, StrMixin, IntFlag): _order_ = 'RED GREEN BLUE' __str__ = StrMixin.__str__ 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): _order_ = 'RED GREEN BLUE' __str__ = StrMixin.__str__ 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') @unittest.skipUnless(threading, 'Threading required for this test.') def test_unique_composite(self): # override __eq__ to be identity only class TestFlag(IntFlag): _order_ = 'one two three four five six seven eight' 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[0] = True threads = [ threading.Thread(target=cycle_enum) for _ in range(8) ] for t in threads: t.start() for t in threads: t.join() # check that only 248 members were created (8 were created originally) self.assertFalse( failed[0], 'at least one thread failed while creating composite members') self.assertEqual(256, len(seen), 'too many composite members created') def test_init_subclass(self): class MyEnum(IntEnum): def __init_subclass__(cls, **kwds): super(MyEnum, cls).__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)) 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) class TestUnique(TestCase): """2.4 doesn't allow class decorators, use function syntax.""" def test_unique_clean(self): class Clean(Enum): one = 1 two = 'dos' tres = 4.0 unique(Clean) class Cleaner(IntEnum): single = 1 double = 2 triple = 3 unique(Cleaner) def test_unique_dirty(self): try: class Dirty(Enum): __order__ = 'one two' one = 1 two = 'dos' tres = 1 unique(Dirty) except ValueError: exc = sys.exc_info()[1] message = exc.args[0] self.assertTrue('tres -> one' in message) try: class Dirtier(IntEnum): __order__ = 'single triple' single = 1 double = 1 triple = 3 turkey = 3 unique(Dirtier) except ValueError: exc = sys.exc_info()[1] message = exc.args[0] self.assertTrue('double -> single' in message) self.assertTrue('turkey -> triple' in message) 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 class TestNamedTuple(TestCase): def test_explicit_indexing(self): class Person(NamedTuple): age = 0 first = 1 last = 2 p1 = Person(17, 'John', 'Doe') p2 = Person(21, 'Jane', 'Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') self.assertEqual(p2.age, 21) self.assertEqual(p2.first, 'Jane') self.assertEqual(p2.last, 'Doe') def test_implicit_indexing(self): class Person(NamedTuple): __order__ = "age first last" age = "person's age" first = "person's first name" last = "person's last name" p1 = Person(17, 'John', 'Doe') p2 = Person(21, 'Jane', 'Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') self.assertEqual(p2.age, 21) self.assertEqual(p2.first, 'Jane') self.assertEqual(p2.last, 'Doe') def test_mixed_indexing(self): class Person(NamedTuple): __order__ = "age last cars" age = "person's age" last = 2, "person's last name" cars = "person's cars" p1 = Person(17, 'John', 'Doe', 3) p2 = Person(21, 'Jane', 'Doe', 9) self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p1[3], 3) self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p2[3], 9) self.assertEqual(p1.age, 17) self.assertEqual(p1.last, 'Doe') self.assertEqual(p1.cars, 3) self.assertEqual(p2.age, 21) self.assertEqual(p2.last, 'Doe') self.assertEqual(p2.cars, 9) def test_issubclass(self): class Person(NamedTuple): age = 0 first = 1 last = 2 self.assertTrue(issubclass(Person, NamedTuple)) self.assertTrue(issubclass(Person, tuple)) def test_isinstance(self): class Person(NamedTuple): age = 0 first = 1 last = 2 p1 = Person(17, 'John', 'Doe') self.assertTrue(isinstance(p1, Person)) self.assertTrue(isinstance(p1, NamedTuple)) self.assertTrue(isinstance(p1, tuple)) def test_explicit_indexing_after_functional_api(self): Person = NamedTuple('Person', (('age', 0), ('first', 1), ('last', 2))) p1 = Person(17, 'John', 'Doe') p2 = Person(21, 'Jane', 'Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') self.assertEqual(p2.age, 21) self.assertEqual(p2.first, 'Jane') self.assertEqual(p2.last, 'Doe') def test_implicit_indexing_after_functional_api(self): Person = NamedTuple('Person', 'age first last') p1 = Person(17, 'John', 'Doe') p2 = Person(21, 'Jane', 'Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') self.assertEqual(p2.age, 21) self.assertEqual(p2.first, 'Jane') self.assertEqual(p2.last, 'Doe') def test_mixed_indexing_after_functional_api(self): Person = NamedTuple('Person', (('age', 0), ('last', 2), ('cars', 3))) p1 = Person(17, 'John', 'Doe', 3) p2 = Person(21, 'Jane', 'Doe', 9) self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p1[3], 3) self.assertEqual(p2[0], 21) self.assertEqual(p2[1], 'Jane') self.assertEqual(p2[2], 'Doe') self.assertEqual(p2[3], 9) self.assertEqual(p1.age, 17) self.assertEqual(p1.last, 'Doe') self.assertEqual(p1.cars, 3) self.assertEqual(p2.age, 21) self.assertEqual(p2.last, 'Doe') self.assertEqual(p2.cars, 9) def test_issubclass_after_functional_api(self): Person = NamedTuple('Person', 'age first last') self.assertTrue(issubclass(Person, NamedTuple)) self.assertTrue(issubclass(Person, tuple)) def test_isinstance_after_functional_api(self): Person = NamedTuple('Person', 'age first last') p1 = Person(17, 'John', 'Doe') self.assertTrue(isinstance(p1, Person)) self.assertTrue(isinstance(p1, NamedTuple)) self.assertTrue(isinstance(p1, tuple)) def test_creation_with_all_keywords(self): Person = NamedTuple('Person', 'age first last') p1 = Person(age=17, first='John', last='Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') def test_creation_with_some_keywords(self): Person = NamedTuple('Person', 'age first last') p1 = Person(17, first='John', last='Doe') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') p1 = Person(17, last='Doe', first='John') self.assertEqual(p1[0], 17) self.assertEqual(p1[1], 'John') self.assertEqual(p1[2], 'Doe') self.assertEqual(p1.age, 17) self.assertEqual(p1.first, 'John') self.assertEqual(p1.last, 'Doe') def test_custom_new(self): class Book(NamedTuple): title = 0 author = 1 genre = 2 def __new__(cls, string): args = [s.strip() for s in string.split(';')] return super(Book, cls).__new__(cls, *tuple(args)) b1 = Book('The Last Mohican; John Doe; Historical') self.assertEqual(b1.title, 'The Last Mohican') self.assertEqual(b1.author, 'John Doe') self.assertEqual(b1.genre, 'Historical') def test_defaults_in_class(self): class Character(NamedTuple): name = 0 gender = 1, None, 'male' klass = 2, None, 'fighter' for char in ( {'name':'John Doe'}, {'name':'William Pickney', 'klass':'scholar'}, {'name':'Sarah Doughtery', 'gender':'female'}, {'name':'Sissy Moonbeam', 'gender':'female', 'klass':'sorceress'}, ): c = Character(**char) for name, value in (('name', None), ('gender','male'), ('klass','fighter')): if name in char: value = char[name] self.assertEqual(getattr(c, name), value) def test_defaults_in_class_that_are_falsey(self): class Point(NamedTuple): x = 0, 'horizondal coordinate', 0 y = 1, 'vertical coordinate', 0 p = Point() self.assertEqual(p.x, 0) self.assertEqual(p.y, 0) def test_pickle_namedtuple_with_module(self): if isinstance(LifeForm, Exception): raise LifeForm lf = LifeForm('this', 'that', 'theother') test_pickle_dump_load(self.assertEqual, lf) def test_pickle_namedtuple_without_module(self): if isinstance(DeathForm, Exception): raise DeathForm df = DeathForm('sickly green', '2x4', 'foul') test_pickle_dump_load(self.assertEqual, df) def test_subclassing(self): if isinstance(ThatsIt, Exception): raise ThatsIt ti = ThatsIt('Henry', 'Weinhardt') self.assertEqual(ti.blah, 'Henry') self.assertTrue(ti.what(), 'Henry') test_pickle_dump_load(self.assertEqual, ti) def test_contains(self): Book = NamedTuple('Book', 'title author genre') b = Book('Teckla', 'Steven Brust', 'fantasy') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertTrue('fantasy' in b) def test_fixed_size(self): class Book(NamedTuple): _size_ = TupleSize.fixed title = 0 author = 1 genre = 2 b = Book('Teckla', 'Steven Brust', 'fantasy') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertTrue('fantasy' in b) self.assertEqual(b.title, 'Teckla') self.assertEqual(b.author, 'Steven Brust') self.assertRaises(TypeError, Book, 'Teckla', 'Steven Brust') self.assertRaises(TypeError, Book, 'Teckla') def test_minimum_size(self): class Book(NamedTuple): _size_ = TupleSize.minimum title = 0 author = 1 b = Book('Teckla', 'Steven Brust', 'fantasy') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertTrue('fantasy' in b) self.assertEqual(b.title, 'Teckla') self.assertEqual(b.author, 'Steven Brust') b = Book('Teckla', 'Steven Brust') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertEqual(b.title, 'Teckla') self.assertEqual(b.author, 'Steven Brust') self.assertRaises(TypeError, Book, 'Teckla') def test_variable_size(self): class Book(NamedTuple): _size_ = TupleSize.variable title = 0 author = 1 genre = 2 b = Book('Teckla', 'Steven Brust', 'fantasy') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertTrue('fantasy' in b) self.assertEqual(b.title, 'Teckla') self.assertEqual(b.author, 'Steven Brust') self.assertEqual(b.genre, 'fantasy') b = Book('Teckla', 'Steven Brust') self.assertTrue('Teckla' in b) self.assertTrue('Steven Brust' in b) self.assertEqual(b.title, 'Teckla') self.assertEqual(b.author, 'Steven Brust') self.assertRaises(AttributeError, getattr, b, 'genre') self.assertRaises(TypeError, Book, title='Teckla', genre='fantasy') self.assertRaises(TypeError, Book, author='Steven Brust') def test_combining_namedtuples(self): class Point(NamedTuple): x = 0, 'horizontal coordinate', 1 y = 1, 'vertical coordinate', -1 class Color(NamedTuple): r = 0, 'red component', 11 g = 1, 'green component', 29 b = 2, 'blue component', 37 Pixel1 = NamedTuple('Pixel', Point+Color, module=__name__) class Pixel2(Point, Color): "a colored dot" class Pixel3(Point): r = 2, 'red component', 11 g = 3, 'green component', 29 b = 4, 'blue component', 37 self.assertEqual(Pixel1._fields_, 'x y r g b'.split()) self.assertEqual(Pixel1.x.__doc__, 'horizontal coordinate') self.assertEqual(Pixel1.x.default, 1) self.assertEqual(Pixel1.y.__doc__, 'vertical coordinate') self.assertEqual(Pixel1.y.default, -1) self.assertEqual(Pixel1.r.__doc__, 'red component') self.assertEqual(Pixel1.r.default, 11) self.assertEqual(Pixel1.g.__doc__, 'green component') self.assertEqual(Pixel1.g.default, 29) self.assertEqual(Pixel1.b.__doc__, 'blue component') self.assertEqual(Pixel1.b.default, 37) self.assertEqual(Pixel2._fields_, 'x y r g b'.split()) self.assertEqual(Pixel2.x.__doc__, 'horizontal coordinate') self.assertEqual(Pixel2.x.default, 1) self.assertEqual(Pixel2.y.__doc__, 'vertical coordinate') self.assertEqual(Pixel2.y.default, -1) self.assertEqual(Pixel2.r.__doc__, 'red component') self.assertEqual(Pixel2.r.default, 11) self.assertEqual(Pixel2.g.__doc__, 'green component') self.assertEqual(Pixel2.g.default, 29) self.assertEqual(Pixel2.b.__doc__, 'blue component') self.assertEqual(Pixel2.b.default, 37) self.assertEqual(Pixel3._fields_, 'x y r g b'.split()) self.assertEqual(Pixel3.x.__doc__, 'horizontal coordinate') self.assertEqual(Pixel3.x.default, 1) self.assertEqual(Pixel3.y.__doc__, 'vertical coordinate') self.assertEqual(Pixel3.y.default, -1) self.assertEqual(Pixel3.r.__doc__, 'red component') self.assertEqual(Pixel3.r.default, 11) self.assertEqual(Pixel3.g.__doc__, 'green component') self.assertEqual(Pixel3.g.default, 29) self.assertEqual(Pixel3.b.__doc__, 'blue component') self.assertEqual(Pixel3.b.default, 37) def test_inherit_from_existing(self): class OERecTuple(NamedTuple): # @classmethod def record_key(cls, rec): return rec[1].upper() # @classmethod def _review_(cls, args): # set the last two args: _key and _primary args[-1] = cls.record_key(args) # OERec = OERecTuple('OERec', ['id', 'xmlid', 'key']) oe_1 = OERec(1, 'abc123') oe_2 = OERec(2, 'xyz789') self.assertEqual(oe_1, (1, 'abc123', 'ABC123')) self.assertEqual(len(oe_1), 3) self.assertEqual(oe_1.key, 'ABC123') def test_function_api_type(self): class Tester(NamedTuple): def howdy(self): return 'backwards', list(reversed(self)) Testee = NamedTuple('Testee', 'a c e', type=Tester) t = Testee(1, 2, 3) self.assertEqual(t.howdy(), ('backwards', [3, 2, 1])) def test_asdict(self): class Point(NamedTuple): x = 0, 'horizontal coordinate', 1 y = 1, 'vertical coordinate', -1 class Color(NamedTuple): r = 0, 'red component', 11 g = 1, 'green component', 29 b = 2, 'blue component', 37 Pixel = NamedTuple('Pixel', Point+Color, module=__name__) pixel = Pixel(99, -101, 255, 128, 0) self.assertEqual(pixel._asdict(), {'x':99, 'y':-101, 'r':255, 'g':128, 'b':0}) def test_make(self): class Point(NamedTuple): x = 0, 'horizontal coordinate', 1 y = 1, 'vertical coordinate', -1 self.assertEqual(Point(4, 5), (4, 5)) self.assertEqual(Point._make((4, 5)), (4, 5)) def test_replace(self): class Color(NamedTuple): r = 0, 'red component', 11 g = 1, 'green component', 29 b = 2, 'blue component', 37 purple = Color(127, 0, 127) mid_gray = purple._replace(g=127) self.assertEqual(mid_gray, (127, 127, 127)) class TestNamedConstant(TestCase): def test_constantness(self): class K(NamedConstant): PI = 3.141596 TAU = 2 * PI self.assertEqual(K.PI, 3.141596) self.assertEqual(K.TAU, 2 * K.PI) with self.assertRaisesRegex(AttributeError, r'cannot rebind constant'): K.PI = 9 with self.assertRaisesRegex(AttributeError, r'cannot delete constant'): del K.PI with self.assertRaisesRegex(AttributeError, r'cannot rebind constant'): K('PI', 3) self.assertTrue(K.PI in K) self.assertTrue(K.TAU in K) def test_duplicates(self): class CardNumber(NamedConstant): ACE = 11 TWO = 2 THREE = 3 FOUR = 4 FIVE = 5 SIX = 6 SEVEN = 7 EIGHT = 8 NINE = 9 TEN = 10 JACK = 10 QUEEN = 10 KING = 10 self.assertFalse(CardNumber.TEN is CardNumber.JACK) self.assertEqual(CardNumber.TEN, CardNumber.JACK) self.assertEqual(CardNumber.TEN, 10) def test_extend_constants(self): class CardSuit(NamedConstant): HEARTS = 1 SPADES = 2 DIAMONTS = 3 CLUBS = 4 self.assertEqual(CardSuit.HEARTS, 1) stars = CardSuit('STARS', 5) self.assertIs(stars, CardSuit.STARS) self.assertEqual(CardSuit.STARS, 5) self.assertTrue(CardSuit.STARS in CardSuit) def test_constant_with_docstring(self): class Stuff(NamedConstant): Artifact = constant(7, "lucky number!") Bowling = 11 HillWomp = constant(29, 'blah blah') self.assertEqual(Stuff.Artifact, 7) self.assertEqual(Stuff.Artifact.__doc__, 'lucky number!') self.assertEqual(Stuff.Bowling, 11) self.assertEqual(Stuff.Bowling.__doc__, None) self.assertEqual(Stuff.HillWomp, 29) self.assertEqual(Stuff.HillWomp.__doc__, 'blah blah') def test_deep_copy(self): import copy class APITypes(aenum.Constant): STRING = "string" INT = "int" APITypes('string') d = {"first": APITypes.STRING} copy.deepcopy(d) self.assertTrue(d['first'] is APITypes.STRING) def test_subclass_w_same_value(self): class Foo(aenum.Constant): BLA = 'bla1' ABA = 'aba1' class Bar(aenum.Constant): BLA = Foo.BLA ABA = 'aba2' self.assertEqual(Foo.BLA, Bar.BLA) self.assertFalse(Foo.BLA is Bar.BLA) class TestStarImport(TestCase): def test_all_exports_names(self): scope = {} exec('from aenum import *', scope, scope) self.assertIn('Enum', scope) class TestStackoverflowAnswers(TestCase): def test_self_referential_directions(self): # https://stackoverflow.com/a/64000706/208880 class Directions(Enum): _order_ = 'NORTH WEST SOUTH EAST' # NORTH = 1, 0 WEST = 0, 1 SOUTH = -1, 0 EAST = 0, -1 # def __init__(self, x, y): self.x = x self.y = y if len(self.__class__): # make links all = list(self.__class__) left, right = all[0], all[-1] self.left = left self.right = right left.right = self right.left = self # D = Directions self.assertEqual(D.NORTH.value, (1, 0)) self.assertTrue(D.NORTH.left is D.WEST) self.assertTrue(D.SOUTH.right is D.WEST) def test_self_referential_rock_paper_scissors(self): # https://stackoverflow.com/a/57085357/208880 class RPS(Enum): _order_ = 'Rock, Paper, Scissors' # Rock = "rock" Paper = "paper" Scissors = "scissors" # def __init__(self, value): if len(self.__class__): # make links all = list(self.__class__) first, previous = all[0], all[-1] first.beats = self self.beats = previous # self.assertTrue(RPS.Rock.beats is RPS.Scissors) self.assertTrue(RPS.Scissors.beats is RPS.Paper) self.assertTrue(RPS.Paper.beats is RPS.Rock) def test_arduino_headers(self): # https://stackoverflow.com/q/65048495/208880 class CHeader(Enum): def __init_subclass__(cls, **kwds): # write Enums to C header file cls_name = cls.__name__ header_path = getattr(cls, '_%s__header' % cls_name) with open(header_path, 'w') as fh: fh.write('initial header stuff here\n') for enum in cls: fh.write('#define %s %r\n' % (enum.name, enum.value)) class Arduino(CHeader): _order_ = 'ONE TWO' __header = os.path.join(tempdir, 'arduino.h') ONE = 1 TWO = 2 with open(os.path.join(tempdir, 'arduino.h')) as fh: data = fh.read() self.assertEqual(textwrap.dedent("""\ initial header stuff here #define ONE 1 #define TWO 2 """), data, ) def test_lowercase_compare(self): # https://stackoverflow.com/q/65139026/208880 class CompareLowerCase(Enum): def __init_subclass__(cls, **kwds): super(CompareLowerCase, cls).__init_subclass__(**kwds) cls.lowered_names = set([m.name.lower() for m in cls]) @classmethod def has_name(cls, name): return name.lower() in cls.lowered_names # class LabelEnum(CompareLowerCase, StrEnum): ENUM_ONE = "Enum One" ENUM_TWO = "Enum Two" ENUM_THREE = "Enum Three" FOUR = "FOUR" FIVE = "FIVE" SIX = "SIX" # self.assertTrue(LabelEnum.has_name('Enum_Three')) class TestExtendEnum(TestCase): def test_extend_enum_plain(self): class Color(Enum): red = 1 green = 2 blue = 3 self.assertRaisesRegex(TypeError, 'already in use as', extend_enum, Color, 'blue', 5) # extend_enum(Color, 'brown', 4) self.assertEqual(Color.brown.name, 'brown') self.assertEqual(Color.brown.value, 4) self.assertTrue(Color.brown in Color) self.assertEqual(Color(4), Color.brown) self.assertEqual(Color['brown'], Color.brown) self.assertEqual(len(Color), 4) # extend_enum(Color, 'mauve') self.assertEqual(Color.mauve.name, 'mauve') self.assertEqual(Color.mauve.value, 5) self.assertTrue(Color.mauve in Color) self.assertEqual(Color(5), Color.mauve) self.assertEqual(Color['mauve'], Color.mauve) self.assertEqual(len(Color), 5) def test_extend_enum_alias(self): class Color(Enum): red = 1 green = 2 blue = 3 extend_enum(Color, 'rojo', 1) self.assertEqual(Color.rojo.name, 'red') self.assertEqual(Color.rojo.value, 1) self.assertTrue(Color.rojo in Color) self.assertEqual(Color(1), Color.rojo) self.assertEqual(Color['rojo'], Color.red) self.assertEqual(len(Color), 3) def test_extend_enum_unique(self): class Color(UniqueEnum): red = 1 green = 2 blue = 3 self.assertRaisesRegex(ValueError, r'<Color.rojo: 1> is a duplicate of <Color.red: 1>', extend_enum, Color, 'rojo', 1) # self.assertEqual(Color.red.name, 'red') self.assertEqual(Color.red.value, 1) self.assertTrue(Color.red in Color) self.assertEqual(Color(1), Color.red) self.assertEqual(Color['red'], Color.red) self.assertEqual(Color.green.name, 'green') self.assertEqual(Color.green.value, 2) self.assertTrue(Color.green in Color) self.assertEqual(Color(2), Color.green) self.assertEqual(Color['blue'], Color.blue) self.assertEqual(Color.blue.name, 'blue') self.assertEqual(Color.blue.value, 3) self.assertTrue(Color.blue in Color) self.assertEqual(Color(3), Color.blue) self.assertEqual(len(Color), 3) # extend_enum(Color, 'brown', 4) self.assertEqual(Color.brown.name, 'brown') self.assertEqual(Color.brown.value, 4) self.assertTrue(Color.brown in Color) self.assertEqual(Color(4), Color.brown) self.assertEqual(Color['brown'], Color.brown) self.assertEqual(len(Color), 4) # self.assertRaisesRegex(ValueError, '', extend_enum, Color, 'verde', 2) # extend_enum(Color, 'mauve') self.assertEqual(Color.mauve.name, 'mauve') self.assertEqual(Color.mauve.value, 5) self.assertTrue(Color.mauve in Color) self.assertEqual(Color(5), Color.mauve) self.assertEqual(Color['mauve'], Color.mauve) self.assertEqual(len(Color), 5) def test_extend_enum_shadow_property(self): class Color(Enum): red = 1 green = 2 blue = 3 extend_enum(Color, 'value', 4) self.assertEqual(Color.value.name, 'value') self.assertEqual(Color.value.value, 4) self.assertTrue(Color.value in Color) self.assertEqual(Color(4), Color.value) self.assertEqual(Color['value'], Color.value) self.assertEqual(len(Color), 4) self.assertEqual(Color.red.value, 1) def test_extend_enum_shadow_base(self): class hohum(object): def cyan(self): "cyanize a color" return self.value class Color(hohum, Enum): red = 1 green = 2 blue = 3 self.assertRaisesRegex(TypeError, r'already in use in superclass', extend_enum, Color, 'cyan', 4) self.assertEqual(len(Color), 3) self.assertEqual(list(Color), [Color.red, Color.green, Color.blue]) def test_extend_enum_multivalue(self): class Color(MultiValueEnum): red = 1, 4, 7 green = 2, 5, 8 blue = 3, 6, 9 extend_enum(Color, 'brown', 10, 20) self.assertEqual(Color.brown.name, 'brown') self.assertEqual(Color.brown.value, 10) self.assertTrue(Color.brown in Color) self.assertEqual(Color(10), Color.brown) self.assertEqual(Color(20), Color.brown) self.assertEqual(Color['brown'], Color.brown) self.assertEqual(len(Color), 4) # self.assertRaisesRegex(ValueError, 'no values specified for MultiValue enum', extend_enum, Color, 'mauve') def test_extend_enum_multivalue_alias(self): class Color(MultiValueEnum): red = 1, 4, 7 green = 2, 5, 8 blue = 3, 6, 9 self.assertRaisesRegex(ValueError, r'<Color.rojo: 7> is a duplicate of <Color.red: 1>', extend_enum, Color, 'rojo', 7) self.assertEqual(Color.red.name, 'red') self.assertEqual(Color.red.value, 1) self.assertTrue(Color.red in Color) self.assertEqual(Color(1), Color.red) self.assertEqual(Color(4), Color.red) self.assertEqual(Color(7), Color.red) self.assertEqual(Color['red'], Color.red) self.assertEqual(Color.green.name, 'green') self.assertEqual(Color.green.value, 2) self.assertTrue(Color.green in Color) self.assertEqual(Color(2), Color.green) self.assertEqual(Color(5), Color.green) self.assertEqual(Color(8), Color.green) self.assertEqual(Color['blue'], Color.blue) self.assertEqual(Color.blue.name, 'blue') self.assertEqual(Color.blue.value, 3) self.assertTrue(Color.blue in Color) self.assertEqual(Color(3), Color.blue) self.assertEqual(Color(6), Color.blue) self.assertEqual(Color(9), Color.blue) self.assertEqual(len(Color), 3) def test_extend_enum_multivalue_str(self): class M(str, MultiValueEnum): VALUE_1 = 'value_1', 'VALUE_1' VALUE_2 = 'value_2', 'VALUE_2' VALUE_3 = 'value_3', 'VALUE_3' self.assertTrue(M._member_type_ is str) extend_enum(M, 'VALUE_4', 'value_4', 'VALUE_4') self.assertEqual(list(M), [M.VALUE_1, M.VALUE_2, M.VALUE_3, M.VALUE_4]) self.assertTrue(M('value_4') is M.VALUE_4) self.assertTrue(M('VALUE_4') is M.VALUE_4) self.assertTrue(M.VALUE_4.name == 'VALUE_4') self.assertTrue(M.VALUE_4.value == 'value_4') def test_extend_intenum(self): class Index(IntEnum): DeviceType = 0x1000 ErrorRegister = 0x1001 for name, value in ( ('ControlWord', 0x6040), ('StatusWord', 0x6041), ('OperationMode', 0x6060), ): extend_enum(Index, name, value) self.assertEqual(len(Index), 5) self.assertEqual(list(Index), [Index.DeviceType, Index.ErrorRegister, Index.ControlWord, Index.StatusWord, Index.OperationMode]) self.assertEqual(Index.DeviceType.value, 0x1000) self.assertEqual(Index.StatusWord.value, 0x6041) def test_extend_multi_init(self): try: from http import HTTPStatus length = len(HTTPStatus) except ImportError: class HTTPStatus(IntEnum): def __new__(cls, value, phrase, description): obj = int.__new__(cls, value) obj._value_ = value obj.phrase = phrase obj.description = description return obj CONTINUE = 100, 'Continue', 'Request received, please continue' SWITCHING_PROTOCOLS = 101, 'Switching Protocols', 'Switching to new protocol; obey Upgrade header' PROCESSING = 102, 'Processing', '' length = 3 extend_enum(HTTPStatus, 'BAD_SPAM', 513, 'Too greasy', 'for a train') extend_enum(HTTPStatus, 'BAD_EGGS', 514, 'Too green', '') self.assertEqual(len(HTTPStatus), length+2) self.assertEqual( list(HTTPStatus)[-2:], [HTTPStatus.BAD_SPAM, HTTPStatus.BAD_EGGS], ) self.assertEqual(HTTPStatus.BAD_SPAM.value, 513) self.assertEqual(HTTPStatus.BAD_SPAM.name, 'BAD_SPAM') self.assertEqual(HTTPStatus.BAD_SPAM.phrase, 'Too greasy') self.assertEqual(HTTPStatus.BAD_SPAM.description, 'for a train') self.assertEqual(HTTPStatus.BAD_EGGS.value, 514) self.assertEqual(HTTPStatus.BAD_EGGS.name, 'BAD_EGGS') self.assertEqual(HTTPStatus.BAD_EGGS.phrase, 'Too green') self.assertEqual(HTTPStatus.BAD_EGGS.description, '') def test_extend_flag(self): class Color(Flag): BLACK = 0 RED = 1 GREEN = 2 BLUE = 4 extend_enum(Color, 'MAGENTA') self.assertTrue(Color(8) is Color.MAGENTA) self.assertTrue(isinstance(Color.MAGENTA, Color)) self.assertEqual(Color.MAGENTA.value, 8) extend_enum(Color, 'PURPLE', 11) self.assertTrue(Color(11) is Color.PURPLE) self.assertTrue(isinstance(Color.PURPLE, Color)) self.assertEqual(Color.PURPLE.value, 11) self.assertTrue(issubclass(Color, Flag)) def test_extend_flag_backwards(self): class Color(Flag): BLACK = 0 RED = 1 GREEN = 2 BLUE = 4 extend_enum(Color, 'PURPLE', 11) self.assertTrue(Color(11) is Color.PURPLE) self.assertTrue(isinstance(Color.PURPLE, Color)) self.assertEqual(Color.PURPLE.value, 11) self.assertTrue(issubclass(Color, Flag)) # extend_enum(Color, 'MAGENTA') self.assertTrue(Color(8) is Color.MAGENTA) self.assertTrue(isinstance(Color.MAGENTA, Color)) self.assertEqual(Color.MAGENTA.value, 8) # extend_enum(Color, 'mauve') self.assertEqual(Color.mauve.name, 'mauve') self.assertEqual(Color.mauve.value, 16) self.assertTrue(Color.mauve in Color) self.assertEqual(Color(16), Color.mauve) self.assertEqual(Color['mauve'], Color.mauve) self.assertEqual(len(Color), 5) def test_extend_intflag(self): class Color(IntFlag): BLACK = 0 RED = 1 GREEN = 2 BLUE = 4 extend_enum(Color, 'MAGENTA') self.assertTrue(Color(8) is Color.MAGENTA) self.assertTrue(isinstance(Color.MAGENTA, Color)) self.assertEqual(Color.MAGENTA.value, 8) extend_enum(Color, 'PURPLE', 11) self.assertTrue(Color(11) is Color.PURPLE) self.assertTrue(isinstance(Color.PURPLE, Color)) self.assertEqual(Color.PURPLE.value, 11) self.assertTrue(issubclass(Color, Flag)) # extend_enum(Color, 'mauve') self.assertEqual(Color.mauve.name, 'mauve') self.assertEqual(Color.mauve.value, 16) self.assertTrue(Color.mauve in Color) self.assertEqual(Color(16), Color.mauve) self.assertEqual(Color['mauve'], Color.mauve) self.assertEqual(len(Color), 5) def test_extend_intflag_backwards(self): class Color(IntFlag): BLACK = 0 RED = 1 GREEN = 2 BLUE = 4 extend_enum(Color, 'PURPLE', 11) self.assertTrue(Color(11) is Color.PURPLE) self.assertTrue(isinstance(Color.PURPLE, Color)) self.assertEqual(Color.PURPLE.value, 11) self.assertTrue(issubclass(Color, Flag)) # extend_enum(Color, 'MAGENTA') self.assertTrue(Color(8) is Color.MAGENTA) self.assertTrue(isinstance(Color.MAGENTA, Color)) self.assertEqual(Color.MAGENTA.value, 8) # extend_enum(Color, 'mauve') self.assertEqual(Color.mauve.name, 'mauve') self.assertEqual(Color.mauve.value, 16) self.assertTrue(Color.mauve in Color) self.assertEqual(Color(16), Color.mauve) self.assertEqual(Color['mauve'], Color.mauve) self.assertEqual(len(Color), 5) def test_extend_strenum(self): class Color(StrEnum): RED = auto() GREEN = auto() BLUE = auto() extend_enum(Color, 'BLACK') self.assertEqual(Color.BLACK.name, 'BLACK') self.assertEqual(Color.BLACK.value, 'black') self.assertEqual(len(Color), 4) class TestIssues(TestCase): def test_auto_multi_int(self): class Measurement(int, MultiValueEnum, AddValueEnum): _order_ = 'one two three' _start_ = 0 one = "20110721" two = "20120911" three = "20110518" self.assertEqual([m.value for m in Measurement], [0, 1, 2]) self.assertEqual([m.name for m in Measurement], ['one', 'two', 'three']) self.assertIs(Measurement('20110721'), Measurement.one) self.assertIs(Measurement(0), Measurement.one) self.assertIs(Measurement('20120911'), Measurement.two) self.assertIs(Measurement(1), Measurement.two) self.assertIs(Measurement('20110518'), Measurement.three) self.assertIs(Measurement(2), Measurement.three) def test_auto_kwds(self): class Item(Enum): _order_ = 'A B' A = auto(size=100, requirements={}) B = auto(size=200, requirements={A: 1}) # def __new__(cls, value, size, requirements): obj = object.__new__(cls) obj._value_ = value obj.size = size # fix requirements new_requirements = {} for k, v in requirements.items(): if isinstance(k, auto): k = k.enum_member new_requirements[k] = v obj.requirements = new_requirements return obj self.assertEqual((Item.A.value, Item.A.size, Item.A.requirements), (1, 100, {})) self.assertEqual((Item.B.value, Item.B.size, Item.B.requirements), (2, 200, {Item.A: 1})) def test_auto_kwds_and_gnv(self): class Item(Enum): _order_ = 'A B' # def _generate_next_value_(name, start, count, values, *args, **kwds): return name A = auto(size=100, requirements={}) B = auto(size=200, requirements={A: 1}) # def __new__(cls, value, size, requirements): obj = object.__new__(cls) obj._value_ = value obj.size = size # fix requirements new_requirements = {} for k, v in requirements.items(): if isinstance(k, auto): k = k.enum_member new_requirements[k] = v obj.requirements = new_requirements return obj self.assertEqual((Item.A.value, Item.A.size, Item.A.requirements), ('A', 100, {})) self.assertEqual((Item.B.value, Item.B.size, Item.B.requirements), ('B', 200, {Item.A: 1})) def test_extend_flag(self): class FlagTest(Flag): # Or IntFlag NONE = 0 LOW = 1 MID = 2 extend_enum(FlagTest, 'HIGH', 4) self.assertEqual(FlagTest.LOW | FlagTest.HIGH, FlagTest(5)) self.assertEqual((FlagTest.LOW | FlagTest.HIGH).value, 5) def test_extend_unhashable(self): class TestEnum(Enum): ABC = { 'id': 0, 'value': 'abc' } DEF = { 'id': 1, 'value': 'def' } rand = uuid.uuid4().hex new_value = { 'id': 99, 'value': 'new', } extend_enum(TestEnum, rand, new_value) # Test conversion of global constants # 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 CONVERT_TEST_SIGABRT = 4 # and this one CONVERT_TEST_SIGIOT = 4 CONVERT_TEST_EIO = 7 CONVERT_TEST_EBUS = 7 # and this one CONVERT_STRING_TEST_NAME_D = 5 CONVERT_STRING_TEST_NAME_C = 5 CONVERT_STRING_TEST_NAME_B = 5 CONVERT_STRING_TEST_NAME_A = 5 # This one should sort first. CONVERT_STRING_TEST_NAME_E = 5 CONVERT_STRING_TEST_NAME_F = 5 # global names for StrEnum._convert_ test CONVERT_STR_TEST_2 = 'goodbye' CONVERT_STR_TEST_1 = 'hello' # We also need values that cannot be compared: UNCOMPARABLE_A = 5 UNCOMPARABLE_C = (9, 1) # naming order is broken on purpose UNCOMPARABLE_B = 'value' COMPLEX_C = 1j COMPLEX_A = 2j COMPLEX_B = 3j class TestConvert(TestCase): def tearDown(self): # Reset the module-level test variables to their original integer # values, otherwise the already created enum values get converted # instead. g = globals() for suffix in ['A', 'B', 'C', 'D', 'E', 'F']: g['CONVERT_TEST_NAME_%s' % suffix] = 5 g['CONVERT_STRING_TEST_NAME_%s' % suffix] = 5 for suffix, value in (('A', 5), ('B', (9, 1)), ('C', 'value')): g['UNCOMPARABLE_%s' % suffix] = value for suffix, value in (('A', 2j), ('B', 3j), ('C', 1j)): g['COMPLEX_%s' % suffix] = value for suffix, value in (('1', 'hello'), ('2', 'goodbye')): g['CONVERT_STR_TEST_%s' % suffix] = value g['CONVERT_TEST_SIGABRT'] = 4 g['CONVERT_TEST_SIGIOT'] = 4 g['CONVERT_TEST_EIO'] = 7 g['CONVERT_TEST_EBUS'] = 7 def test_convert_value_lookup_priority(self): test_type = IntEnum._convert_( 'UnittestConvert', MODULE, 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') self.assertEqual(test_type(4).name, 'CONVERT_TEST_SIGABRT') self.assertEqual(test_type(7).name, 'CONVERT_TEST_EBUS') self.assertEqual( list(test_type), [ test_type.CONVERT_TEST_SIGABRT, test_type.CONVERT_TEST_NAME_A, test_type.CONVERT_TEST_EBUS, ], ) def test_convert_int(self): test_type = IntEnum._convert_( 'UnittestConvert', MODULE, 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. int_dir = dir(int) + [ 'CONVERT_TEST_NAME_A', 'CONVERT_TEST_NAME_B', 'CONVERT_TEST_NAME_C', 'CONVERT_TEST_NAME_D', 'CONVERT_TEST_NAME_E', 'CONVERT_TEST_NAME_F', 'CONVERT_TEST_SIGABRT', 'CONVERT_TEST_SIGIOT', 'CONVERT_TEST_EIO', 'CONVERT_TEST_EBUS', ] extra = [name for name in dir(test_type) if name not in enum_dir(test_type)] missing = [name for name in enum_dir(test_type) if name not in dir(test_type)] self.assertEqual( extra + missing, [], msg='extra names: %r; missing names: %r' % (extra, missing), ) @unittest.skipUnless(PY3, 'everything is comparable on Python 2') def test_convert_uncomparable(self): uncomp = Enum._convert_( 'Uncomparable', MODULE, filter=lambda x: x.startswith('UNCOMPARABLE_')) # Should be ordered by `name` only: self.assertEqual( list(uncomp), [uncomp.UNCOMPARABLE_A, uncomp.UNCOMPARABLE_B, uncomp.UNCOMPARABLE_C], list(uncomp), ) @unittest.skipUnless(PY3, 'everything is comparable on Python 2') def test_convert_complex(self): uncomp = Enum._convert_( 'Uncomparable', MODULE, filter=lambda x: x.startswith('COMPLEX_')) # Should be ordered by `name` only: self.assertEqual( list(uncomp), [uncomp.COMPLEX_A, uncomp.COMPLEX_B, uncomp.COMPLEX_C], ) def test_convert_str(self): test_type = StrEnum._convert_( 'UnittestConvert', MODULE, filter=lambda x: x.startswith('CONVERT_STR_'), as_global=True) # 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. extra = [name for name in dir(test_type) if name not in enum_dir(test_type)] missing = [name for name in enum_dir(test_type) if name not in dir(test_type)] self.assertEqual( extra + missing, [], msg='extra names: %r; missing names: %r' % (extra, missing), ) self.assertEqual(repr(test_type.CONVERT_STR_TEST_1), '%s.CONVERT_STR_TEST_1' % SHORT_MODULE) self.assertEqual(str(test_type.CONVERT_STR_TEST_2), 'goodbye') self.assertEqual(format(test_type.CONVERT_STR_TEST_1), 'hello') def test_convert_repr_and_str(self): test_type = IntEnum._convert_( 'UnittestConvert', MODULE, filter=lambda x: x.startswith('CONVERT_STRING_TEST_'), as_global=True) self.assertEqual(repr(test_type.CONVERT_STRING_TEST_NAME_A), '%s.CONVERT_STRING_TEST_NAME_A' % SHORT_MODULE) self.assertEqual(str(test_type.CONVERT_STRING_TEST_NAME_A), '5') self.assertEqual(format(test_type.CONVERT_STRING_TEST_NAME_A), '5') # helpers def enum_dir(cls): interesting = set(cls._member_names_ + [ '__class__', '__contains__', '__doc__', '__getitem__', '__iter__', '__len__', '__members__', '__module__', '__name__', ]) if cls._new_member_ is not object.__new__: interesting.add('__new__') if cls.__init_subclass__ is not Enum.__init_subclass__: interesting.add('__init_subclass__') if hasattr(object, '__qualname__'): interesting.add('__qualname__') for method in ('__init__', '__format__', '__repr__', '__str__'): if getattr(cls, method) not in (getattr(Enum, method), getattr(Flag, method)): interesting.add(method) if cls._member_type_ is object: return sorted(interesting) else: # return whatever mixed-in data type has return sorted(set(dir(cls._member_type_)) | interesting) def member_dir(member): if member.__class__._member_type_ is object: allowed = set(['__class__', '__doc__', '__eq__', '__hash__', '__module__', 'name', 'value']) else: allowed = set(dir(member)) for cls in member.__class__.mro(): for name, obj in cls.__dict__.items(): if name[0] == '_': continue if isinstance(obj, enum.property): if obj.fget is not None or name not in member._member_map_: allowed.add(name) else: allowed.discard(name) else: allowed.add(name) return sorted(allowed) if __name__ == '__main__': tempdir = tempfile.mkdtemp() test = None try: if PY3: test_v3.tempdir = tempdir test = unittest.main(exit=False) sys.stdout.flush() for name, reason in test.result.skipped: print("%s: %s" % (name, reason)) finally: shutil.rmtree(tempdir, True) if test: sys.exit(len(test.result.errors or test.result.failures) and 1 or 0)