Add some extra test cases.

This is to ensure that all `second_play_test` cases are covered.
Axelrod-Python · Jun 3, 2017 · 741b300 · 741b300
1 parent 294b27c
commit 741b300
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/axelrod/tests/strategies/test_lookerup.py b/axelrod/tests/strategies/test_lookerup.py
@@ -311,6 +311,9 @@ def test_strategy(self):
         actions = [(C, C), (C, D), (D, C), (C, D)]
         self.versus_test(axelrod.Alternator(), expected_actions=actions)
 
+        actions = [(C, D), (D, D), (D, D)]
+        self.versus_test(axelrod.Defector(), expected_actions=actions)
+
     def test_cooperator_table(self):
         lookup_table = {((), (), ()): C}
         actions = [(C, D)] * 5

diff --git a/axelrod/tests/strategies/test_titfortat.py b/axelrod/tests/strategies/test_titfortat.py
@@ -81,7 +81,7 @@ class TestTitFor2Tats(TestPlayer):
     def test_strategy(self):
         # Will punish sequence of 2 defections but will forgive
         opponent = axelrod.MockPlayer(actions=[D, D, D, C, C])
-        actions = [(C, D), (C, D), (D, D), (D, C), (C, C)]
+        actions = [(C, D), (C, D), (D, D), (D, C), (C, C), (C, D)]
         self.versus_test(opponent, expected_actions=actions)
 
 
@@ -104,6 +104,14 @@ def test_strategy(self):
         actions = [(C, D), (D, C), (D, C), (C, D), (D, C)]
         self.versus_test(opponent, expected_actions=actions)
 
+        actions = [(C, C), (C, C)]
+        self.versus_test(opponent=axelrod.Cooperator(),
+                         expected_actions=actions)
+
+        actions = [(C, D), (D, D), (D, D)]
+        self.versus_test(opponent=axelrod.Defector(),
+                         expected_actions=actions)
+
 
 class TestDynamicTwoTitsForTat(TestPlayer):
 
@@ -201,6 +209,13 @@ def test_strategy(self):
         actions = [(D, C), (C, D)] * 8
         self.versus_test(axelrod.TitForTat(), expected_actions=actions)
 
+        actions = [(D, C), (C, C), (C, C)]
+        self.versus_test(axelrod.Cooperator(),
+                         expected_actions=actions)
+
+        actions = [(D, D), (D, D), (D, D)]
+        self.versus_test(axelrod.Defector(), expected_actions=actions)
+
 
 class TestAntiTitForTat(TestPlayer):
 
@@ -238,6 +253,13 @@ def test_strategy(self):
         actions = [(C, D), (D, C), (D, C), (D, C), (C, D), (D, C)]
         self.versus_test(opponent, expected_actions=actions)
 
+        actions = [(C, C), (C, C), (C, C)]
+        self.versus_test(axelrod.Cooperator(),
+                         expected_actions=actions)
+
+        actions = [(C, D), (D, D), (D, D)]
+        self.versus_test(axelrod.Defector(), expected_actions=actions)
+
 
 class TestHardTitFor2Tats(TestPlayer):
 
@@ -513,7 +535,7 @@ def test_strategy(self):
                          attrs={"retaliating": False})
 
         opponent = axelrod.MockPlayer(actions=[C, C, C, C, D, C])
-        actions = [(C, C)] * 4 + [(C, D), (D, C)]
+        actions = [(C, C)] * 4 + [(C, D), (D, C), (C, C)]
         self.versus_test(opponent, expected_actions=actions,
                          attrs={"retaliating": False})