photon/csdl/tests/test_enum.py

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)