diff options
| author | Jaime Fernandez <jaime.frio@gmail.com> | 2015-05-08 21:08:43 -0700 |
|---|---|---|
| committer | Jaime Fernandez <jaime.frio@gmail.com> | 2015-05-10 03:11:42 -0700 |
| commit | c7f121abb24e19c34246c8ad834dfdd5bd5ce8b2 (patch) | |
| tree | b8eab139785549d959dee3ea4482c6ebb9c2ae47 /numpy/random/tests/test_regression.py | |
| parent | 30e3d41b6d11c18e17eb283a61bbe9bbf4bb4d8f (diff) | |
| download | numpy-c7f121abb24e19c34246c8ad834dfdd5bd5ce8b2.tar.gz | |
BUG: np.random.beta with small parameters produces NaNs
Fixes #5851
Diffstat (limited to 'numpy/random/tests/test_regression.py')
| -rw-r--r-- | numpy/random/tests/test_regression.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/numpy/random/tests/test_regression.py b/numpy/random/tests/test_regression.py index 1a5854e82..52be0d4d5 100644 --- a/numpy/random/tests/test_regression.py +++ b/numpy/random/tests/test_regression.py @@ -93,5 +93,13 @@ class TestRegression(TestCase): np.random.multivariate_normal([0], [[0]], size=np.int_(1)) np.random.multivariate_normal([0], [[0]], size=np.int64(1)) + def test_beta_small_parameters(self): + # Test that beta with small a and b parameters does not produce + # NaNs due to roundoff errors causing 0 / 0, gh-5851 + np.random.seed(1234567890) + x = np.random.beta(0.0001, 0.0001, size=100) + assert_(not np.any(np.isnan(x)), 'Nans in np.random.beta') + + if __name__ == "__main__": run_module_suite() |