add largelist for slt

Author: my-vegetable-has-exploded

datafusion/physical-expr/src/array_expressions.rs

@@ -3094,31 +3094,4 @@ mod tests {
 
         make_array(&[arr1, arr2]).expect("failed to initialize function array")
     }
-
-    #[test]
-    fn test_array_distinct() {
-        // test: from array [[1, 3, 2, 3, 1, 2, 4], [1, 1]] to [[1, 2, 3, 4], [1]]
-        let data = vec![
-            Some(vec![
-                Some(1),
-                Some(3),
-                Some(2),
-                Some(3),
-                Some(1),
-                Some(2),
-                Some(4),
-            ]),
-            Some(vec![Some(1), Some(1)]),
-        ];
-        let excepted = vec![
-            Some(vec![Some(1), Some(2), Some(3), Some(4)]),
-            Some(vec![Some(1)]),
-        ];
-        let input =
-            Arc::new(ListArray::from_iter_primitive::<Int64Type, _, _>(data)) as ArrayRef;
-        let excepted = ListArray::from_iter_primitive::<Int64Type, _, _>(excepted);
-        let result = array_distinct(&[input]).unwrap();
-        let result = as_list_array(&result).unwrap();
-        assert_eq!(result, &excepted);
-    }
 }

datafusion/sqllogictest/test_files/array.slt

@@ -206,6 +206,14 @@ AS VALUES
   (make_array([5,6], [5,6], NULL))
 ;
 
+statement ok
+CREATE TABLE array_distinct_table_1D_large
+AS VALUES
+  (arrow_cast(make_array(1, 1, 2, 2, 3), 'LargeList(Int64)')),
+  (arrow_cast(make_array(1, 2, 3, 4, 5), 'LargeList(Int64)')),
+  (arrow_cast(make_array(3, 5, 3, 3, 3), 'LargeList(Int64)'))
+;
+
 statement ok
 CREATE TABLE array_intersect_table_1D
 AS VALUES
@@ -2691,6 +2699,14 @@ from array_distinct_table_2D;
 [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
 [, [5, 6]]
 
+query ?
+select array_distinct(column1)
+from array_distinct_table_1D_large;
+----
+[1, 2, 3]
+[1, 2, 3, 4, 5]
+[3, 5]
+
 query ???
 select array_intersect(column1, column2),
        array_intersect(column3, column4),