Adding WorseAndWorse stratergy attempt 2

Author: AdamPohl

axelrod/strategies/worse_and_worse.py

@@ -0,0 +1,35 @@
+from axelrod import Actions, Player
+from random import randint
+
+class WorseAndWorse (Player):
+    """
+    Defects with probability of 'current turn / total no. of turns'. Therefore
+    it is more and more likely to defect as the round goes on.
+
+    Names:
+        - worse_and_worse: [PRISON1998]
+
+    """
+
+    name = 'Worse and worse'
+    classifier = {
+        'memory_depth': float('inf'),
+        'stochastic': True,
+        'makes_use_of': set(['length']),
+        'long_run_time': False,
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def strategy(self, opponent):
+        current_round = len(self.history) + 1
+        expected_length = self.match_attributes['length']
+        try:
+            if randint(0, expected_length) < (current_round):
+                return Actions.D
+            return Actions.C
+        except:
+            if randint(0, 200) < (current_round):
+                return Actions.D
+            return Actions.C

axelrod/tests/unit/test_worse_and_worse.py

@@ -0,0 +1,33 @@
+"""Test for the Worse and Worse strategy."""
+
+import axelrod
+
+from .test_player import TestPlayer
+
+C, D = axelrod.Actions.C, axelrod.Actions.D
+
+
+class TestWorseAndWorse(TestPlayer):
+
+    name = "Worse and worse"
+    player = axelrod.WorseAndWorse
+    expected_classifier = {
+        'memory_depth': float('inf'),
+        'stochastic': True,
+        'makes_use_of': set(['length']),
+        'long_run_time': False,
+        'inspects_source': False,
+        'manipulates_source': False,
+        'manipulates_state': False
+    }
+
+    def test_strategy(self):
+        """
+        Test that the stratergy chooses to defect according to the correct
+        probability.
+        """
+        self.responses_test([], [], [C, C, D, D, D], random_seed=1,
+        tournament_length=5)
+
+        self.responses_test([], [], [D, D, D, D, D], random_seed=2,
+        tournament_length=5)