Fix tests to accomodate changes in NumPy

nolitsa/tests/test_d2.py

@@ -5,7 +5,7 @@
 import numpy as np
 
 from nolitsa import d2, utils
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 def test_c2():
@@ -74,7 +74,3 @@ def test_ttmle():
 
     desired = np.pi * (c[1:] / (c[1:] - c[0]))
     assert_allclose(desired, d2.ttmle(r, c, zero=False)[1])
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_data.py

@@ -5,7 +5,7 @@
 import numpy as np
 
 from nolitsa import data, utils
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 def test_falpha():
@@ -26,7 +26,3 @@ def test_falpha():
                               np.diff(np.log(freq[1:])))
 
             assert_allclose(-alpha, desired)
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_delay.py

@@ -3,7 +3,7 @@
 from __future__ import absolute_import, division, print_function
 import numpy as np
 from nolitsa import delay
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 class TestAcorr(object):
@@ -67,7 +67,3 @@ def test_adfd():
     dim, maxtau = 7, 25
     desired = np.sqrt(sqsum(dim - 1)) * b * np.arange(maxtau)
     assert_allclose(delay.adfd(x, dim=dim, maxtau=maxtau), desired)
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_dimension.py

@@ -5,7 +5,7 @@
 import numpy as np
 
 from nolitsa import dimension
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 class TestAFN(object):
@@ -95,7 +95,3 @@ def test_curve(self):
 
         f1 = dimension.fnn(x, dim=dim, tau=25)[0]
         np.allclose(f1, desired)
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_lyapunov.py

@@ -5,7 +5,7 @@
 import numpy as np
 
 from nolitsa import lyapunov
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 def test_mle():
@@ -43,7 +43,3 @@ def test_mle_embed():
 
     assert_allclose(desired, lyapunov.mle_embed(x, dim=dim, tau=tau,
                                                 maxt=25)[0], atol=1e-1)
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_noise.py

@@ -4,7 +4,7 @@
 import numpy as np
 
 from nolitsa import noise
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 class TestNoRed(object):
@@ -76,7 +76,3 @@ def test_line(self):
                         z[-c:] = x[-c:]
 
                     assert_allclose(y, z)
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_surrogates.py

@@ -5,7 +5,7 @@
 import numpy as np
 
 from nolitsa import surrogates, noise, utils
-from numpy.testing import assert_allclose, run_module_suite
+from numpy.testing import assert_allclose
 
 
 def test_ft():
@@ -56,7 +56,3 @@ def test_mismatch():
     x[desired[0] - 1:desired[1] + neigh] = 10.0
 
     assert_allclose(desired, surrogates.mismatch(x, neigh=neigh)[0])
-
-
-if __name__ == '__main__':
-    run_module_suite()

nolitsa/tests/test_utils.py

@@ -7,7 +7,7 @@
 
 from time import sleep
 from nolitsa import utils
-from numpy.testing import assert_, assert_allclose, run_module_suite
+from numpy.testing import assert_, assert_allclose
 
 
 def test_corrupt():
@@ -209,7 +209,3 @@ def test_stationary(self):
     def test_non_stationary(self):
         x = np.arange(1000)
         assert_(utils.statcheck(x)[1] < 1E-30)
-
-
-if __name__ == '__main__':
-    run_module_suite()