diff options
| author | Matti Picus <matti.picus@gmail.com> | 2020-04-11 21:16:17 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-04-11 21:16:17 +0300 |
| commit | 3ac8f65681e2ee55c8b8bbb25f1d4aacba07d9bc (patch) | |
| tree | 0e543a7d7a40161c03f3abee0cd2e265d3a91e7f /numpy/random/tests/test_random.py | |
| parent | 8f7adad487ebac177964a577f5d263da975d8333 (diff) | |
| parent | 9dda11f9c0c99d6735a070fa4744465585bab6aa (diff) | |
| download | numpy-3ac8f65681e2ee55c8b8bbb25f1d4aacba07d9bc.tar.gz | |
Merge pull request #15876 from panpiort8/multinomial_pvals_1d
BUG: Check that `pvals` is 1D in `_generator.multinomial`.
Diffstat (limited to 'numpy/random/tests/test_random.py')
| -rw-r--r-- | numpy/random/tests/test_random.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py index a9aa15083..c5f79d2c1 100644 --- a/numpy/random/tests/test_random.py +++ b/numpy/random/tests/test_random.py @@ -91,6 +91,12 @@ class TestMultinomial: assert_raises(TypeError, np.random.multinomial, 1, p, float(1)) + def test_multidimensional_pvals(self): + assert_raises(ValueError, np.random.multinomial, 10, [[0, 1]]) + assert_raises(ValueError, np.random.multinomial, 10, [[0], [1]]) + assert_raises(ValueError, np.random.multinomial, 10, [[[0], [1]], [[1], [0]]]) + assert_raises(ValueError, np.random.multinomial, 10, np.array([[0, 1], [1, 0]])) + class TestSetState: def setup(self): |