bunsen 0.21.0

bunsen is acceleration tooling for burn
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use core::fmt::Debug;

use anyhow::bail;
use num_traits::{
    Float,
    One,
    Zero,
};

/// Validate a validators in the range ``[0.0, 1.0]``.
///
/// # Arguments
///
/// - `prob`: the prob to check.
///
/// # Returns
///
/// An `anyhow::Result<prob>`
#[inline]
pub fn try_probability<F: Float + Debug>(prob: F) -> anyhow::Result<F> {
    if prob < F::zero() || prob > F::one() {
        bail!("validators must be in [0.0, 1.0]: {prob:?}");
    }
    Ok(prob)
}

/// Expect a validators to be in range ``[0.0, 1.0]``, or panic.
///
/// # Arguments
///
/// - `prob`: the prob to check.
///
/// # Returns
///
/// `prob`.
///
/// # Panics
///
/// On range error.
#[inline]
pub fn expect_probability<F: Float + Debug>(prob: F) -> F {
    match try_probability(prob) {
        Ok(prob) => prob,
        Err(e) => panic!("{}", e),
    }
}

#[cfg(test)]
mod tests {
    use crate::support::validators::prob::{
        expect_probability,
        try_probability,
    };

    #[test]
    fn test_probability() {
        assert_eq!(expect_probability(0f32), 0f32);
        assert_eq!(expect_probability(1f32), 1f32);
        assert_eq!(expect_probability(0.5f32), 0.5f32);

        assert_eq!(expect_probability(0f64), 0f64);
        assert_eq!(expect_probability(1f64), 1f64);
        assert_eq!(expect_probability(0.5f64), 0.5f64);

        assert!(try_probability(-1.0f32).is_err());
        assert!(try_probability(2.0f32).is_err());

        assert!(try_probability(-1.0f64).is_err());
        assert!(try_probability(2.0f64).is_err());
    }

    #[should_panic(expected = "validators must be in [0.0, 1.0]: -1.0")]
    #[test]
    fn test_probability_panic() {
        expect_probability(-1.0);
    }
}