pbt 0.4.11

Property-based testing with `derive` macros, aware of mutual induction & instantiability.
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

//! Implementations for `core::num::NonZero<_>` types.

use {
    crate::{
        construct::{Construct, Literal, TypeFormer, visit_self, visit_self_owned},
        reflection::{Type, register, type_of},
        scc::StronglyConnectedComponents,
        shrink::shrink,
    },
    core::num::NonZero,
    std::collections::BTreeSet,
};

/// Generate an arbitrary nonzero value for an
/// unsigned integer of fixed but unspecified width.
#[macro_export]
macro_rules! arbitrary_nonzero_unsigned {
    // TODO: iterate over a `u64` as 64 booleans
    // instead of recomputing each
    ($u:ty, $prng:ident) => {{
        let mut acc: $u = 1;

        while ($prng.rand() & 3) != 0 {
            #[allow(
                clippy::allow_attributes,
                clippy::default_numeric_fallback,
                reason = "type varies"
            )]
            if acc.cast_signed() < 0 {
                acc = <$u>::MAX;
                break;
            }
            acc <<= 1_u8;
            acc |= <$u>::from(($prng.rand() & 1) != 0);
        }
        acc
    }};
}

/// Implement `Construct` for `NonZero<$u>`.
macro_rules! impl_for {
    ($u:ty) => {
        impl Construct for NonZero<$u> {
            #[inline]
            fn register_all_immediate_dependencies(
                visited: &mut BTreeSet<Type>,
                sccs: &mut StronglyConnectedComponents,
            ) {
                if !visited.insert(type_of::<Self>()) {
                    return;
                }
                let () = register::<$u>(visited.clone(), sccs);
            }

            #[inline]
            fn type_former() -> TypeFormer<Self> {
                TypeFormer::Literal(Literal {
                    deserialize: |s| NonZero::new(s.parse().ok()?),
                    // SAFETY: Internals of `arbitrary_nonzero_unsigned`
                    // prevent `0` (see above).
                    generate: |prng| unsafe {
                        NonZero::new_unchecked(arbitrary_nonzero_unsigned!($u, prng))
                    },
                    serialize: |value: &Self| value.get().to_string(),
                    shrink: |u| Box::new(shrink(u.get()).filter_map(NonZero::new)),
                })
            }

            #[inline]
            fn visit_deep<V: Construct>(&self) -> impl Iterator<Item = V> {
                visit_self(self).chain(visit_self_owned(self.get()))
            }
        }
    };
}

impl_for!(u8);
impl_for!(u16);
impl_for!(u32);
impl_for!(u64);
impl_for!(u128);
impl_for!(usize);

impl Construct for NonZero<char> {
    #[inline]
    fn register_all_immediate_dependencies(
        visited: &mut BTreeSet<Type>,
        sccs: &mut StronglyConnectedComponents,
    ) {
        if !visited.insert(type_of::<Self>()) {
            return;
        }
        let () = register::<char>(visited.clone(), sccs);
    }

    #[inline]
    #[expect(clippy::as_conversions, reason = "intentional")]
    fn type_former() -> TypeFormer<Self> {
        TypeFormer::Literal(Literal {
            deserialize: |s| NonZero::new(s.parse().ok()?),
            generate: |prng| loop {
                let u: u32 = arbitrary_nonzero_unsigned!(u32, prng);
                if let Ok(c) = char::try_from(u) {
                    // SAFETY: Internals of `arbitrary_nonzero_unsigned`
                    // prevent `0` (see above).
                    return unsafe { NonZero::new_unchecked(c) };
                }
            },
            serialize: |value: &Self| value.get().to_string(),
            shrink: |c| {
                Box::new(
                    shrink(c.get() as u32).filter_map(|u| NonZero::new(char::try_from(u).ok()?)),
                )
            },
        })
    }

    #[inline]
    fn visit_deep<V: Construct>(&self) -> impl Iterator<Item = V> {
        visit_self(self).chain(visit_self_owned(self.get()))
    }
}