Added tests for grumpy strategy

axelrod/tests/test_grumpy.py

@@ -0,0 +1,47 @@
+"""
+Test for the grudger strategy
+"""
+import unittest
+import axelrod
+
+class TestGrumpy(unittest.TestCase):
+
+    def test_initial_nice_strategy(self):
+        """
+        Starts by cooperating
+        """
+        P1 = axelrod.Grumpy()
+        P2 = axelrod.Player()
+        self.assertEqual(P1.strategy(P2), 'C')
+
+    def test_initial_grumpy_strategy(self):
+        """
+        Starts by defecting if grumpy
+        """
+        P1 = axelrod.Grumpy(starting_state = 'Grumpy')
+        P2 = axelrod.Player()
+        self.assertEqual(P1.strategy(P2), 'D')
+
+    def test_effect_of_strategy(self):
+        """
+        Tests that grumpy will play c until threshold is ht at which point it will become grumpy.
+        Player will then not become nice until lower nice threshold is hit.
+        """
+        P1 = axelrod.Grumpy(grumpy_threshold = 3, nice_threshold=0)
+        P2 = axelrod.Player()
+        P1.history = ['C', 'D', 'D', 'D']
+        P2.history = ['C', 'C', 'C', 'C']
+        self.assertEqual(P1.strategy(P2), 'C')
+        P1.history = ['C', 'C', 'D', 'D', 'D']
+        P2.history = ['D', 'D', 'D', 'D', 'D']
+        self.assertEqual(P1.strategy(P2), 'D')
+        P1.history = ['C', 'C', 'D', 'D', 'D', 'D', 'D', 'D']
+        P2.history = ['D', 'D', 'D', 'D', 'D', 'C', 'C', 'C']
+        self.assertEqual(P1.strategy(P2), 'D')
+        P1.history = ['C', 'C', 'D', 'D', 'D', 'D', 'D', 'D', 'D', 'D', 'D']
+        P2.history = ['D', 'D', 'D', 'D', 'D', 'C', 'C', 'C', 'C', 'C', 'C']
+        self.assertEqual(P1.strategy(P2), 'C')
+
+    def test_representation(self):
+        P1 = axelrod.Grumpy()
+        self.assertEqual(str(P1), 'Grumpy')