def test_add_fsm_to_state_when_memory_is_empty(self):
fsm = FSM(1)
nested_fsm = FSM('1_1')
fsm.add_fsm_to_memory(1, nested_fsm)
print "test_add_fsm_to_state_when_memory_is_empty: %s" % fsm.memory
self.assertEqual(fsm.memory.get(1), [nested_fsm])
def test_add_fsm_to_state_that_is_already_in_memory(self):
fsm = FSM(1)
first_fsm = self.get_test_fsm()
fsm.memory = {1: [first_fsm]}
second_fsm = self.get_test_fsm()
fsm.add_fsm_to_memory(1, second_fsm)
print "test_add_fsm_to_state_that_is_already_in_memory%s" % fsm.memory
self.assertEqual(len(fsm.memory), 1)
self.assertEqual(fsm.memory.get(1), [first_fsm, second_fsm])
def test_add_fsm_to_state_that_is_not_in_memory_and_memory_is_not_empty(
self):
fsm = FSM(1)
first_fsm = self.get_test_fsm()
fsm.memory = {1: [first_fsm]}
second_fsm = self.get_test_fsm()
fsm.add_fsm_to_memory(2, second_fsm)
print 'test_add_fsm_to_state_that_is_not_in_memory_and_memory_is_not_empty: %s' % fsm.memory
self.assertEqual(len(fsm.memory), 2)
self.assertEqual(fsm.memory.get(1), [first_fsm])
self.assertEqual(fsm.memory.get(1), [second_fsm])