Skip to content

Commit

Permalink
Merge pull request #288 from proptest-rs/const-generic-arbitrary
Browse files Browse the repository at this point in the history
Const generic arbitrary
  • Loading branch information
cameron1024 authored Jan 14, 2023
2 parents b513a32 + 603b12c commit 00f29af
Showing 1 changed file with 12 additions and 15 deletions.
27 changes: 12 additions & 15 deletions proptest/src/arbitrary/arrays.rs
Original file line number Diff line number Diff line change
Expand Up @@ -12,27 +12,24 @@
use crate::arbitrary::{any_with, Arbitrary};
use crate::array::UniformArrayStrategy;

macro_rules! array {
($($n: expr),*) => { $(
impl<A: Arbitrary> Arbitrary for [A; $n] {
type Parameters = A::Parameters;
type Strategy = UniformArrayStrategy<A::Strategy, [A; $n]>;
fn arbitrary_with(args: Self::Parameters) -> Self::Strategy {
let base = any_with::<A>(args);
UniformArrayStrategy::new(base)
}
}
)* };
impl<A: Arbitrary, const N: usize> Arbitrary for [A; N] {
type Parameters = A::Parameters;
type Strategy = UniformArrayStrategy<A::Strategy, [A; N]>;

fn arbitrary_with(args: Self::Parameters) -> Self::Strategy {
let base = any_with::<A>(args);
UniformArrayStrategy::new(base)
}
}

array!(
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
);

#[cfg(test)]
mod test {
no_panic_test!(
array_16 => [u8; 16]
);

no_panic_test!(
array_1024 => [u8; 1024]
);
}

0 comments on commit 00f29af

Please sign in to comment.