Skip to content

Commit b353a39

Browse files
authored
stop asking for zero random bits of data (#18668)
1 parent 0ab7291 commit b353a39

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

chia/_tests/core/data_layer/test_data_layer_util.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,13 @@ def definition(left_hash: bytes32, right_hash: bytes32) -> bytes32:
159159
assert definition(left_hash=left_hash, right_hash=right_hash) == reference
160160

161161

162+
def get_random_bytes(length: int, r: Random) -> bytes:
163+
if length == 0:
164+
return b""
165+
166+
return r.getrandbits(length * 8).to_bytes(length, "big")
167+
168+
162169
def test_leaf_hash(seeded_random: Random) -> None:
163170
def definition(key: bytes, value: bytes) -> bytes32:
164171
return SerializedProgram.to((key, value)).get_tree_hash()
@@ -169,12 +176,14 @@ def definition(key: bytes, value: bytes) -> bytes32:
169176
length = 0
170177
else:
171178
length = seeded_random.randrange(100)
172-
key = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
179+
180+
key = get_random_bytes(length=length, r=seeded_random)
181+
173182
if cycle in (1, 2):
174183
length = 0
175184
else:
176185
length = seeded_random.randrange(100)
177-
value = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
186+
value = get_random_bytes(length=length, r=seeded_random)
178187
reference = definition(key=key, value=value)
179188
data.append((key, value, reference))
180189

@@ -197,7 +206,7 @@ def definition(key: bytes) -> bytes32:
197206
length = 0
198207
else:
199208
length = seeded_random.randrange(100)
200-
key = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
209+
key = get_random_bytes(length=length, r=seeded_random)
201210
reference = definition(key=key)
202211
data.append((key, reference))
203212

0 commit comments

Comments
 (0)