Skip to content

Commit

Permalink
Add tests for limit and filter
Browse files Browse the repository at this point in the history
  • Loading branch information
Nicolas Trinquier committed Feb 24, 2019
1 parent 0ca0412 commit 32a2f85
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions rust/arrow/src/compute/array_ops.rs
Original file line number Diff line number Diff line change
Expand Up @@ -450,4 +450,26 @@ mod tests {
assert_eq!(5, min(&a).unwrap());
assert_eq!(9, max(&a).unwrap());
}

#[test]
fn test_filter_array() {
let a = Int32Array::from(vec![5, 6, 7, 8, 9]);
let b = BooleanArray::from(vec![true, false, false, true, false]);
let c = filter(&a, &b).unwrap();
let d = c.as_ref().as_any().downcast_ref::<Int32Array>().unwrap();
assert_eq!(2, d.len());
assert_eq!(5, d.value(0));
assert_eq!(8, d.value(1));
}

#[test]
fn test_limit_array() {
let a = Int32Array::from(vec![5, 6, 7, 8, 9]);
let c = limit(&a, 3).unwrap();
let d = c.as_ref().as_any().downcast_ref::<Int32Array>().unwrap();
assert_eq!(3, d.len());
assert_eq!(5, d.value(0));
assert_eq!(6, d.value(1));
assert_eq!(7, d.value(2));
}
}

0 comments on commit 32a2f85

Please sign in to comment.