Fix unary tests for fhevm-engine

zama-ai · Aug 13, 2024 · 7c91ba3 · 7c91ba3
1 parent d25b730
commit 7c91ba3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/fhevm-engine/src/tests/operators.rs b/fhevm-engine/src/tests/operators.rs
@@ -369,6 +369,10 @@ fn compute_expected_unary_output(inp: &BigInt, op: SupportedFheOperations, bits:
                     let inp: u32 = inp.try_into().unwrap();
                     BigInt::from(inp.not())
                 }
+                64 => {
+                    let inp: u64 = inp.try_into().unwrap();
+                    BigInt::from(inp.not())
+                }
                 other => {
                     panic!("unknown bits: {other}")
                 }
@@ -388,6 +392,10 @@ fn compute_expected_unary_output(inp: &BigInt, op: SupportedFheOperations, bits:
                     let inp: i32 = inp.try_into().unwrap();
                     BigInt::from(-inp as u32)
                 }
+                64 => {
+                    let inp: i64 = inp.try_into().unwrap();
+                    BigInt::from(-inp as u64)
+                }
                 other => {
                     panic!("unknown bits: {other}")
                 }