Fixed lut & added test for lut

backend/src/vec_znx.rs

@@ -111,6 +111,16 @@ impl<D: AsMut<[u8]> + AsRef<[u8]>> VecZnx<D> {
         }
     }
 
+    pub fn rotate(&mut self, k: i64){
+        unsafe{
+            (0..self.cols()).for_each(|i|{
+                (0..self.size()).for_each(|j|{
+                    znx::znx_rotate_inplace_i64(self.n() as u64, k, self.at_mut_ptr(i, j));
+                });
+            })
+        }
+    }
+
     pub fn rsh(&mut self, basek: usize, k: usize, scratch: &mut Scratch) {
         let n: usize = self.n();
         let cols: usize = self.cols();