34 lines
701 B
Python
34 lines
701 B
Python
|
# standard imports
|
||
|
import unittest
|
||
|
|
||
|
# local imports
|
||
|
from schiz import State
|
||
|
from schiz.error import (
|
||
|
StateExists,
|
||
|
StateInvalid,
|
||
|
)
|
||
|
|
||
|
|
||
|
class TestStateReport(unittest.TestCase):
|
||
|
|
||
|
def setUp(self):
|
||
|
self.states = State(4)
|
||
|
self.states.add('foo')
|
||
|
self.states.add('bar')
|
||
|
self.states.add('baz')
|
||
|
|
||
|
|
||
|
def test_list_pure(self):
|
||
|
for k in ['FOO', 'BAR', 'BAZ']:
|
||
|
getattr(self.states, k)
|
||
|
|
||
|
|
||
|
def test_list_alias(self):
|
||
|
self.states.alias('xyzzy', self.states.FOO | self.states.BAZ)
|
||
|
for k in ['FOO', 'BAR', 'BAZ', 'XYZZY']:
|
||
|
getattr(self.states, k)
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
unittest.main()
|