Fix test_compute_fp

QuantEcon · Nov 17, 2016 · cf4ceab · cf4ceab
1 parent e35b46f
commit cf4ceab
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/quantecon/tests/test_compute_fp.py b/quantecon/tests/test_compute_fp.py
@@ -77,15 +77,16 @@ def test_imitation_game_method(self):
 
  for mu in [self.mu_1, self.mu_2]:
  for i in self.unit_inverval:
- fp_computed = \
- compute_fixed_point(self.T, i, mu=mu, **self.kwargs)
+ fp_computed = compute_fixed_point(self.T, i, method=method,
+   mu=mu, **self.kwargs)
  self.assertTrue(
  abs(self.T(fp_computed, mu=mu) - fp_computed) <= error_tol
  )
 
  # numpy array input
  i = np.asarray(self.unit_inverval)
- fp_computed = compute_fixed_point(self.T, i, mu=mu, **self.kwargs)
+ fp_computed = compute_fixed_point(self.T, i, method=method, mu=mu,
+ **self.kwargs)
  self.assertTrue(
  abs(self.T(fp_computed, mu=mu) - fp_computed).max() <=
  error_tol