@@ -1858,7 +1858,10 @@ mod tests {
18581858 }
18591859
18601860 let input_schema = Arc :: new ( Schema :: new ( vec ! [
1861- Arc :: new( Field :: new( "list" , DataType :: new_list( DataType :: UInt32 , true ) , true ) )
1861+ Arc :: new( Field :: new( "col_1" , DataType :: Utf8 , true ) ) ,
1862+ Arc :: new( Field :: new( "col_2" , DataType :: Utf8 , true ) ) ,
1863+ Arc :: new( Field :: new( "col_3" , DataType :: Utf8 , true ) ) ,
1864+ Arc :: new( Field :: new( "list" , DataType :: new_list( DataType :: UInt32 , true ) , true ) ) ,
18621865 ] ) ) ;
18631866
18641867 let input_list = ListArray :: from_iter_primitive :: < UInt32Type , _ , _ > ( vec ! [
@@ -1875,7 +1878,10 @@ mod tests {
18751878 let batch = RecordBatch :: try_new (
18761879 input_schema,
18771880 vec ! [
1878- Arc :: new( input_list)
1881+ new_null_array( & DataType :: Utf8 , input_list. len( ) ) ,
1882+ new_null_array( & DataType :: Utf8 , input_list. len( ) ) ,
1883+ new_null_array( & DataType :: Utf8 , input_list. len( ) ) ,
1884+ Arc :: new( input_list) ,
18791885 ]
18801886 ) . unwrap ( ) ;
18811887 let schema = batch. schema ( ) ;
@@ -1887,7 +1893,7 @@ mod tests {
18871893 let idx_col: Arc < dyn PhysicalExpr > = Arc :: new ( Column :: new ( "idx" , 0 ) ) ;
18881894 let item_col: Arc < dyn PhysicalExpr > = Arc :: new ( Column :: new ( "item" , 1 ) ) ;
18891895 AllListElementMatchMiniLambda :: new (
1890- Arc :: new ( Column :: new ( "list" , 0 ) ) ,
1896+ Arc :: new ( Column :: new ( "list" , 3 ) ) ,
18911897 create_both_odd_or_even ( & idx_col, & item_col, is_even)
18921898 )
18931899 }
0 commit comments