diff options
| author | Daniel da Silva <daniel@meltingwax.net> | 2014-04-04 21:58:11 -0400 |
|---|---|---|
| committer | Daniel da Silva <daniel@meltingwax.net> | 2014-04-04 21:58:11 -0400 |
| commit | ac56b60a72df1ee939f3aa0826389276fc4b0b64 (patch) | |
| tree | be7b8ce188507dd3e9da64be746cb6e886de372d /numpy/random/tests/test_random.py | |
| parent | 547765dd9b2131e1e8b1ef646d4cc5ba9b4dd791 (diff) | |
| download | numpy-ac56b60a72df1ee939f3aa0826389276fc4b0b64.tar.gz | |
BUG: Explicitly reject nan values for p in binomial(n, p). Fixes #4571.
Adds check with np.isnan(p) and raises ValueError if check is positive.
Diffstat (limited to 'numpy/random/tests/test_random.py')
| -rw-r--r-- | numpy/random/tests/test_random.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py index c30037e1e..db4093ab4 100644 --- a/numpy/random/tests/test_random.py +++ b/numpy/random/tests/test_random.py @@ -18,6 +18,10 @@ class TestBinomial(TestCase): assert_(random.binomial(0, p) == 0) np.testing.assert_array_equal(random.binomial(zeros, p), zeros) + def test_p_is_nan(self): + # Issue #4571. + assert_raises(ValueError, random.binomial, 1, np.nan) + class TestMultinomial(TestCase): def test_basic(self): |