test: use deterministic PRG (#127)

privacy-scaling-explorations · May 13, 2024 · 0e3cd1a · 0e3cd1a
1 parent cfa42a5
commit 0e3cd1a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ot/mpz-ot-core/src/ferret/mod.rs b/ot/mpz-ot-core/src/ferret/mod.rs
@@ -54,6 +54,7 @@ mod tests {
         ideal_mpcot::{IdealMpcot, MpcotMsgForReceiver, MpcotMsgForSender},
     };
     use mpz_core::{lpn::LpnParameters, prg::Prg};
+    use rand::SeedableRng;
 
     const LPN_PARAMETERS_TEST: LpnParameters = LpnParameters {
         n: 9600,
@@ -63,7 +64,7 @@ mod tests {
 
     #[test]
     fn ferret_test() {
-        let mut prg = Prg::new();
+        let mut prg = Prg::from_seed([1u8; 16].into());
         let delta = prg.random_block();
         let mut ideal_cot = IdealCOT::new_with_delta(delta);
         let mut ideal_mpcot = IdealMpcot::init_with_delta(delta);

diff --git a/ot/mpz-ot-core/src/ferret/mpcot/mod.rs b/ot/mpz-ot-core/src/ferret/mpcot/mod.rs
@@ -15,10 +15,11 @@ mod tests {
     };
     use crate::ideal::ideal_spcot::{IdealSpcot, SpcotMsgForReceiver, SpcotMsgForSender};
     use mpz_core::prg::Prg;
+    use rand::SeedableRng;
 
     #[test]
     fn mpcot_general_test() {
-        let mut prg = Prg::new();
+        let mut prg = Prg::from_seed([1u8; 16].into());
         let delta = prg.random_block();
         let mut ideal_spcot = IdealSpcot::new_with_delta(delta);
 
@@ -94,7 +95,7 @@ mod tests {
 
     #[test]
     fn mpcot_regular_test() {
-        let mut prg = Prg::new();
+        let mut prg = Prg::from_seed([2u8; 16].into());
         let delta = prg.random_block();
         let mut ideal_spcot = IdealSpcot::new_with_delta(delta);