Fix memory depth bug revealed by test

Axelrod-Python · Feb 6, 2019 · 49eda27 · 49eda27
1 parent f027f93
commit 49eda27
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/axelrod/strategies/shortmem.py b/axelrod/strategies/shortmem.py
@@ -22,7 +22,7 @@ class ShortMem(Player):
 
     name = "ShortMem"
     classifier = {
-        "memory_depth": 10,
+        "memory_depth": float('inf'),
         "stochastic": False,
         "makes_use_of": set(),
         "long_run_time": False,

diff --git a/axelrod/tests/strategies/test_player.py b/axelrod/tests/strategies/test_player.py
@@ -491,6 +491,7 @@ def test_memory_depth_upper_bound(self, strategies, seed, turns):
         memory = player.classifier["memory_depth"]
         if memory < float("inf"):
             for strategy in strategies:
+                player.reset()
                 opponent = strategy()
                 self.assertTrue(
                     test_memory(

diff --git a/axelrod/tests/strategies/test_shortmem.py b/axelrod/tests/strategies/test_shortmem.py
@@ -7,7 +7,7 @@
 C, D = axelrod.Action.C, axelrod.Action.D
 
 
-class TestCooperator(TestPlayer):
+class TestShortMem(TestPlayer):
 
     name = "ShortMem"
     player = axelrod.ShortMem