status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix utils.hash_function typing

This commit is contained in:
Hsiao-Wei Wang 2020-04-17 18:15:46 +08:00
parent 6fdee75475
commit cafd98b9e8
No known key found for this signature in database
GPG Key ID: 95B070122902DEA4
1 changed files with 6 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/core/pyspec/eth2spec/utils/hash_function.py
View File

@ -1,28 +1,17 @@
from hashlib import sha256 from hashlib import sha256
from typing import Dict, Union
ZERO_BYTES32 = b'\x00' * 32 ZERO_BYTES32 = b'\x00' * 32
def _hash(x): def _hash(x: Union[bytes, bytearray, memoryview]) -> bytes:
return sha256(x).digest() return sha256(x).digest()
# Minimal collection of (key, value) pairs, for fast hash-retrieval, to save on repetitive computation cost. hash_cache: Dict[bytes, bytes] = {}
# Key = the hash input
# Value = the hash output
hash_cache = []
def add_zero_hashes_to_cache(): def hash(x: bytes) -> bytes:
zerohashes = [(None, ZERO_BYTES32)] if x in hash_cache:
for layer in range(1, 32): return hash_cache[x]
k = zerohashes[layer - 1][1] + zerohashes[layer - 1][1]
zerohashes.append((k, _hash(k)))
hash_cache.extend(zerohashes[1:])
def hash(x):
for (k, h) in hash_cache:
if x == k:
return h
return _hash(x) return _hash(x)