markov test update

Author: HumphreyYang

quantecon/markov/tests/test_core.py

@@ -225,12 +225,12 @@ def test_left_eigen_vec(self):
         stationary_distributions = self.stationary
 
         if self.n_stat_dists == 1:
-            assert_allclose(np.dot(stationary_distributions, mc.P),
+            assert_allclose(stationary_distributions @ mc.P,
                             stationary_distributions, atol=self.TOL)
         else:
             for i in range(self.n_stat_dists):
                 curr_v = stationary_distributions[i, :]
-                assert_allclose(np.dot(curr_v, mc.P), curr_v, atol=self.TOL)
+                assert_allclose(curr_v @ mc.P, curr_v, atol=self.TOL)
 
 
 def test_simulate_shape():

quantecon/markov/tests/test_gth_solve.py

@@ -108,7 +108,7 @@ def __call__(self, x):
 
 class StationaryDistLeftEigenVec(AddDescription):
     def __call__(self, A, x):
-        assert_allclose(np.dot(x, A), x, atol=TOL)
+        assert_allclose(x @ A, x, atol=TOL)
 
 
 class StationaryDistEqualToKnown(AddDescription):