debug for testcases

worldcoin · Jan 22, 2025 · edd980b · edd980b
1 parent 7e415c9
commit edd980b
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 8 deletions.
diff --git a/iris-mpc-gpu/tests/bucket_threshold.rs b/iris-mpc-gpu/tests/bucket_threshold.rs
@@ -110,7 +110,7 @@ mod bucket_threshold_test {
         let mod_ = 1u64 << (16 + B_BITS);
         let mut res = Vec::with_capacity(code_input.len());
         for (c, m) in code_input.into_iter().zip(mask_input) {
-            let r = ((m as u64) * A - ((c as u64) << B_BITS) - 1) % mod_;
+            let r = (((m as u64) * A).wrapping_sub(((c as u64) << B_BITS).wrapping_sub(1))) % mod_;
             let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
             res.push(msb)
         }

diff --git a/iris-mpc-gpu/tests/buckets.rs b/iris-mpc-gpu/tests/buckets.rs
@@ -101,7 +101,8 @@ mod buckets_test {
 
             let mut count = 0;
             for (c, m) in code_input.iter().cloned().zip(mask_input.iter().cloned()) {
-                let r = ((m as u64) * a - ((c as u64) << B_BITS) - 1) % mod_;
+                let r =
+                    (((m as u64) * a).wrapping_sub(((c as u64) << B_BITS).wrapping_sub(1))) % mod_;
                 let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
                 count += msb as u32;
             }

diff --git a/iris-mpc-gpu/tests/one_bucket.rs b/iris-mpc-gpu/tests/one_bucket.rs
@@ -47,7 +47,7 @@ mod one_bucket_test {
     fn rep_share<R: Rng>(value: u16, rng: &mut R) -> (u16, u16, u16) {
         let a = rng.gen();
         let b = rng.gen();
-        let c= value.wrapping_sub(a).wrapping_sub(b);
+        let c = value.wrapping_sub(a).wrapping_sub(b);
 
         (a, b, c)
     }
@@ -94,7 +94,7 @@ mod one_bucket_test {
         let mod_ = 1u64 << (16 + B_BITS);
         let mut count = 0;
         for (c, m) in code_input.into_iter().zip(mask_input) {
-            let r = ((m as u64) * A - ((c as u64) << B_BITS) - 1) % mod_;
+            let r = (((m as u64) * A).wrapping_sub(((c as u64) << B_BITS).wrapping_sub(1))) % mod_;
             let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
             count += msb as u32;
         }

diff --git a/iris-mpc-gpu/tests/threshold.rs b/iris-mpc-gpu/tests/threshold.rs
@@ -48,7 +48,7 @@ mod threshold_test {
     fn rep_share<R: Rng>(value: u16, rng: &mut R) -> (u16, u16, u16) {
         let a = rng.gen();
         let b = rng.gen();
-        let c= value.wrapping_sub(a).wrapping_sub(b);
+        let c = value.wrapping_sub(a).wrapping_sub(b);
 
         (a, b, c)
     }
@@ -110,7 +110,7 @@ mod threshold_test {
         let mod_ = 1u64 << (16 + B_BITS);
         let mut res = Vec::with_capacity(code_input.len());
         for (c, m) in code_input.into_iter().zip(mask_input) {
-            let r = ((m as u64) * A - ((c as u64) << B_BITS) - 1) % mod_;
+            let r = (((m as u64) * A).wrapping_sub(((c as u64) << B_BITS).wrapping_sub(1))) % mod_;
             let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
             res.push(msb)
         }

diff --git a/iris-mpc-gpu/tests/threshold_and_or_tree.rs b/iris-mpc-gpu/tests/threshold_and_or_tree.rs
@@ -46,7 +46,7 @@ mod test_threshold_and_or_tree_test {
     fn rep_share<R: Rng>(value: u16, rng: &mut R) -> (u16, u16, u16) {
         let a = rng.gen();
         let b = rng.gen();
-        let c= value.wrapping_sub(a).wrapping_sub(b);
+        let c = value.wrapping_sub(a).wrapping_sub(b);
 
         (a, b, c)
     }
@@ -93,7 +93,7 @@ mod test_threshold_and_or_tree_test {
         let mod_ = 1u64 << (16 + B_BITS);
         let mut res = false;
         for (c, m) in code_input.into_iter().zip(mask_input) {
-            let r = ((m as u64) * A - ((c as u64) << B_BITS) - 1) % mod_;
+            let r = (((m as u64) * A).wrapping_sub(((c as u64) << B_BITS).wrapping_sub(1))) % mod_;
             let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
             res |= msb;
         }