Support arrays

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, )* )