Process review comments for random and test_random
- add pragma no cover - format using ruff - fix module factor in test_randompull/130/head
parent
0c1e8f2b6a
commit
0124b33202
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue