diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/src/lib.rs | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/lib/src/lib.rs b/lib/src/lib.rs index e954eb4..ed76cba 100644 --- a/lib/src/lib.rs +++ b/lib/src/lib.rs @@ -82,6 +82,23 @@ impl<T: GenerateRandom> GenerateRandom for Option<T> { } } +impl<T: GenerateRandom, const N: usize> GenerateRandom for [T; N] { + fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self { + use core::mem::MaybeUninit; + + let mut arr: [MaybeUninit<T>; N] = unsafe { MaybeUninit::uninit().assume_init() }; + + for elem in arr.iter_mut() { + *elem = MaybeUninit::new(T::generate_random(rng)); + } + + let ret = unsafe { std::mem::transmute_copy(&arr) }; + std::mem::forget(arr); + + ret + } +} + macro_rules! impl_generate_random_tuple { ( $t0:ident $( $t:ident )* ) => { impl< $t0, $( $t, )* > GenerateRandom for ( $t0, $( $t, )* ) |