Adding a reset method to FSM.

The test checks that the reset method changes the value of the state.

Author: drvinceknight

axelrod/strategies/finite_state_machines.py

@@ -54,6 +54,7 @@ def __init__(self, transitions=None, initial_state=None, initial_action=None):
             initial_state = 1
             initial_action = C
         Player.__init__(self)
+        self.initial_state = initial_state
         self.initial_action = initial_action
         self.fsm = SimpleFSM(transitions, initial_state)
 
@@ -67,6 +68,10 @@ def strategy(self, opponent):
             self.state = self.fsm.state
             return action
 
+    def reset(self):
+        Player.reset(self)
+        self.fsm.state = self.initial_state
+
 
 class Fortress3(FSMPlayer):
     """Finite state machine player specified in DOI:10.1109/CEC.2006.1688322.

axelrod/tests/unit/test_finite_state_machines.py

@@ -111,6 +111,12 @@ def test_transitions(self):
         fsm = player.fsm
         self.assertTrue(check_state_transitions(fsm.state_transitions))
 
+    def test_reset_initial_state(self):
+        player = self.player()
+        player.fsm.state = -1
+        player.reset()
+        self.assertFalse(player.fsm.state == -1)
+
 
 class TestFortress3(TestFSMPlayer):