@@ -530,10 +530,10 @@ pub mod tests {
530530 use ark_sponge:: poseidon:: { PoseidonParameters , PoseidonSponge } ;
531531 use ark_std:: rand:: {
532532 distributions:: { Distribution , Uniform } ,
533- rngs:: StdRng ,
534- Rng ,
533+ Rng , SeedableRng ,
535534 } ;
536535 use ark_std:: test_rng;
536+ use rand_chacha:: ChaCha20Rng ;
537537
538538 struct TestInfo < F : PrimeField , P : Polynomial < F > , S : CryptographicSponge > {
539539 num_iters : usize ,
@@ -544,14 +544,14 @@ pub mod tests {
544544 enforce_degree_bounds : bool ,
545545 max_num_queries : usize ,
546546 num_equations : Option < usize > ,
547- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
548- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
547+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
548+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
549549 sponge : fn ( ) -> S ,
550550 }
551551
552552 pub fn bad_degree_bound_test < F , P , PC , S > (
553- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
554- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
553+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
554+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
555555 sponge : fn ( ) -> S ,
556556 ) -> Result < ( ) , PC :: Error >
557557 where
@@ -566,7 +566,7 @@ pub mod tests {
566566 ] ;
567567
568568 for challenge_gen in challenge_generators {
569- let rng = & mut test_rng ( ) ;
569+ let rng = & mut ChaCha20Rng :: from_rng ( test_rng ( ) ) . unwrap ( ) ;
570570 let max_degree = 100 ;
571571 let pp = PC :: setup ( max_degree, None , rng) ?;
572572 for _ in 0 ..10 {
@@ -674,7 +674,7 @@ pub mod tests {
674674 ] ;
675675
676676 for challenge_gen in challenge_gens {
677- let rng = & mut test_rng ( ) ;
677+ let rng = & mut ChaCha20Rng :: from_rng ( test_rng ( ) ) . unwrap ( ) ;
678678 // If testing multivariate polynomials, make the max degree lower
679679 let max_degree = match num_vars {
680680 Some ( _) => max_degree. unwrap_or ( Uniform :: from ( 2 ..=10 ) . sample ( rng) ) ,
@@ -819,7 +819,7 @@ pub mod tests {
819819 ] ;
820820
821821 for challenge_gen in challenge_gens {
822- let rng = & mut test_rng ( ) ;
822+ let rng = & mut ChaCha20Rng :: from_rng ( test_rng ( ) ) . unwrap ( ) ;
823823 // If testing multivariate polynomials, make the max degree lower
824824 let max_degree = match num_vars {
825825 Some ( _) => max_degree. unwrap_or ( Uniform :: from ( 2 ..=10 ) . sample ( rng) ) ,
@@ -979,8 +979,8 @@ pub mod tests {
979979
980980 pub fn single_poly_test < F , P , PC , S > (
981981 num_vars : Option < usize > ,
982- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
983- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
982+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
983+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
984984 sponge : fn ( ) -> S ,
985985 ) -> Result < ( ) , PC :: Error >
986986 where
@@ -1006,8 +1006,8 @@ pub mod tests {
10061006 }
10071007
10081008 pub fn linear_poly_degree_bound_test < F , P , PC , S > (
1009- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1010- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1009+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1010+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
10111011 sponge : fn ( ) -> S ,
10121012 ) -> Result < ( ) , PC :: Error >
10131013 where
@@ -1033,8 +1033,8 @@ pub mod tests {
10331033 }
10341034
10351035 pub fn single_poly_degree_bound_test < F , P , PC , S > (
1036- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1037- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1036+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1037+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
10381038 sponge : fn ( ) -> S ,
10391039 ) -> Result < ( ) , PC :: Error >
10401040 where
@@ -1060,8 +1060,8 @@ pub mod tests {
10601060 }
10611061
10621062 pub fn quadratic_poly_degree_bound_multiple_queries_test < F , P , PC , S > (
1063- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1064- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1063+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1064+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
10651065 sponge : fn ( ) -> S ,
10661066 ) -> Result < ( ) , PC :: Error >
10671067 where
@@ -1087,8 +1087,8 @@ pub mod tests {
10871087 }
10881088
10891089 pub fn single_poly_degree_bound_multiple_queries_test < F , P , PC , S > (
1090- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1091- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1090+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1091+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
10921092 sponge : fn ( ) -> S ,
10931093 ) -> Result < ( ) , PC :: Error >
10941094 where
@@ -1114,8 +1114,8 @@ pub mod tests {
11141114 }
11151115
11161116 pub fn two_polys_degree_bound_single_query_test < F , P , PC , S > (
1117- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1118- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1117+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1118+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
11191119 sponge : fn ( ) -> S ,
11201120 ) -> Result < ( ) , PC :: Error >
11211121 where
@@ -1142,8 +1142,8 @@ pub mod tests {
11421142
11431143 pub fn full_end_to_end_test < F , P , PC , S > (
11441144 num_vars : Option < usize > ,
1145- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1146- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1145+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1146+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
11471147 sponge : fn ( ) -> S ,
11481148 ) -> Result < ( ) , PC :: Error >
11491149 where
@@ -1170,8 +1170,8 @@ pub mod tests {
11701170
11711171 pub fn full_end_to_end_equation_test < F , P , PC , S > (
11721172 num_vars : Option < usize > ,
1173- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1174- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1173+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1174+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
11751175 sponge : fn ( ) -> S ,
11761176 ) -> Result < ( ) , PC :: Error >
11771177 where
@@ -1198,8 +1198,8 @@ pub mod tests {
11981198
11991199 pub fn single_equation_test < F , P , PC , S > (
12001200 num_vars : Option < usize > ,
1201- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1202- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1201+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1202+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
12031203 sponge : fn ( ) -> S ,
12041204 ) -> Result < ( ) , PC :: Error >
12051205 where
@@ -1226,8 +1226,8 @@ pub mod tests {
12261226
12271227 pub fn two_equation_test < F , P , PC , S > (
12281228 num_vars : Option < usize > ,
1229- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1230- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1229+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1230+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
12311231 sponge : fn ( ) -> S ,
12321232 ) -> Result < ( ) , PC :: Error >
12331233 where
@@ -1253,8 +1253,8 @@ pub mod tests {
12531253 }
12541254
12551255 pub fn two_equation_degree_bound_test < F , P , PC , S > (
1256- rand_poly : fn ( usize , Option < usize > , & mut StdRng ) -> P ,
1257- rand_point : fn ( Option < usize > , & mut StdRng ) -> P :: Point ,
1256+ rand_poly : fn ( usize , Option < usize > , & mut ChaCha20Rng ) -> P ,
1257+ rand_point : fn ( Option < usize > , & mut ChaCha20Rng ) -> P :: Point ,
12581258 sponge : fn ( ) -> S ,
12591259 ) -> Result < ( ) , PC :: Error >
12601260 where
0 commit comments