tests for BinaryViewVector filter

Signed-off-by: Andrew Duffy <[email protected]>

Author: a10y

vortex-compute/src/filter/vector/binaryview.rs

@@ -42,3 +42,175 @@ where
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use vortex_mask::Mask;
+    use vortex_vector::binaryview::StringVectorMut;
+    use vortex_vector::{VectorMutOps, VectorOps};
+
+    use super::*;
+    use crate::filter::MaskIndices;
+
+    #[test]
+    fn test_filter_binary_view_vector_with_mask() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_values("baz", 1);
+        let vec = vec.freeze();
+
+        let mask = Mask::from_iter([true, false, true, false, true]);
+
+        let filtered = (&vec).filter(&mask);
+
+        assert_eq!(filtered.len(), 3);
+        assert_eq!(filtered.validity().true_count(), 3);
+        assert_eq!(filtered.get_ref(0), Some("hello"));
+        assert_eq!(filtered.get_ref(1), Some("foo"));
+        assert_eq!(filtered.get_ref(2), Some("baz"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_with_mask_indices() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_values("baz", 1);
+        let vec = vec.freeze();
+
+        let indices = unsafe { MaskIndices::new_unchecked(&[0, 2, 4]) };
+
+        let filtered = (&vec).filter(&indices);
+
+        assert_eq!(filtered.len(), 3);
+        assert_eq!(filtered.validity().true_count(), 3);
+        assert_eq!(filtered.get_ref(0), Some("hello"));
+        assert_eq!(filtered.get_ref(1), Some("foo"));
+        assert_eq!(filtered.get_ref(2), Some("baz"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_with_nulls() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_nulls(1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_nulls(1);
+        let vec = vec.freeze();
+
+        let mask = Mask::from_iter([true, true, false, true, false]);
+
+        let filtered = (&vec).filter(&mask);
+
+        assert_eq!(filtered.len(), 3);
+        assert_eq!(filtered.validity().true_count(), 2);
+        assert_eq!(filtered.get_ref(0), Some("hello"));
+        assert_eq!(filtered.get_ref(1), None);
+        assert_eq!(filtered.get_ref(2), Some("bar"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_all_true() {
+        let mut vec = StringVectorMut::with_capacity(3);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        let vec = vec.freeze();
+
+        let mask = Mask::new_true(3);
+
+        let filtered = (&vec).filter(&mask);
+
+        assert_eq!(filtered.len(), 3);
+        assert_eq!(filtered.get_ref(0), Some("hello"));
+        assert_eq!(filtered.get_ref(1), Some("world"));
+        assert_eq!(filtered.get_ref(2), Some("foo"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_all_false() {
+        let mut vec = StringVectorMut::with_capacity(3);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        let vec = vec.freeze();
+
+        let mask = Mask::new_false(3);
+
+        let filtered = (&vec).filter(&mask);
+
+        assert_eq!(filtered.len(), 0);
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_mut_with_mask() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_values("baz", 1);
+
+        let mask = Mask::from_iter([true, false, true, false, true]);
+
+        (&mut vec).filter(&mask);
+
+        assert_eq!(vec.len(), 3);
+        let frozen = vec.freeze();
+        assert_eq!(frozen.len(), 3);
+        assert_eq!(frozen.validity().true_count(), 3);
+        assert_eq!(frozen.get_ref(0), Some("hello"));
+        assert_eq!(frozen.get_ref(1), Some("foo"));
+        assert_eq!(frozen.get_ref(2), Some("baz"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_mut_with_mask_indices() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_values("world", 1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_values("baz", 1);
+
+        let indices = unsafe { MaskIndices::new_unchecked(&[0, 2, 4]) };
+
+        (&mut vec).filter(&indices);
+
+        assert_eq!(vec.len(), 3);
+        let frozen = vec.freeze();
+        assert_eq!(frozen.len(), 3);
+        assert_eq!(frozen.validity().true_count(), 3);
+        assert_eq!(frozen.get_ref(0), Some("hello"));
+        assert_eq!(frozen.get_ref(1), Some("foo"));
+        assert_eq!(frozen.get_ref(2), Some("baz"));
+    }
+
+    #[test]
+    fn test_filter_binary_view_vector_mut_with_nulls() {
+        let mut vec = StringVectorMut::with_capacity(5);
+        vec.append_values("hello", 1);
+        vec.append_nulls(1);
+        vec.append_values("foo", 1);
+        vec.append_values("bar", 1);
+        vec.append_nulls(1);
+
+        let mask = Mask::from_iter([true, true, false, true, false]);
+
+        (&mut vec).filter(&mask);
+
+        assert_eq!(vec.len(), 3);
+        let frozen = vec.freeze();
+        assert_eq!(frozen.len(), 3);
+        assert_eq!(frozen.validity().true_count(), 2);
+        assert_eq!(frozen.get_ref(0), Some("hello"));
+        assert_eq!(frozen.get_ref(1), None);
+        assert_eq!(frozen.get_ref(2), Some("bar"));
+    }
+}