Skip to content

Commit 741b300

Browse files
committed
Add some extra test cases.
This is to ensure that all `second_play_test` cases are covered.
1 parent 294b27c commit 741b300

File tree

2 files changed

+27
-2
lines changed

2 files changed

+27
-2
lines changed

axelrod/tests/strategies/test_lookerup.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -311,6 +311,9 @@ def test_strategy(self):
311311
actions = [(C, C), (C, D), (D, C), (C, D)]
312312
self.versus_test(axelrod.Alternator(), expected_actions=actions)
313313

314+
actions = [(C, D), (D, D), (D, D)]
315+
self.versus_test(axelrod.Defector(), expected_actions=actions)
316+
314317
def test_cooperator_table(self):
315318
lookup_table = {((), (), ()): C}
316319
actions = [(C, D)] * 5

axelrod/tests/strategies/test_titfortat.py

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ class TestTitFor2Tats(TestPlayer):
8181
def test_strategy(self):
8282
# Will punish sequence of 2 defections but will forgive
8383
opponent = axelrod.MockPlayer(actions=[D, D, D, C, C])
84-
actions = [(C, D), (C, D), (D, D), (D, C), (C, C)]
84+
actions = [(C, D), (C, D), (D, D), (D, C), (C, C), (C, D)]
8585
self.versus_test(opponent, expected_actions=actions)
8686

8787

@@ -104,6 +104,14 @@ def test_strategy(self):
104104
actions = [(C, D), (D, C), (D, C), (C, D), (D, C)]
105105
self.versus_test(opponent, expected_actions=actions)
106106

107+
actions = [(C, C), (C, C)]
108+
self.versus_test(opponent=axelrod.Cooperator(),
109+
expected_actions=actions)
110+
111+
actions = [(C, D), (D, D), (D, D)]
112+
self.versus_test(opponent=axelrod.Defector(),
113+
expected_actions=actions)
114+
107115

108116
class TestDynamicTwoTitsForTat(TestPlayer):
109117

@@ -201,6 +209,13 @@ def test_strategy(self):
201209
actions = [(D, C), (C, D)] * 8
202210
self.versus_test(axelrod.TitForTat(), expected_actions=actions)
203211

212+
actions = [(D, C), (C, C), (C, C)]
213+
self.versus_test(axelrod.Cooperator(),
214+
expected_actions=actions)
215+
216+
actions = [(D, D), (D, D), (D, D)]
217+
self.versus_test(axelrod.Defector(), expected_actions=actions)
218+
204219

205220
class TestAntiTitForTat(TestPlayer):
206221

@@ -238,6 +253,13 @@ def test_strategy(self):
238253
actions = [(C, D), (D, C), (D, C), (D, C), (C, D), (D, C)]
239254
self.versus_test(opponent, expected_actions=actions)
240255

256+
actions = [(C, C), (C, C), (C, C)]
257+
self.versus_test(axelrod.Cooperator(),
258+
expected_actions=actions)
259+
260+
actions = [(C, D), (D, D), (D, D)]
261+
self.versus_test(axelrod.Defector(), expected_actions=actions)
262+
241263

242264
class TestHardTitFor2Tats(TestPlayer):
243265

@@ -513,7 +535,7 @@ def test_strategy(self):
513535
attrs={"retaliating": False})
514536

515537
opponent = axelrod.MockPlayer(actions=[C, C, C, C, D, C])
516-
actions = [(C, C)] * 4 + [(C, D), (D, C)]
538+
actions = [(C, C)] * 4 + [(C, D), (D, C), (C, C)]
517539
self.versus_test(opponent, expected_actions=actions,
518540
attrs={"retaliating": False})
519541

0 commit comments

Comments
 (0)