tests

Signed-off-by: Alexander Droste <[email protected]>

Author: 0ax1

vortex-compute/src/expand/buffer.rs

@@ -374,4 +374,40 @@ mod tests {
         let mask = Mask::from_iter([true, true, true, false]);
         (&mut buf).expand(&mask);
     }
+
+    #[test]
+    fn test_expand_into_new_buffer() {
+        let src = [10u32, 20, 30, 40, 50];
+        // Alternating pattern with gaps: [T, F, T, F, T, F, T, F, T, F]
+        let mask = Mask::from_iter([
+            true, false, true, false, true, false, true, false, true, false,
+        ]);
+
+        let result = expand_into_new_buffer(&src, &mask);
+        assert_eq!(result.len(), 10);
+        assert_eq!(result.as_slice()[0], 10);
+        assert_eq!(result.as_slice()[2], 20);
+        assert_eq!(result.as_slice()[4], 30);
+        assert_eq!(result.as_slice()[6], 40);
+        assert_eq!(result.as_slice()[8], 50);
+    }
+
+    #[test]
+    fn test_scatter_into_slice_from() {
+        let src = [1u32, 2, 3, 4, 5];
+        let mut dest = vec![0u32; 8];
+        let mask = Mask::from_iter([true, true, false, true, true, false, true, false]);
+
+        let mask_values = match &mask {
+            Mask::Values(mv) => mv,
+            _ => panic!("Expected Mask::Values"),
+        };
+
+        scatter_into_slice_from(&src, &mut dest, mask_values);
+        assert_eq!(dest[0], 1);
+        assert_eq!(dest[1], 2);
+        assert_eq!(dest[3], 3);
+        assert_eq!(dest[4], 4);
+        assert_eq!(dest[6], 5);
+    }
 }