Browse Source

add inv test

Samir Menon 2 years ago
parent
commit
e89da50eb2
1 changed files with 11 additions and 0 deletions
  1. 11 0
      src/ntt.rs

+ 11 - 0
src/ntt.rs

@@ -388,6 +388,17 @@ mod test {
         assert_eq!(v1[2048 + 50], 100);
     }
 
+    #[test]
+    fn ntt_inverse_correct() {
+        let params = get_params();
+        let mut v1 = vec![100; 2*2048];
+        ntt_inverse(&params, v1.as_mut_slice());
+        assert_eq!(v1[0], 100);
+        assert_eq!(v1[2048], 100);
+        assert_eq!(v1[50], 0);
+        assert_eq!(v1[2048 + 50], 0);
+    }
+
     #[test]
     fn ntt_correct() {
         let params = get_params();