Add a memory test to default FSM test.

Axelrod-Python · Jan 16, 2019 · 6763610 · 6763610
1 parent b4e6c41
commit 6763610
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/axelrod/tests/strategies/test_finite_state_machines.py b/axelrod/tests/strategies/test_finite_state_machines.py
@@ -586,6 +586,13 @@ def test_strategy(self):
             opponent, expected_actions=actions, init_kwargs={"transitions": transitions}
         )
 
+    def test_memory(self):
+        """
+        Test the memory depth using implemented algorithm
+        """
+        transitions = self.player().fsm._state_transitions
+        self.assertEqual(get_memory_from_transitions(transitions), self.expected_classifier["memory_depth"])
+
 
 class TestFortress3(TestFSMPlayer):