Merge pull request statsmodels#5156 from jbrockmendel/nomains

TST: Remove __main__ sections in tests

Author: josef-pkt

statsmodels/base/tests/test_data.py

@@ -910,8 +910,3 @@ def test_raise_nonfinite_exog():
     assert_raises(MissingDataError, OLS, y, x)
     x[1, 1] = np.nan
     assert_raises(MissingDataError, OLS, y, x)
-
-
-if __name__ == "__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])

statsmodels/base/tests/test_shrink_pickle.py

@@ -283,19 +283,3 @@ def setup(self):
         y_count = np.random.poisson(np.exp(x.sum(1) - x.mean()))
         model = sm.Poisson(y_count, x)
         self.results = model.fit_regularized(method='l1', disp=0, alpha=10)
-
-
-if __name__ == '__main__':
-    for cls in [TestRemoveDataPickleOLS, TestRemoveDataPickleWLS,
-                TestRemoveDataPicklePoisson,
-                TestRemoveDataPicklePoissonRegularized,
-                TestRemoveDataPickleNegativeBinomial,
-                TestRemoveDataPickleLogit, TestRemoveDataPickleRLM,
-                TestRemoveDataPickleGLM]:
-        print(cls)
-        cls.setup_class()
-        tt = cls()
-        tt.setup()
-        tt.test_remove_data_pickle()
-        tt.test_remove_data_docstring()
-        tt.test_pickle_wrapper()

statsmodels/compat/tests/test_itercompat.py

@@ -34,10 +34,3 @@ def test_combinations():
     actual = list(combinations(lrange(4), 3))
     desired = [(0, 1, 2), (0, 1, 3), (0, 2, 3), (1, 2, 3)]
     assert_(actual == desired, '%r not equal %r' % (actual, desired))
-
-
-
-
-if __name__ == '__main__':
-    test_zip_longest()
-    test_combinations()

statsmodels/discrete/tests/test_discrete.py

@@ -2376,8 +2376,3 @@ def test_unchanging_degrees_of_freedom():
     # Test that the call to `fit_regularized` didn't modify model.df_model inplace.
     assert_equal(res3.df_model, res1.df_model)
     assert_equal(res3.df_resid, res1.df_resid)
-
-
-if __name__ == "__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])

statsmodels/discrete/tests/test_sandwich_cov.py

@@ -722,13 +722,3 @@ def setup_class(cls):
     def test_kwd(self):
         # test corrected keyword name
         assert_allclose(self.res1b.bse, self.res1.bse, rtol=1e-12)
-
-
-
-if __name__ == '__main__':
-    tt = TestPoissonClu()
-    tt.setup_class()
-    tt.test_basic()
-    tt = TestNegbinClu()
-    tt.setup_class()
-    tt.test_basic()

statsmodels/distributions/tests/test_discrete.py

@@ -109,7 +109,3 @@ def test_logpmf(self):
         nb_logpmf = nbinom.logpmf(2, n, p)
         tnb_logpmf = sm.distributions.zinegbin.logpmf(2, 5, 1, 1, 0.005)
         assert_allclose(nb_logpmf, tnb_logpmf, rtol=1e-2, atol=1e-2)
-
-if __name__ == "__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])

statsmodels/distributions/tests/test_edgeworth.py

@@ -4,7 +4,7 @@
 
 import numpy as np
 from numpy.polynomial.hermite_e import HermiteE
-from numpy.testing import (run_module_suite, assert_equal, assert_raises,
+from numpy.testing import (assert_equal, assert_raises,
                            assert_allclose)
 import numpy.testing as npt
 
@@ -188,7 +188,3 @@ def check_distribution_rvs(distfn, args, alpha, rvs):
         D,pval = stats.kstest(distfn.rvs, distfn.cdf, args=args, N=1000)
         npt.assert_(pval > alpha, "D = " + str(D) + "; pval = " + str(pval) +
                "; alpha = " + str(alpha) + "\nargs = " + str(args))
-
-
-if __name__ == "__main__":
-    run_module_suite()

statsmodels/distributions/tests/test_mixture.py

@@ -103,8 +103,3 @@ def test_mv_mixture_rvs_fixed(self):
                        [ 3.60521933,  1.57316531,  0.82784584],
                        [ 3.86102275,  0.6211812 ,  1.33016426],
                        [ 3.91074761,  2.037155  ,  2.22247051]]))
-
-
-if __name__ == "__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])

statsmodels/duration/tests/test_phreg.py

@@ -414,8 +414,3 @@ def plf(params):
                          list(itertools.product(fnames, ties, entry_f, strata_f)))
 def test_r(fname, ties, entry_f, strata_f):
     TestPHReg.do1(fname, ties, entry_f, strata_f)
-
-
-if __name__ == "__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])

statsmodels/genmod/families/tests/test_link.py

@@ -114,7 +114,3 @@ def test_invlogit_stability():
          812.2382651982667, 495.06449978924525]
     zinv = logit.inverse(z)
     assert_equal(zinv, np.ones_like(z))
-
-if __name__=="__main__":
-    import pytest
-    pytest.main([__file__, '-vvs', '-x', '--pdb'])