fix bug in any kernel (#744)

jorgecarleitao · Jan 9, 2022 · a4383b1 · a4383b1
1 parent 1c78fe9
commit a4383b1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/compute/boolean.rs b/src/compute/boolean.rs
@@ -194,7 +194,7 @@ pub fn any(array: &BooleanArray) -> bool {
         array.into_iter().any(|v| v == Some(true))
     } else {
         let vals = array.values();
-        vals.null_count() != 0
+        vals.null_count() != vals.len()
     }
 }
 

diff --git a/tests/it/compute/boolean.rs b/tests/it/compute/boolean.rs
@@ -440,4 +440,10 @@ fn test_any_all() {
     let array = BooleanArray::from_iter([true, false, true, true].map(Some));
     assert!(!all(&array));
     assert!(any(&array));
+    let array = BooleanArray::from(&[Some(true)]);
+    assert!(any(&array));
+    assert!(all(&array));
+    let array = BooleanArray::from(&[Some(false)]);
+    assert!(!any(&array));
+    assert!(!all(&array));
 }