| 1964 | } |
| 1965 | |
| 1966 | fn create_test_projection_batch_data(schema: &Schema) -> RecordBatch { |
| 1967 | // set test data for each column |
| 1968 | let array0 = UInt32Array::from(vec![1, 2, 3]); |
| 1969 | let array1 = StringArray::from(vec!["foo", "bar", "baz"]); |
| 1970 | let array2 = BooleanArray::from(vec![true, false, true]); |
| 1971 | |
| 1972 | let mut union_builder = UnionBuilder::new_dense(); |
| 1973 | union_builder.append::<Int32Type>("a", 1).unwrap(); |
| 1974 | union_builder.append::<Float64Type>("b", 10.1).unwrap(); |
| 1975 | union_builder.append_null::<Float64Type>("b").unwrap(); |
| 1976 | let array3 = union_builder.build().unwrap(); |
| 1977 | |
| 1978 | let array4 = NullArray::new(3); |
| 1979 | let array5 = Float64Array::from(vec![Some(1.1), None, Some(3.3)]); |
| 1980 | let array6_values = vec![ |
| 1981 | Some(vec![Some(10), Some(10), Some(10)]), |
| 1982 | Some(vec![Some(20), Some(20), Some(20)]), |
| 1983 | Some(vec![Some(30), Some(30)]), |
| 1984 | ]; |
| 1985 | let array6 = ListArray::from_iter_primitive::<Int32Type, _, _>(array6_values); |
| 1986 | let array7_values = vec![vec![11, 12, 13], vec![22, 23, 24], vec![33, 34, 35]]; |
| 1987 | let array7 = FixedSizeBinaryArray::try_from_iter(array7_values.into_iter()).unwrap(); |
| 1988 | |
| 1989 | let array8_values = ArrayData::builder(DataType::Int32) |
| 1990 | .len(9) |
| 1991 | .add_buffer(Buffer::from_slice_ref([40, 41, 42, 43, 44, 45, 46, 47, 48])) |
| 1992 | .build() |
| 1993 | .unwrap(); |
| 1994 | let array8_data = ArrayData::builder(schema.field(8).data_type().clone()) |
| 1995 | .len(3) |
| 1996 | .add_child_data(array8_values) |
| 1997 | .build() |
| 1998 | .unwrap(); |
| 1999 | let array8 = FixedSizeListArray::from(array8_data); |
| 2000 | |
| 2001 | let array9_id: ArrayRef = Arc::new(Int32Array::from(vec![1001, 1002, 1003])); |
| 2002 | let array9_list: ArrayRef = |
| 2003 | Arc::new(ListArray::from_iter_primitive::<Int8Type, _, _>(vec![ |
| 2004 | Some(vec![Some(-10)]), |
| 2005 | Some(vec![Some(-20), Some(-20), Some(-20)]), |
| 2006 | Some(vec![Some(-30)]), |
| 2007 | ])); |
| 2008 | let array9 = ArrayDataBuilder::new(schema.field(9).data_type().clone()) |
| 2009 | .add_child_data(array9_id.into_data()) |
| 2010 | .add_child_data(array9_list.into_data()) |
| 2011 | .len(3) |
| 2012 | .build() |
| 2013 | .unwrap(); |
| 2014 | let array9 = StructArray::from(array9); |
| 2015 | |
| 2016 | let array10_input = vec![Some(1_i32), None, None]; |
| 2017 | let mut array10_builder = PrimitiveRunBuilder::<Int16Type, Int32Type>::new(); |
| 2018 | array10_builder.extend(array10_input); |
| 2019 | let array10 = array10_builder.finish(); |
| 2020 | |
| 2021 | let array11 = BooleanArray::from(vec![false, false, true]); |
| 2022 | |
| 2023 | let array12_values = StringArray::from(vec!["x", "yy", "zzz"]); |