summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--numpy/random/tests/test_regression.py21
1 files changed, 8 insertions, 13 deletions
diff --git a/numpy/random/tests/test_regression.py b/numpy/random/tests/test_regression.py
index 9e473f9b1..9f7455fe5 100644
--- a/numpy/random/tests/test_regression.py
+++ b/numpy/random/tests/test_regression.py
@@ -10,21 +10,19 @@ import numpy as np
class TestRegression(TestCase):
def test_VonMises_range(self):
- """Make sure generated random variables are in [-pi, pi].
-
- Regression test for ticket #986.
- """
+ # Make sure generated random variables are in [-pi, pi].
+ # Regression test for ticket #986.
for mu in np.linspace(-7., 7., 5):
r = random.mtrand.vonmises(mu, 1, 50)
assert_(np.all(r > -np.pi) and np.all(r <= np.pi))
def test_hypergeometric_range(self):
- """Test for ticket #921"""
+ # Test for ticket #921
assert_(np.all(np.random.hypergeometric(3, 18, 11, size=10) < 4))
assert_(np.all(np.random.hypergeometric(18, 3, 11, size=10) > 0))
def test_logseries_convergence(self):
- """Test for ticket #923"""
+ # Test for ticket #923
N = 1000
np.random.seed(0)
rvsn = np.random.logseries(0.8, size=N)
@@ -56,7 +54,7 @@ class TestRegression(TestCase):
raise AssertionError
def test_shuffle_mixed_dimension(self):
- """Test for trac ticket #2074"""
+ # Test for trac ticket #2074
for t in [[1, 2, 3, None],
[(1, 1), (2, 2), (3, 3), None],
[1, (2, 2), (3, 3), None],
@@ -77,12 +75,9 @@ class TestRegression(TestCase):
assert_array_equal(m.choice(10, size=10, p=np.ones(10.)/10), res)
def test_multivariate_normal_size_types(self):
- """Test for multivariate_normal issue with 'size' argument.
-
- Check that the multivariate_normal size argument can be a
- numpy integer.
-
- """
+ # Test for multivariate_normal issue with 'size' argument.
+ # Check that the multivariate_normal size argument can be a
+ # numpy integer.
np.random.multivariate_normal([0], [[0]], size=1)
np.random.multivariate_normal([0], [[0]], size=np.int_(1))
np.random.multivariate_normal([0], [[0]], size=np.int64(1))