Process review comments for random and test_random

- add pragma no cover
- format using ruff
- fix module factor in test_random
pull/130/head
Martijn de Milliano 2026-06-30 20:47:41 +02:00
parent 0c1e8f2b6a
commit 0124b33202
2 changed files with 7 additions and 4 deletions

View File

@ -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)

View File

@ -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)