tweak fill_random to support any rng element

rust-rse · Jan 9, 2019 · 938c394 · 938c394
1 parent 08f908e
commit 938c394
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/tests/mod.rs b/src/tests/mod.rs
@@ -1,7 +1,7 @@
 #![allow(dead_code)]
 
 use super::{galois_8, Error, SBSError};
-use rand::{thread_rng, Rng};
+use rand::{self, thread_rng, Rng};
 
 type ReedSolomon = crate::ReedSolomon<galois_8::Field>;
 type ShardByShard<'a> = crate::ShardByShard<'a, galois_8::Field>;
@@ -32,9 +32,11 @@ where
     }
 }
 
-pub fn fill_random(arr: &mut [u8]) {
+pub fn fill_random<T>(arr: &mut [T]) 
+    where rand::distributions::Standard: rand::distributions::Distribution<T>
+{
     for a in arr.iter_mut() {
-        *a = rand::random::<u8>();
+        *a = rand::random::<T>();
     }
 }