def test_value_enum_only(self):
venum = TypedEnum(abc)
venum.value = 'a'
venum.value = 1
venum.value = abc.a
with self.assertRaises(ValueError):
venum.value = 6
def test_setting_enum(self):
'''enum should be able to be set with another enum object'''
venum = TypedEnum(abc)
venum2 = TypedEnum(abc)
venum2.__set__(None, 'b')
venum.__set__(None, venum2)
self.assertEqual(venum.value, venum2.value)
self.assertEqual(venum.value, 1)
def test_descriptor(self):
venum = TypedEnum(abc)
venum.__set__(None, 1)
self.assertEqual(venum.__get__(None, None), 'b')
def test_setting(self):
venum = TypedEnum(abc)
venum.value = 1
self.assertEqual(venum.name, 'b')