diff options
Diffstat (limited to 'numpy/random/tests')
| -rw-r--r-- | numpy/random/tests/test_random.py | 5 | ||||
| -rw-r--r-- | numpy/random/tests/test_regression.py | 5 |
2 files changed, 2 insertions, 8 deletions
diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py index 4546a0184..61c6e912d 100644 --- a/numpy/random/tests/test_random.py +++ b/numpy/random/tests/test_random.py @@ -3,7 +3,7 @@ import warnings import numpy as np from numpy.testing import ( - run_module_suite, assert_, assert_raises, assert_equal, assert_warns, + assert_, assert_raises, assert_equal, assert_warns, assert_no_warnings, assert_array_equal, assert_array_almost_equal, suppress_warnings ) @@ -1634,6 +1634,3 @@ class TestSingleEltArrayInput(object): out = func(self.argOne, self.argTwo[0], self.argThree) assert_equal(out.shape, self.tgtShape) - -if __name__ == "__main__": - run_module_suite() diff --git a/numpy/random/tests/test_regression.py b/numpy/random/tests/test_regression.py index 572f4c087..3b4b4ed40 100644 --- a/numpy/random/tests/test_regression.py +++ b/numpy/random/tests/test_regression.py @@ -2,7 +2,7 @@ from __future__ import division, absolute_import, print_function import sys from numpy.testing import ( - run_module_suite, assert_, assert_array_equal, assert_raises, + assert_, assert_array_equal, assert_raises, ) from numpy import random from numpy.compat import long @@ -133,6 +133,3 @@ class TestRegression(object): # Force Garbage Collection - should not segfault. import gc gc.collect() - -if __name__ == "__main__": - run_module_suite() |