From 96fd8ecf7c88c878e75fc36c7f24e8cf5263b6ee Mon Sep 17 00:00:00 2001 From: Kyle Altendorf <sda@fstab.net> Date: Fri, 24 Jan 2025 10:01:07 -0500 Subject: [PATCH] more --- chia/_tests/core/data_layer/test_merkle_blob.py | 6 +++--- chia/data_layer/util/merkle_blob.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/chia/_tests/core/data_layer/test_merkle_blob.py b/chia/_tests/core/data_layer/test_merkle_blob.py index b7cecd3bf1a8..0229a1ee1ddd 100644 --- a/chia/_tests/core/data_layer/test_merkle_blob.py +++ b/chia/_tests/core/data_layer/test_merkle_blob.py @@ -234,10 +234,10 @@ def generate_kvid(seed: int) -> tuple[KVId, KVId]: return kv_ids[0], kv_ids[1] -def generate_hash(seed: int) -> bytes: +def generate_hash(seed: int) -> bytes32: seed_bytes = seed.to_bytes(8, byteorder="big", signed=True) hash_obj = hashlib.sha256(seed_bytes) - return hash_obj.digest() + return bytes32(hash_obj.digest()) def test_insert_delete_loads_all_keys() -> None: @@ -337,7 +337,7 @@ def test_proof_of_inclusion_merkle_blob() -> None: num_deletes = 1 + repeats * 10 kv_ids: list[tuple[KVId, KVId]] = [] - hashes: list[bytes] = [] + hashes: list[bytes32] = [] for _ in range(num_inserts): seed += 1 key, value = generate_kvid(seed) diff --git a/chia/data_layer/util/merkle_blob.py b/chia/data_layer/util/merkle_blob.py index 29a5950edef2..02745d364277 100644 --- a/chia/data_layer/util/merkle_blob.py +++ b/chia/data_layer/util/merkle_blob.py @@ -560,7 +560,7 @@ def get_nodes_with_indexes(self, index: TreeIndex = TreeIndex(0)) -> list[tuple[ return this + left_nodes + right_nodes - def batch_insert(self, keys_values: list[tuple[KVId, KVId]], hashes: list[bytes]) -> None: + def batch_insert(self, keys_values: list[tuple[KVId, KVId]], hashes: list[bytes32]) -> None: indexes: list[TreeIndex] = [] if len(self.key_to_index) <= 1: