checkpoint: into main from release/2.4.4 @ b353a39 (#18670)

Source hash: b353a39
Remaining commits: 0

chia/_tests/core/data_layer/test_data_layer_util.py

@@ -159,6 +159,13 @@ def definition(left_hash: bytes32, right_hash: bytes32) -> bytes32:
             assert definition(left_hash=left_hash, right_hash=right_hash) == reference
 
 
+def get_random_bytes(length: int, r: Random) -> bytes:
+    if length == 0:
+        return b""
+
+    return r.getrandbits(length * 8).to_bytes(length, "big")
+
+
 def test_leaf_hash(seeded_random: Random) -> None:
     def definition(key: bytes, value: bytes) -> bytes32:
         return SerializedProgram.to((key, value)).get_tree_hash()
@@ -169,12 +176,14 @@ def definition(key: bytes, value: bytes) -> bytes32:
             length = 0
         else:
             length = seeded_random.randrange(100)
-        key = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
+
+        key = get_random_bytes(length=length, r=seeded_random)
+
         if cycle in (1, 2):
             length = 0
         else:
             length = seeded_random.randrange(100)
-        value = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
+        value = get_random_bytes(length=length, r=seeded_random)
         reference = definition(key=key, value=value)
         data.append((key, value, reference))
 
@@ -197,7 +206,7 @@ def definition(key: bytes) -> bytes32:
             length = 0
         else:
             length = seeded_random.randrange(100)
-        key = seeded_random.getrandbits(length * 8).to_bytes(length, "big")
+        key = get_random_bytes(length=length, r=seeded_random)
         reference = definition(key=key)
         data.append((key, reference))