diff options
Diffstat (limited to 'numpy/random/tests')
-rw-r--r-- | numpy/random/tests/test_generator_mt19937.py | 2 | ||||
-rw-r--r-- | numpy/random/tests/test_random.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/numpy/random/tests/test_generator_mt19937.py b/numpy/random/tests/test_generator_mt19937.py index 2a30d39e2..2397a2fac 100644 --- a/numpy/random/tests/test_generator_mt19937.py +++ b/numpy/random/tests/test_generator_mt19937.py @@ -116,7 +116,7 @@ class TestMultinomial: assert_array_equal(non_contig, contig) def test_multidimensional_pvals(self): - assert_raises(ValueError, random.multinomial, 10, [[0, 1], [1, 0]]) + assert_raises(ValueError, random.multinomial, 10, [[0, 1]]) assert_raises(ValueError, random.multinomial, 10, [[0], [1]]) assert_raises(ValueError, random.multinomial, 10, [[[0], [1]], [[1], [0]]]) assert_raises(ValueError, random.multinomial, 10, np.array([[0, 1], [1, 0]])) diff --git a/numpy/random/tests/test_random.py b/numpy/random/tests/test_random.py index 935afbf71..c5f79d2c1 100644 --- a/numpy/random/tests/test_random.py +++ b/numpy/random/tests/test_random.py @@ -92,7 +92,7 @@ class TestMultinomial: 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]]) assert_raises(ValueError, np.random.multinomial, 10, [[[0], [1]], [[1], [0]]]) assert_raises(ValueError, np.random.multinomial, 10, np.array([[0, 1], [1, 0]])) |