fix(zk): generate m mod t in padding test

zama-ai · Sep 27, 2024 · 5d522ff · 5d522ff
1 parent 3956f96
commit 5d522ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/tfhe-zk-pok/src/proofs/pke.rs b/tfhe-zk-pok/src/proofs/pke.rs
@@ -1322,7 +1322,7 @@ mod tests {
                 .map(|_| (rng.gen::<u64>() % t) as i64)
                 .collect::<Vec<_>>();
             while tmp.iter().all(|&x| (x as u64) < effective_cleartext_t) {
-                tmp.fill_with(|| rng.gen());
+                tmp.fill_with(|| (rng.gen::<u64>() % t) as i64);
             }
 
             tmp
@@ -1364,9 +1364,6 @@ mod tests {
         let mut metadata = [0u8; METADATA_LEN];
         metadata.fill_with(|| rng.gen::<u8>());
 
-        let mut fake_metadata = [255u8; METADATA_LEN];
-        fake_metadata.fill_with(|| rng.gen::<u8>());
-
         let mut m_roundtrip = vec![0i64; k];
         for i in 0..k {
             let mut dot = 0i128;

diff --git a/tfhe-zk-pok/src/proofs/pke_v2.rs b/tfhe-zk-pok/src/proofs/pke_v2.rs
@@ -2344,7 +2344,7 @@ mod tests {
                 .map(|_| (rng.gen::<u64>() % t) as i64)
                 .collect::<Vec<_>>();
             while tmp.iter().all(|&x| (x as u64) < effective_cleartext_t) {
-                tmp.fill_with(|| rng.gen());
+                tmp.fill_with(|| (rng.gen::<u64>() % t) as i64);
             }
 
             tmp
@@ -2386,9 +2386,6 @@ mod tests {
         let mut metadata = [0u8; METADATA_LEN];
         metadata.fill_with(|| rng.gen::<u8>());
 
-        let mut fake_metadata = [255u8; METADATA_LEN];
-        fake_metadata.fill_with(|| rng.gen::<u8>());
-
         let mut m_roundtrip = vec![0i64; k];
         for i in 0..k {
             let mut dot = 0i128;