diff options
author | Robert Kern <robert.kern@gmail.com> | 2019-06-25 20:52:03 -0700 |
---|---|---|
committer | Robert Kern <robert.kern@gmail.com> | 2019-06-25 20:52:03 -0700 |
commit | 45473e3e7d661ebdfa97afa757b9cbba103a155e (patch) | |
tree | b29f146b9c04fb8c61ceeea4494ba5346fd8bc45 /numpy/random/tests/test_smoke.py | |
parent | 4668b46310e74b2f3c78eb213b6d796456278942 (diff) | |
download | numpy-45473e3e7d661ebdfa97afa757b9cbba103a155e.tar.gz |
ENH: Add SFC64 BitGenerator.
Diffstat (limited to 'numpy/random/tests/test_smoke.py')
-rw-r--r-- | numpy/random/tests/test_smoke.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/numpy/random/tests/test_smoke.py b/numpy/random/tests/test_smoke.py index c512cb26c..4263335f6 100644 --- a/numpy/random/tests/test_smoke.py +++ b/numpy/random/tests/test_smoke.py @@ -5,7 +5,7 @@ from functools import partial import numpy as np import pytest from numpy.testing import assert_equal, assert_, assert_array_equal -from numpy.random import (Generator, MT19937, PCG64, Philox, entropy) +from numpy.random import (Generator, MT19937, PCG64, Philox, SFC64, entropy) @pytest.fixture(scope='module', params=(np.bool, np.int8, np.int16, np.int32, np.int64, @@ -145,7 +145,7 @@ class RNG(object): assert_(comp_state(jumped_state, rejumped_state)) else: bitgen_name = self.rg.bit_generator.__class__.__name__ - if bitgen_name not in ('',): + if bitgen_name not in ('SFC64',): raise AttributeError('no "jumped" in %s' % bitgen_name) pytest.skip('Jump is not supported by {0}'.format(bitgen_name)) @@ -752,6 +752,18 @@ class TestPhilox(RNG): cls._extra_setup() +class TestSFC64(RNG): + @classmethod + def setup_class(cls): + cls.bit_generator = SFC64 + cls.advance = None + cls.seed = [12345] + cls.rg = Generator(cls.bit_generator(*cls.seed)) + cls.initial_state = cls.rg.bit_generator.state + cls.seed_vector_bits = 192 + cls._extra_setup() + + class TestEntropy(object): def test_entropy(self): e1 = entropy.random_entropy() |