diff options
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() |