@@ -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+
162169def 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