29 lines
893 B
Python
29 lines
893 B
Python
from csdl.enum import CEnum, CEnumSymbol
|
|
import nose.tools
|
|
|
|
class TestEnum(CEnum):
|
|
FIRST = 0
|
|
SPECIAL = 2, 'a special case'
|
|
BIG = 9001
|
|
|
|
def test_enumsymbol():
|
|
assert isinstance(TestEnum.FIRST, int)
|
|
assert TestEnum.FIRST == 0
|
|
assert TestEnum.SPECIAL == 2
|
|
assert TestEnum.FIRST.name == 'FIRST'
|
|
assert TestEnum.SPECIAL.name == 'SPECIAL'
|
|
assert TestEnum.FIRST.description == 'FIRST'
|
|
assert TestEnum.SPECIAL.description == 'a special case'
|
|
assert repr(TestEnum.FIRST) == '<FIRST>'
|
|
assert repr(TestEnum.SPECIAL) == '<SPECIAL>'
|
|
|
|
def test_cenum():
|
|
assert len(TestEnum.values()) == 3
|
|
assert len(list(iter(TestEnum))) == 3
|
|
|
|
def test_cenum_from_int():
|
|
big_hopefully = TestEnum.from_int(9001)
|
|
assert isinstance(big_hopefully, CEnumSymbol)
|
|
assert big_hopefully.name == 'BIG'
|
|
nose.tools.assert_raises(ValueError, TestEnum.from_int, 50)
|