def test_get_transition_from_memory_when_there_is_match(self):
fsm = FSM(1)
fsm.memory = {1: [['a']], 2 : [['b', 'c'], ['d']]}
(action, next_state) = fsm.get_transition('b', '2')
self.assertEqual(next_state,2)
(action, next_state) = fsm.get_transition('d', 2)
self.assertEqual(next_state,2)
def test_get_transition_from_memory_when_there_is_a_match(self):
fsm = FSM(1)
first_fsm = self.get_test_fsm()
fsm.memory = {1: [first_fsm]}
(_, _, next_state) = fsm.get_transition('a', 1)
self.assertEqual(next_state, 1)
self.assertEqual(fsm.memory.get(1), [first_fsm])
def test_get_normal_transition_when_there_is_a_match(self):
fsm = FSM(1)
fsm.state_transitions = {('b', 1): (None, None, 2)}
first_fsm = self.get_test_fsm()
fsm.memory = {2: [first_fsm]}
(_, _, next_state) = fsm.get_transition('b', 1)
self.assertEqual(next_state, 2)
self.assertEqual(fsm.memory.get(2), [first_fsm])
def test_get_normal_transition_when_there_is_no_match_but_such_transition_exist(
self):
fsm = FSM(1)
#fsm.state_transitions = {('b', 1):(None, 2)}
first_fsm = FSM('1_1')
first_fsm.state_transitions = {
('a', '1_1'): (None, None, '1_2'),
(fsm.END_PAR_TRANSITION, '1_2'): (None, None, 2)
}
fsm.memory = {2: [first_fsm]}
(_, _, next_state) = fsm.get_transition('a', 2)
print "test_get_normal_transition_when_there_is_no_match_but_such_transition_exist:%s" % fsm.memory
self.assertEqual(fsm.memory, {2: []})
self.assertEqual(next_state, 2)