@@ -71,7 +71,7 @@ def test_various(self):
71
71
72
72
class TestSimulatePlay (unittest .TestCase ):
73
73
74
- def test_various (self ):
74
+ def test_opponent_play (self ):
75
75
p1 = TestOpponent ()
76
76
p2 = TestOpponent ()
77
77
self .assertEqual (simulate_play (p1 , p2 ), (C , C ))
@@ -81,15 +81,14 @@ def test_various(self):
81
81
self .assertEqual (p2 .defections , 0 )
82
82
83
83
# TestOpponent always returns C
84
- for h1 in [C , D ]:
85
- for h2 in [C , D ]:
86
- self .assertEqual (simulate_play (p1 , p2 , h1 , h2 ), (C , C ))
87
- self .assertEqual (p1 .cooperations , 3 )
88
- self .assertEqual (p2 .cooperations , 3 )
89
- self .assertEqual (p1 .defections , 2 )
90
- self .assertEqual (p2 .defections , 2 )
84
+ for _ in range (4 ):
85
+ self .assertEqual (simulate_play (p1 , p2 , C , C ), (C , C ))
86
+ self .assertEqual (p1 .cooperations , 5 )
87
+ self .assertEqual (p2 .cooperations , 5 )
88
+ self .assertEqual (p1 .defections , 0 )
89
+ self .assertEqual (p2 .defections , 0 )
91
90
92
- def test_various2 (self ):
91
+ def test_simulated_play (self ):
93
92
p1 = axelrod .Cooperator ()
94
93
p2 = axelrod .Defector ()
95
94
self .assertEqual (simulate_play (p1 , p2 ), (C , D ))
0 commit comments