diff options
| author | Kevin Sheppard <kevin.sheppard@gmail.com> | 2021-02-26 23:06:56 +0000 |
|---|---|---|
| committer | Kevin Sheppard <kevin.sheppard@gmail.com> | 2021-02-26 23:44:50 +0000 |
| commit | b1015adfdbce55b7ee9211baca2f51284d67694a (patch) | |
| tree | 7b9a745abd5912105c95b9dcb7bb099c5a4951eb /numpy/random/tests | |
| parent | e900be2bf367d2a90af922e56c27aadcd0581bdf (diff) | |
| download | numpy-b1015adfdbce55b7ee9211baca2f51284d67694a.tar.gz | |
Port error to RandomState
Diffstat (limited to 'numpy/random/tests')
| -rw-r--r-- | numpy/random/tests/test_generator_mt19937.py | 2 | ||||
| -rw-r--r-- | numpy/random/tests/test_randomstate.py | 8 |
2 files changed, 9 insertions, 1 deletions
diff --git a/numpy/random/tests/test_generator_mt19937.py b/numpy/random/tests/test_generator_mt19937.py index 9de044774..446b350dd 100644 --- a/numpy/random/tests/test_generator_mt19937.py +++ b/numpy/random/tests/test_generator_mt19937.py @@ -147,7 +147,7 @@ class TestMultinomial: 1.0e-09, 1.0e-09, 1.0e-09, 1.0e-09], dtype=np.float32) pvals = x / x.sum() random = Generator(MT19937(1432985819)) - match = r"[\w\s]*pvals are cast to 64-bit floating" + match = r"[\w\s]*pvals array is cast to 64-bit floating" with pytest.raises(ValueError, match=match): random.multinomial(1, pvals) diff --git a/numpy/random/tests/test_randomstate.py b/numpy/random/tests/test_randomstate.py index b16275b70..861813a95 100644 --- a/numpy/random/tests/test_randomstate.py +++ b/numpy/random/tests/test_randomstate.py @@ -167,6 +167,14 @@ class TestMultinomial: contig = random.multinomial(100, pvals=np.ascontiguousarray(pvals)) assert_array_equal(non_contig, contig) + def test_multinomial_pvals_float32(self): + x = np.array([9.9e-01, 9.9e-01, 1.0e-09, 1.0e-09, 1.0e-09, 1.0e-09, + 1.0e-09, 1.0e-09, 1.0e-09, 1.0e-09], dtype=np.float32) + pvals = x / x.sum() + match = r"[\w\s]*pvals array is cast to 64-bit floating" + with pytest.raises(ValueError, match=match): + random.multinomial(1, pvals) + class TestSetState: def setup(self): |