TEST: Updates for pytest

Author: oyamad

quantecon/game_theory/tests/test_brd.py

@@ -13,7 +13,7 @@
 class TestBRD:
     '''Test the methods of BRD'''
 
-    def setUp(self):
+    def setup_method(self):
         '''Setup a BRD instance'''
         # 2x2 coordination game with action 1 risk-dominant
         payoff_matrix = [[4, 0],
@@ -40,7 +40,7 @@ def test_time_series_2(self):
 class TestKMR:
     '''Test the methods of KMR'''
 
-    def setUp(self):
+    def setup_method(self):
         payoff_matrix = [[4, 0],
                          [3, 2]]
         self.N = 4
@@ -57,7 +57,7 @@ def test_time_series(self):
 class TestSamplingBRD:
     '''Test the methods of SamplingBRD'''
 
-    def setUp(self):
+    def setup_method(self):
         payoff_matrix = [[4, 0],
                          [3, 2]]
         self.N = 4
@@ -69,13 +69,3 @@ def test_time_series(self):
                                    random_state=np.random.RandomState(seed))
              for i in range(2)]
         assert_array_equal(x[0], x[1])
-
-
-if __name__ == '__main__':
-    import sys
-    import nose
-
-    argv = sys.argv[:]
-    argv.append('--verbose')
-    argv.append('--nocapture')
-    nose.main(argv=argv, defaultTest=__file__)

quantecon/game_theory/tests/test_fictplay.py

@@ -4,7 +4,6 @@
 Tests for fictplay.py
 
 """
-
 import numpy as np
 from numpy.testing import assert_array_almost_equal
 from scipy.stats import norm
@@ -14,7 +13,7 @@
 
 class TestFictitiousPlayDecreaingGain:
 
-    def setUp(self):
+    def setup_method(self):
         '''Setup a FictitiousPlay instance'''
         # symmetric 2x2 coordination game
         matching_pennies = [[(1, -1), (-1, 1)],
@@ -37,7 +36,7 @@ def test_time_series(self):
 
 class TestFictitiousPlayConstantGain:
 
-    def setUp(self):
+    def setup_method(self):
         matching_pennies = [[(1, -1), (-1, 1)],
                             [(-1, 1), (1, -1)]]
         self.fp = FictitiousPlay(matching_pennies, gain=0.1)
@@ -58,7 +57,7 @@ def test_time_series(self):
 
 class TestStochasticFictitiosuPlayDecreaingGain:
 
-    def setUp(self):
+    def setup_method(self):
         matching_pennies = [[(1, -1), (-1, 1)],
                             [(-1, 1), (1, -1)]]
         distribution = norm()
@@ -83,7 +82,7 @@ def test_time_series(self):
 
 class TestStochasticFictitiosuPlayConstantGain:
 
-    def setUp(self):
+    def setup_method(self):
         matching_pennies = [[(1, -1), (-1, 1)],
                             [(-1, 1), (1, -1)]]
         distribution = norm()
@@ -104,13 +103,3 @@ def test_time_series(self):
              for i in range(2)]
         assert_array_almost_equal(x[0][0], x[1][0])
         assert_array_almost_equal(x[0][1], x[1][1])
-
-
-if __name__ == '__main__':
-    import sys
-    import nose
-
-    argv = sys.argv[:]
-    argv.append('--verbose')
-    argv.append('--nocapture')
-    nose.main(argv=argv, defaultTest=__file__)

quantecon/game_theory/tests/test_localint.py

@@ -4,7 +4,6 @@
 Tests for localint.py
 
 """
-
 import numpy as np
 from numpy.testing import assert_array_equal, assert_equal
 
@@ -14,7 +13,7 @@
 class TestLocalInteraction:
     '''Test the methods of LocalInteraction'''
 
-    def setUp(self):
+    def setup_method(self):
         '''Setup a LocalInteraction instance'''
         payoff_matrix = np.asarray([[4, 0], [2, 3]])
         adj_matrix = np.asarray([[0, 1, 3],
@@ -56,13 +55,3 @@ def test_time_series_asynchronous_revision_with_player_index(self):
                                                actions=init_actions,
                                                player_ind_seq=player_ind_seq),
                            x)
-
-
-if __name__ == '__main__':
-    import sys
-    import nose
-
-    argv = sys.argv[:]
-    argv.append('--verbose')
-    argv.append('--nocapture')
-    nose.main(argv=argv, defaultTest=__file__)

quantecon/game_theory/tests/test_logitdyn.py

@@ -4,7 +4,6 @@
 Tests for logitdyn.py
 
 """
-
 import numpy as np
 from numpy.testing import assert_array_equal, assert_array_almost_equal_nulp
 
@@ -14,7 +13,7 @@
 class TestLogitDynamics:
     '''Test the methods of LogitDynamics'''
 
-    def setUp(self):
+    def setup_method(self):
         '''Setup a LogitDynamics instance'''
         # symmetric 2x2 coordination game
         payoff_matrix = [[4, 0],
@@ -54,13 +53,3 @@ def test_set_choice_probs_with_asymmetric_payoff_matrix():
     cdfs_computed = cdfs_computed / cdfs_computed[..., [-1]]  # Normalized
 
     assert_array_almost_equal_nulp(cdfs_computed, cdfs)
-
-
-if __name__ == '__main__':
-    import sys
-    import nose
-
-    argv = sys.argv[:]
-    argv.append('--verbose')
-    argv.append('--nocapture')
-    nose.main(argv=argv, defaultTest=__file__)