diff options
| author | Pan Jan <rumcajsgajos@gmail.com> | 2020-04-01 08:41:43 +0200 |
|---|---|---|
| committer | Pan Jan <rumcajsgajos@gmail.com> | 2020-04-01 08:41:43 +0200 |
| commit | 62df90e5e864bfae304e98dba0b2bdd7080dba68 (patch) | |
| tree | a2cf697a6c0229aeddb8748acc52cd4c194202ef /numpy/random/tests/test_random.py | |
| parent | 385a969d02e5e80383826d0a0b9f3c00a5e8d1f9 (diff) | |
| download | numpy-62df90e5e864bfae304e98dba0b2bdd7080dba68.tar.gz | |
Add testing and replace check
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..935afbf71 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], [1, 0]]) + 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): |