MCPcopy Create free account
hub / github.com/apache/datafusion / test_remove_list_null_values_list

Function test_remove_list_null_values_list

datafusion/common/src/utils/mod.rs:1672–1693  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1670
1671 #[test]
1672 fn test_remove_list_null_values_list() {
1673 let list = Arc::new(create_i32_list(
1674 vec![100, 20, 10, 0, 0, 0, 0, 1, 50],
1675 OffsetBuffer::<i32>::from_lengths(vec![3, 4, 0, 2, 0]),
1676 Some(NullBuffer::from(vec![true, false, false, true, false])),
1677 )) as ArrayRef;
1678
1679 let res = remove_list_null_values(&list).unwrap();
1680 let res = res.as_list::<i32>();
1681
1682 let expected = Arc::new(create_i32_list(
1683 vec![100, 20, 10, 1, 50],
1684 OffsetBuffer::<i32>::from_lengths(vec![3, 0, 0, 2, 0]),
1685 Some(NullBuffer::from(vec![true, false, false, true, false])),
1686 )) as ArrayRef;
1687 let expected = expected.as_list::<i32>();
1688
1689 assert_eq!(res, expected);
1690 // check above skips inner value of nulls
1691 assert_eq!(res.values(), expected.values());
1692 assert_eq!(res.offsets(), expected.offsets());
1693 }
1694
1695 #[test]
1696 fn test_list_array_values_row_number() {

Callers

nothing calls this directly

Calls 3

newFunction · 0.85
remove_list_null_valuesFunction · 0.85
create_i32_listFunction · 0.70

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…