Publically re-export rand with test_utils

arrow/src/util/bench_util.rs

@@ -19,17 +19,17 @@
 
 use crate::array::*;
 use crate::datatypes::*;
-use crate::util::test_util::seedable_rng;
-use arrow_buffer::{Buffer, IntervalMonthDayNano};
-use half::f16;
-use rand::distr::uniform::SampleUniform;
-use rand::rng;
-use rand::Rng;
-use rand::SeedableRng;
-use rand::{
+use crate::util::rand::distr::uniform::SampleUniform;
+use crate::util::rand::rng;
+use crate::util::rand::Rng;
+use crate::util::rand::SeedableRng;
+use crate::util::rand::{
     distr::{Alphanumeric, Distribution, StandardUniform},
     prelude::StdRng,
 };
+use crate::util::test_util::seedable_rng;
+use arrow_buffer::{Buffer, IntervalMonthDayNano};
+use half::f16;
 use std::ops::Range;
 
 /// Creates an random (but fixed-seeded) array of a given size and null density

arrow/src/util/data_gen.rs

@@ -19,7 +19,7 @@
 
 use std::sync::Arc;
 
-use rand::{
+use crate::util::rand::{
     distr::uniform::{SampleRange, SampleUniform},
     Rng,
 };

arrow/src/util/mod.rs

@@ -30,5 +30,8 @@ pub use arrow_cast::pretty;
 pub mod string_writer;
 #[cfg(any(test, feature = "test_utils"))]
 pub mod test_util;
+/// Public re-export rand crate as it appears in the public API
+#[cfg(any(test, feature = "test_utils"))]
+pub use rand;
 
 pub use arrow_cast::display;

arrow/src/util/test_util.rs

@@ -17,7 +17,7 @@
 
 //! Utils to make testing easier
 
-use rand::{rngs::StdRng, Rng, SeedableRng};
+use crate::util::rand::{rngs::StdRng, Rng, SeedableRng};
 use std::{env, error::Error, fs, io::Write, path::PathBuf};
 
 /// Returns a vector of size `n`, filled with randomly generated bytes.