gnosis/lighthouse
1
0
Fork 0
mirror of https://github.com/sigp/lighthouse.git synced 2026-04-28 02:03:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement basic vec tree hash cache

Browse Source
This commit is contained in:
Paul Hauner
2019-03-28 13:05:24 +11:00
parent 7af6eecb44
commit 224a967cce
3 changed files with 128 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
eth2/utils/ssz/src/cached_tree_hash/tests.rs
View File

@@ -1,5 +1,5 @@
use super::*;
use int_to_bytes::int_to_bytes32;
use int_to_bytes::{int_to_bytes32, int_to_bytes8};
#[derive(Clone)]
pub struct Inner {
@@ -10,6 +10,8 @@ pub struct Inner {
}
impl CachedTreeHash for Inner {
type Item = Self;
fn build_cache_bytes(&self) -> Vec<u8> {
let mut leaves = vec![];
@@ -21,15 +23,6 @@ impl CachedTreeHash for Inner {
merkleize(leaves)
}
fn max_num_leaves(&self) -> usize {
let mut leaves = 0;
leaves += self.a.max_num_leaves();
leaves += self.b.max_num_leaves();
leaves += self.c.max_num_leaves();
leaves += self.d.max_num_leaves();
leaves
}
fn num_bytes(&self) -> usize {
let mut bytes = 0;
bytes += self.a.num_bytes();
@@ -45,7 +38,12 @@ impl CachedTreeHash for Inner {
cache: &mut TreeHashCache,
chunk: usize,
) -> Option<usize> {
let num_leaves = self.max_num_leaves();
let mut num_leaves: usize = 0;
num_leaves += num_unsanitized_leaves(self.a.num_bytes());
num_leaves += num_unsanitized_leaves(self.b.num_bytes());
num_leaves += num_unsanitized_leaves(self.c.num_bytes());
num_leaves += num_unsanitized_leaves(self.d.num_bytes());
let num_nodes = num_nodes(num_leaves);
let num_internal_nodes = num_nodes - num_leaves;
@@ -78,6 +76,26 @@ fn join(many: Vec<Vec<u8>>) -> Vec<u8> {
all
}
#[test]
fn vec_of_u64() {
let data = join(vec![
int_to_bytes8(1),
int_to_bytes8(2),
int_to_bytes8(3),
int_to_bytes8(4),
int_to_bytes8(5),
vec![0; 32 - 8], // padding
]);
let expected = merkleize(data);
let my_vec = vec![1, 2, 3, 4, 5];
let cache = my_vec.build_cache_bytes();
assert_eq!(expected, cache);
}
#[test]
fn merkleize_odd() {
let data = join(vec![