Added strategies Desperate, Willing, Hopeless, Grim

Adding tests Desperate, Hopeless, Willing and Grim in mutual.py

Author: rnjd

axelrod/strategies/_strategies.py

@@ -43,6 +43,7 @@
     ZDGen2, ZDGTFT2, ZDSet2, WinStayLoseShift, WinShiftLoseStay)
 from .mindcontrol import MindController, MindWarper, MindBender
 from .mindreader import MindReader, ProtectedMindReader, MirrorMindReader
+from .mutual import Desperate, Hopeless, Willing
 from .oncebitten import OnceBitten, FoolMeOnce, ForgetfulFoolMeOnce, FoolMeForever
 from .prober import (Prober, Prober2, Prober3, HardProber,
                      NaiveProber, RemorsefulProber)
@@ -92,6 +93,7 @@
     Davis,
     Defector,
     DefectorHunter,
+    Desperate,
     DoubleCrosser,
     Eatherley,
     EventualCycleHunter,
@@ -115,21 +117,22 @@
     GoByMajority20,
     GoByMajority40,
     GoByMajority5,
+    Golden,
+    Gradual,
+    Grofman,
+    Grudger,
+    Grumpy,
     Handshake,
     HardGoByMajority,
     HardGoByMajority10,
     HardGoByMajority20,
     HardGoByMajority40,
     HardGoByMajority5,
-    Golden,
-    Gradual,
-    Grofman,
-    Grudger,
-    Grumpy,
     HardProber,
     HardTitFor2Tats,
     HardTitForTat,
     HesitantQLearner,
+    Hopeless,
     Inverse,
     InversePunisher,
     Joss,
@@ -184,6 +187,7 @@
     TrickyDefector,
     Tullock,
     TwoTitsForTat,
+    Willing,
     WinShiftLoseStay,
     WinStayLoseShift,
     ZDExtort2,

axelrod/strategies/mutual.py

@@ -0,0 +1,60 @@
+from axelrod import Actions, Player, init_args
+
+C, D = Actions.C, Actions.D
+
+class Desperate(Player):
+    """A player that only cooperates after mutual defection"""
+    name = "Desperate"
+    classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def strategy(self, opponent):
+        if not opponent.history:
+            return D
+        if self.history[-1] == D and opponent.history[-1] == D:
+            return C
+        return D
+
+class Hopeless(Player):
+    """A player that only defects after mutual cooperation"""
+    name = "Hopeless"
+    classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def strategy(self, opponent):
+        if not opponent.history:
+            return C
+        if self.history[-1] == C and opponent.history[-1] == C:
+            return D
+        return C
+
+class Willing(Player):
+    """A player that only defects after mutual defection"""
+    name = "Willing"
+    classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def strategy(self, opponent):
+        if not opponent.history:
+            return C
+        if self.history[-1] == D and opponent.history[-1] == D:
+            return D
+        return C

axelrod/tests/unit/test_mutual.py

@@ -0,0 +1,82 @@
+"""Tests for strategies Desperate, Hopeless, Willing, and Grim"""
+import axelrod
+
+from .test_player import TestPlayer
+
+C, D = axelrod.Actions.C, axelrod.Actions.D
+
+class TestDesperate(TestPlayer):
+
+    name = "Desperate"
+    player = axelrod.Desperate
+    expected_classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def test_strategy(self):
+        """
+        Test that initial strategy defects.
+        """
+        self.first_play_test(D)
+
+    def test_responses(self):
+
+        self.responses_test([C]* 4, [D] * 4, [D])
+        self.responses_test([D, D, C], [C, C, D], [D])
+        self.responses_test([D, D, D], [C, C, D], [C])
+
+
+class TestHopeless(TestPlayer):
+
+    name = "Hopeless"
+    player = axelrod.Hopeless
+    expected_classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def test_strategy(self):
+        """
+        Test that initial strategy cooperates.
+        """
+        self.first_play_test(C)
+
+    def test_responses(self):
+
+        self.responses_test([C] * 4, [D] * 4, [C])
+        self.responses_test([D] * 5, [C] * 5, [C])
+        self.responses_test([C, D, C], [C, C, C], [D])
+
+class TestWilling(TestPlayer):
+
+    name = "Willing"
+    player = axelrod.Willing
+    expected_classifier = {
+        'memory_depth': 1,
+        'stochastic': False,
+        'makes_use_of': set(),
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def test_strategy(self):
+        """
+        Test that initial strategy cooperates.
+        """
+        self.first_play_test(C)
+
+    def test_responses(self):
+
+        self.responses_test([C] * 4, [D] * 4, [C])
+        self.responses_test([D] * 5, [C] * 5, [C])
+        self.responses_test([C, C, D], [C, C, D], [D])