Refactoring mutual.py using versus_test

Author: MariosZoulias

axelrod/tests/strategies/test_mutual.py

@@ -22,10 +22,37 @@ class TestDesperate(TestPlayer):
     def test_strategy(self):
         self.first_play_test(C, seed=1)
         self.first_play_test(D, seed=2)
-        self.responses_test([D], [C] * 4, [D] * 4)
-        self.responses_test([D], [D, D, C], [C, C, D])
-        self.responses_test([C], [D, D, D], [C, C, D])
 
+        #Our Player (Desperate) vs Cooperator SEED -->1
+        opponent = axelrod.Cooperator()
+        opponent_actions = [C] * 5
+        actions = [(C, C), (D, C), (D, C), (D, C), (D,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Desperate) vs Cooperator SEED -->2
+        opponent = axelrod.Cooperator()
+        actions = [(D, C), (D, C), (D, C), (D, C), (D,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Desperate) vs Defector SEED -->1
+        opponent = axelrod.Defector()
+        actions = [(C, D), (D, D), (C, D), (D, D), (C,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Desperate) vs Defector SEED -->2
+        opponent = axelrod.Defector()
+        actions = [(D, D), (C, D), (D, D), (C, D), (D,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Desperate) vs Alternator SEED -->1
+        opponent = axelrod.Alternator()
+        actions = [(C, C), (D, D), (C, C), (D, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Desperate) vs Alternator SEED -->2
+        opponent = axelrod.Alternator()
+        actions = [(D, C), (D, D), (C, C), (D, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
 
 class TestHopeless(TestPlayer):
 
@@ -44,9 +71,37 @@ class TestHopeless(TestPlayer):
     def test_strategy(self):
         self.first_play_test(C, seed=1)
         self.first_play_test(D, seed=2)
-        self.responses_test([C], [C] * 4, [D] * 4)
-        self.responses_test([C], [D] * 5, [C] * 5)
-        self.responses_test([D], [C, D, C], [C, C, C])
+
+        #Our Player (Hopeless) vs Cooperator SEED -->1
+        opponent = axelrod.Cooperator()
+        opponent_actions = [C] * 5
+        actions = [(C, C), (D, C), (C, C), (D, C), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Hopeless) vs Cooperator SEED -->2
+        opponent = axelrod.Cooperator()
+        actions = [(D, C), (C, C), (D, C), (C, C), (D,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Hopeless) vs Defector SEED -->1
+        opponent = axelrod.Defector()
+        actions = [(C, D), (C, D), (C, D), (C, D), (C,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Hopeless) vs Defector SEED -->2
+        opponent = axelrod.Defector()
+        actions = [(D, D), (C, D), (C, D), (C, D), (C,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Hopeless) vs Alternator SEED -->1
+        opponent = axelrod.Alternator()
+        actions = [(C, C), (D, D), (C, C), (D, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Hopeless) vs Alternator SEED -->2
+        opponent = axelrod.Alternator()
+        actions = [(D, C), (C, D), (C, C), (D, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
 
 
 class TestWilling(TestPlayer):
@@ -66,6 +121,34 @@ class TestWilling(TestPlayer):
     def test_strategy(self):
         self.first_play_test(C, seed=1)
         self.first_play_test(D, seed=2)
-        self.responses_test([C], [C] * 4, [D] * 4)
-        self.responses_test([C], [D] * 5, [C] * 5)
-        self.responses_test([D], [C, C, D], [C, C, D])
+
+        #Our Player (Willing) vs Cooperator SEED -->1
+        opponent = axelrod.Cooperator()
+        opponent_actions = [C] * 5
+        actions = [(C, C), (C, C), (C, C), (C, C), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Willing) vs Cooperator SEED -->2
+        opponent = axelrod.Cooperator()
+        actions = [(D, C), (C, C), (C, C), (C, C), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Willing) vs Defector SEED -->1
+        opponent = axelrod.Defector()
+        actions = [(C, D), (C, D), (C, D), (C, D), (C,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Willing) vs Defector SEED -->2
+        opponent = axelrod.Defector()
+        actions = [(D, D), (D, D), (D, D), (D, D), (D,D)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)
+
+        #Our Player (Willing) vs Alternator SEED -->1
+        opponent = axelrod.Alternator()
+        actions = [(C, C), (C, D), (C, C), (C, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=1)
+
+        #Our Player (Willing) vs Alternator SEED -->2
+        opponent = axelrod.Alternator()
+        actions = [(D, C), (C, D), (C, C), (C, D), (C,C)]
+        self.versus_test(opponent, expected_actions=actions, seed=2)