Merge pull request #446 from shizejin/fix-AS-algorithm-test

mmcky · web-flow · commit 3772fddb8ea8 · 2018-12-04T15:27:06.000+11:00
FIX: test `method` keyword of `RepeatedGame.equilibrium_payoffs()`.
diff --git a/quantecon/game_theory/tests/test_repeated_game.py b/quantecon/game_theory/tests/test_repeated_game.py
@@ -30,7 +30,7 @@ def setUp(self):
         # Prisoner's dilemma
         bimatrix = [[(9, 9), (1, 10)],
                     [(10, 1), (3, 3)]]
-        vertices = np.array([[3.  ,   3.],
+        vertices = np.array([[3.  , 3.  ],
                              [9.75, 3.  ],
                              [9.  , 9.  ],
                              [3.  , 9.75]])
@@ -44,7 +44,8 @@ def test_abreu_sannikov(self):
         for d in self.game_dicts:
             rpg = RepeatedGame(d['sg'], d['delta'])
             for method in ('abreu_sannikov', 'AS'):
-                hull = rpg.equilibrium_payoffs(options={'u_init': d['u']})
+                hull = rpg.equilibrium_payoffs(method=method,
+                                               options={'u_init': d['u']})
                 assert_allclose(hull.points[hull.vertices], d['vertices'])
 
 if __name__ == '__main__':
