diff --git a/tests/test_random.py b/tests/test_random.py index dc26e48..9b07b71 100644 --- a/tests/test_random.py +++ b/tests/test_random.py @@ -39,13 +39,16 @@ def test_bytes(rng): assert len(rng.bytes(8)) == 8 assert len(rng.bytes(128)) == 128 + @pytest.fixture def rng_nonce(): return Random(b"abcdefghijklmnopqrstuv") + def test_nonce_byte(rng_nonce): assert len(rng_nonce.byte()) == 1 + @pytest.mark.parametrize("length", (1, 8, 128)) def test_nonce_bytes(rng_nonce, length): assert len(rng_nonce.bytes(length)) == length @@ -58,7 +61,7 @@ def test_reseed_sizes(rng, seed_size): Test that reseeding the random number generator works, for various seed sizes. """ # Create seed of required length. - seed = bytes(x % 255 for x in range(seed_size)) + seed = bytes(x % 256 for x in range(seed_size)) assert len(seed) == seed_size rng.reseed(seed) # Pull some bytes from the random number generator to test that it still works. @@ -74,9 +77,9 @@ def test_reseed_multiple(rng): for _ in range(10): # Create seed using random seed size for each call. seed_size = ord(rng.byte()) - seed = bytes(x % 255 for x in range(seed_size)) + seed = bytes(x % 256 for x in range(seed_size)) rng.reseed(seed) - + # Pull some bytes from the random number generator to test that it still works. num_bytes = ord(rng.byte()) rng.bytes(num_bytes) diff --git a/wolfcrypt/random.py b/wolfcrypt/random.py index 04fc6c5..2948eaf 100644 --- a/wolfcrypt/random.py +++ b/wolfcrypt/random.py @@ -85,5 +85,5 @@ class Random: """ assert self.native_object is not None ret = _lib.wc_RNG_DRBG_Reseed(self.native_object, seed, len(seed)) - if ret < 0: + if ret < 0: # pragma: no cover raise WolfCryptApiError("RNG reseed error", ret)