milhouse 0.9.0

Persistent binary merkle tree
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use crate::List;
use proptest::prelude::*;
use ssz_derive::{Decode, Encode};
use tree_hash::Hash256;
use tree_hash_derive::TreeHash;
use typenum::{U4, Unsigned};

mod operations;
mod rebase;
mod tree_hash_and_ssz;

pub fn arb_index(n: usize) -> impl Strategy<Value = usize> {
    any::<proptest::sample::Index>().prop_map(move |index| index.index(n))
}

pub fn arb_list<T, N, S>(strategy: S) -> impl Strategy<Value = Vec<T>>
where
    S: Strategy<Value = T>,
    T: std::fmt::Debug,
    N: Unsigned + std::fmt::Debug,
{
    proptest::collection::vec(strategy, 0..=N::to_usize())
}

pub fn arb_vect<T, N, S>(strategy: S) -> impl Strategy<Value = Vec<T>>
where
    S: Strategy<Value = T>,
    T: std::fmt::Debug,
    N: Unsigned + std::fmt::Debug,
{
    proptest::collection::vec(strategy, N::to_usize())
}

pub fn arb_hash256() -> impl Strategy<Value = Hash256> {
    proptest::array::uniform32(any::<u8>()).prop_map(Hash256::from)
}

/// Struct with multiple fields shared by multiple proptests.
#[derive(Debug, Clone, PartialEq, Encode, Decode, TreeHash)]
pub struct Large {
    pub a: u64,
    pub b: u8,
    pub c: Hash256,
    pub d: List<u64, U4>,
}

pub fn arb_large() -> impl Strategy<Value = Large> {
    (
        any::<u64>(),
        any::<u8>(),
        arb_hash256(),
        arb_list::<_, U4, _>(any::<u64>()),
    )
        .prop_map(|(a, b, c, d)| Large {
            a,
            b,
            c,
            d: List::new(d).unwrap(),
        })
}