def test_extend_enum_multivalue_with_duplicate(self):
with self.assertRaises(ValueError):
class Color(Enum, settings=MultiValue):
red = 1, 'rojo'
green = 2, 'verde'
blue = 3, 'azul'
extend_enum(Color, 'value', 2)
def test_extend_enum_unique_with_duplicate(self):
with self.assertRaises(ValueError):
class Color(Enum, settings=Unique):
red = 1
green = 2
blue = 3
extend_enum(Color, 'value', 1)
def test_extend_enum_plain(self):
class Color(UniqueEnum):
red = 1
green = 2
blue = 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(len(Color), 4)
def test_extend_enum_shadow(self):
class Color(UniqueEnum):
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(len(Color), 4)
self.assertEqual(Color.red.value, 1)
def test_extend_enum_noalias_with_duplicate(self):
class Color(Enum, settings=NoAlias):
red = 1
green = 2
blue = 3
extend_enum(
Color,
'value',
3,
)
self.assertRaises(TypeError, Color, 3)
self.assertFalse(Color.value is Color.blue)
self.assertTrue(Color.value.value, 3)
def test_extend_enum_generate(self):
class Foo(AutoEnum):
def _generate_next_value_(name, start, count, values, *args,
**kwds):
return name
a
b
#
extend_enum(Foo, 'c')
self.assertEqual(Foo.a.value, 'a')
self.assertEqual(Foo.b.value, 'b')
self.assertEqual(Foo.c.value, 'c')