This is an automated email from the ASF dual-hosted git repository.
taiyangli pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-gluten.git
The following commit(s) were added to refs/heads/main by this push:
new 804f08c0c [GLUTEN-6805][CH] support function array_remove/array_repeat
(#7210)
804f08c0c is described below
commit 804f08c0c611ae704529352c84fd54ed012ef2b6
Author: 李扬 <[email protected]>
AuthorDate: Sat Sep 14 10:33:09 2024 +0800
[GLUTEN-6805][CH] support function array_remove/array_repeat (#7210)
* support array_repeat
* support function arrayRemove
* finish dev
* remove logs
* fix failed uts
---
.../org/apache/gluten/utils/CHExpressionUtil.scala | 2 -
.../Parser/scalar_function_parser/arrayExcept.cpp | 8 ++-
.../{arrayExcept.cpp => arrayRemove.cpp} | 72 ++++++++++------------
.../{arrayExcept.cpp => arrayRepeat.cpp} | 69 +++++++++------------
.../utils/clickhouse/ClickHouseTestSettings.scala | 23 +------
.../utils/clickhouse/ClickHouseTestSettings.scala | 24 +-------
.../utils/clickhouse/ClickHouseTestSettings.scala | 25 +-------
.../utils/clickhouse/ClickHouseTestSettings.scala | 25 +-------
8 files changed, 70 insertions(+), 178 deletions(-)
diff --git
a/backends-clickhouse/src/main/scala/org/apache/gluten/utils/CHExpressionUtil.scala
b/backends-clickhouse/src/main/scala/org/apache/gluten/utils/CHExpressionUtil.scala
index 561129f51..3c9fa9888 100644
---
a/backends-clickhouse/src/main/scala/org/apache/gluten/utils/CHExpressionUtil.scala
+++
b/backends-clickhouse/src/main/scala/org/apache/gluten/utils/CHExpressionUtil.scala
@@ -195,8 +195,6 @@ object CHExpressionUtil {
DATE_FORMAT -> DateFormatClassValidator(),
DECODE -> EncodeDecodeValidator(),
ENCODE -> EncodeDecodeValidator(),
- ARRAY_REPEAT -> DefaultValidator(),
- ARRAY_REMOVE -> DefaultValidator(),
DATE_FROM_UNIX_DATE -> DefaultValidator(),
MONOTONICALLY_INCREASING_ID -> DefaultValidator(),
SPARK_PARTITION_ID -> DefaultValidator(),
diff --git a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
index e90fd4070..4accdcac6 100644
--- a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
+++ b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
@@ -50,7 +50,9 @@ public:
/// if (arr1 == null || arr2 == null)
/// return null
/// else
- /// return arrayDistinct(arrayFilter(x -> !has(assumeNotNull(arr2),
x), assumeNotNull(arr1)))
+ /// return arrayDistinctSpark(arrayFilter(x ->
!has(assumeNotNull(arr2), x), assumeNotNull(arr1)))
+ ///
+ /// Note: we should use arrayDistinctSpark instead of arrayDistinct
because of https://github.com/ClickHouse/ClickHouse/issues/69546
const auto * arr1_arg = parsed_args[0];
const auto * arr2_arg = parsed_args[1];
const auto * arr1_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr1_arg});
@@ -85,8 +87,8 @@ public:
// Apply arrayFilter with the lambda function
const auto * array_filter_node = toFunctionNode(actions_dag,
"arrayFilter", {lambda_function, arr1_not_null});
- // Apply arrayDistinct to the result of arrayFilter
- const auto * array_distinct_node = toFunctionNode(actions_dag,
"arrayDistinct", {array_filter_node});
+ // Apply arrayDistinctSpark to the result of arrayFilter
+ const auto * array_distinct_node = toFunctionNode(actions_dag,
"arrayDistinctSpark", {array_filter_node});
/// Return null if any of arr1 or arr2 is null
const auto * arr1_is_null_node = toFunctionNode(actions_dag, "isNull",
{arr1_arg});
diff --git a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayRemove.cpp
similarity index 52%
copy from cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
copy to cpp-ch/local-engine/Parser/scalar_function_parser/arrayRemove.cpp
index e90fd4070..3b5f6dafb 100644
--- a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
+++ b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayRemove.cpp
@@ -30,13 +30,13 @@ extern const int SIZES_OF_COLUMNS_DOESNT_MATCH;
namespace local_engine
{
-class FunctionParserArrayExcept : public FunctionParser
+class FunctionParserArrayRemove : public FunctionParser
{
public:
- FunctionParserArrayExcept(SerializedPlanParser * plan_parser_) :
FunctionParser(plan_parser_) { }
- ~FunctionParserArrayExcept() override = default;
+ FunctionParserArrayRemove(SerializedPlanParser * plan_parser_) :
FunctionParser(plan_parser_) { }
+ ~FunctionParserArrayRemove() override = default;
- static constexpr auto name = "array_except";
+ static constexpr auto name = "array_remove";
String getName() const override { return name; }
const DB::ActionsDAG::Node *
@@ -46,31 +46,31 @@ public:
if (parsed_args.size() != 2)
throw Exception(DB::ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH,
"Function {} requires exactly two arguments", getName());
- /// Parse spark array_except(arr1, arr2)
- /// if (arr1 == null || arr2 == null)
- /// return null
- /// else
- /// return arrayDistinct(arrayFilter(x -> !has(assumeNotNull(arr2),
x), assumeNotNull(arr1)))
- const auto * arr1_arg = parsed_args[0];
- const auto * arr2_arg = parsed_args[1];
- const auto * arr1_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr1_arg});
- const auto * arr2_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr2_arg});
- // std::cout << "actions_dag:" << actions_dag.dumpDAG() << std::endl;
+ /// Parse spark array_remove(arr, elem)
+ /// if (arr == null || elem == null) return null
+ /// else return arrayFilter(x -> x != assumeNotNull(elem),
assumeNotNull(arr))
+ const auto * arr_arg = parsed_args[0];
+ const auto * elem_arg = parsed_args[1];
+ const auto * arr_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr_arg});
+ const auto * elem_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {elem_arg});
+ const auto & arr_not_null_type = assert_cast<const DataTypeArray
&>(*arr_not_null->result_type);
- // Create lambda function x -> !has(arr2, x)
+ /// Create lambda function x -> ifNull(x != assumeNotNull(elem), 1)
+ /// Note that notEquals in CH is not null safe, so we need to wrap it
with ifNull
ActionsDAG lambda_actions_dag;
- const auto * arr2_in_lambda =
&lambda_actions_dag.addInput(arr2_not_null->result_name,
arr2_not_null->result_type);
- const auto & nested_type = assert_cast<const DataTypeArray
&>(*removeNullable(arr1_not_null->result_type)).getNestedType();
- const auto * x_in_lambda = &lambda_actions_dag.addInput("x",
nested_type);
- const auto * has_in_lambda = toFunctionNode(lambda_actions_dag, "has",
{arr2_in_lambda, x_in_lambda});
- const auto * lambda_output = toFunctionNode(lambda_actions_dag, "not",
{has_in_lambda});
+ const auto * x_in_lambda = &lambda_actions_dag.addInput("x",
arr_not_null_type.getNestedType());
+ const auto * elem_in_lambda =
&lambda_actions_dag.addInput(elem_not_null->result_name,
elem_not_null->result_type);
+ const auto * not_equals_in_lambda = toFunctionNode(lambda_actions_dag,
"notEquals", {x_in_lambda, elem_in_lambda});
+ const auto * const_one_in_lambda =
addColumnToActionsDAG(lambda_actions_dag, std::make_shared<DataTypeUInt8>(),
{1});
+ const auto * if_null_in_lambda = toFunctionNode(lambda_actions_dag,
"ifNull", {not_equals_in_lambda, const_one_in_lambda});
+ const auto * lambda_output = if_null_in_lambda;
lambda_actions_dag.getOutputs().push_back(lambda_output);
lambda_actions_dag.removeUnusedActions(Names(1,
lambda_output->result_name));
auto expression_actions_settings =
DB::ExpressionActionsSettings::fromContext(getContext(),
DB::CompileExpressions::yes);
auto lambda_actions =
std::make_shared<DB::ExpressionActions>(std::move(lambda_actions_dag),
expression_actions_settings);
- DB::Names captured_column_names{arr2_in_lambda->result_name};
+ DB::Names captured_column_names{elem_in_lambda->result_name};
NamesAndTypesList lambda_arguments_names_and_types;
lambda_arguments_names_and_types.emplace_back(x_in_lambda->result_name,
x_in_lambda->result_type);
DB::Names required_column_names = lambda_actions->getRequiredColumns();
@@ -80,29 +80,21 @@ public:
lambda_arguments_names_and_types,
lambda_output->result_type,
lambda_output->result_name);
- const auto * lambda_function =
&actions_dag.addFunction(function_capture, {arr2_not_null},
lambda_output->result_name);
+ const auto * lambda_function =
&actions_dag.addFunction(function_capture, {elem_not_null},
lambda_output->result_name);
- // Apply arrayFilter with the lambda function
- const auto * array_filter_node = toFunctionNode(actions_dag,
"arrayFilter", {lambda_function, arr1_not_null});
+ /// Apply arrayFilter with the lambda function
+ const auto * array_filter = toFunctionNode(actions_dag, "arrayFilter",
{lambda_function, arr_not_null});
- // Apply arrayDistinct to the result of arrayFilter
- const auto * array_distinct_node = toFunctionNode(actions_dag,
"arrayDistinct", {array_filter_node});
-
- /// Return null if any of arr1 or arr2 is null
- const auto * arr1_is_null_node = toFunctionNode(actions_dag, "isNull",
{arr1_arg});
- const auto * arr2_is_null_node = toFunctionNode(actions_dag, "isNull",
{arr2_arg});
+ /// Return null if arr or elem is null
+ const auto * arr_is_null = toFunctionNode(actions_dag, "isNull",
{arr_arg});
+ const auto * elem_is_null = toFunctionNode(actions_dag, "isNull",
{elem_arg});
+ const auto * arr_or_elem_is_null = toFunctionNode(actions_dag, "or",
{arr_is_null, elem_is_null});
const auto * null_array_node
- = addColumnToActionsDAG(actions_dag,
std::make_shared<DataTypeNullable>(array_distinct_node->result_type), {});
- const auto * multi_if_node = toFunctionNode(actions_dag, "multiIf", {
- arr1_is_null_node,
- null_array_node,
- arr2_is_null_node,
- null_array_node,
- array_distinct_node,
- });
- return convertNodeTypeIfNeeded(substrait_func, multi_if_node,
actions_dag);
+ = addColumnToActionsDAG(actions_dag,
std::make_shared<DataTypeNullable>(arr_not_null->result_type), {});
+ const auto * if_node = toFunctionNode(actions_dag, "if",
{arr_or_elem_is_null, null_array_node, array_filter});
+ return convertNodeTypeIfNeeded(substrait_func, if_node, actions_dag);
}
};
-static FunctionParserRegister<FunctionParserArrayExcept> register_array_except;
+static FunctionParserRegister<FunctionParserArrayRemove> register_array_remove;
}
diff --git a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayRepeat.cpp
similarity index 52%
copy from cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
copy to cpp-ch/local-engine/Parser/scalar_function_parser/arrayRepeat.cpp
index e90fd4070..8eef0647b 100644
--- a/cpp-ch/local-engine/Parser/scalar_function_parser/arrayExcept.cpp
+++ b/cpp-ch/local-engine/Parser/scalar_function_parser/arrayRepeat.cpp
@@ -1,3 +1,4 @@
+
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
@@ -30,13 +31,13 @@ extern const int SIZES_OF_COLUMNS_DOESNT_MATCH;
namespace local_engine
{
-class FunctionParserArrayExcept : public FunctionParser
+class FunctionParserArrayRepeat : public FunctionParser
{
public:
- FunctionParserArrayExcept(SerializedPlanParser * plan_parser_) :
FunctionParser(plan_parser_) { }
- ~FunctionParserArrayExcept() override = default;
+ FunctionParserArrayRepeat(SerializedPlanParser * plan_parser_) :
FunctionParser(plan_parser_) { }
+ ~FunctionParserArrayRepeat() override = default;
- static constexpr auto name = "array_except";
+ static constexpr auto name = "array_repeat";
String getName() const override { return name; }
const DB::ActionsDAG::Node *
@@ -46,31 +47,29 @@ public:
if (parsed_args.size() != 2)
throw Exception(DB::ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH,
"Function {} requires exactly two arguments", getName());
- /// Parse spark array_except(arr1, arr2)
- /// if (arr1 == null || arr2 == null)
- /// return null
- /// else
- /// return arrayDistinct(arrayFilter(x -> !has(assumeNotNull(arr2),
x), assumeNotNull(arr1)))
- const auto * arr1_arg = parsed_args[0];
- const auto * arr2_arg = parsed_args[1];
- const auto * arr1_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr1_arg});
- const auto * arr2_not_null = toFunctionNode(actions_dag,
"assumeNotNull", {arr2_arg});
- // std::cout << "actions_dag:" << actions_dag.dumpDAG() << std::endl;
+ /// Parse spark array_repeat(elem, n)
+ /// if (n == null) return null
+ /// else return arrayMap(x -> elem, range(greatest(assumeNotNull(n))))
+ const auto * elem_arg = parsed_args[0];
+ const auto * n_arg = parsed_args[1];
+ const auto * n_not_null_arg = toFunctionNode(actions_dag,
"assumeNotNull", {n_arg});
+ const auto * const_zero_node = addColumnToActionsDAG(actions_dag,
n_not_null_arg->result_type, {0});
+ const auto * greatest_node = toFunctionNode(actions_dag, "greatest",
{n_not_null_arg, const_zero_node});
+ const auto * range_node = toFunctionNode(actions_dag, "range",
{greatest_node});
+ const auto & range_type = assert_cast<const DataTypeArray &
>(*removeNullable(range_node->result_type));
- // Create lambda function x -> !has(arr2, x)
+ // Create lambda function x -> elem
ActionsDAG lambda_actions_dag;
- const auto * arr2_in_lambda =
&lambda_actions_dag.addInput(arr2_not_null->result_name,
arr2_not_null->result_type);
- const auto & nested_type = assert_cast<const DataTypeArray
&>(*removeNullable(arr1_not_null->result_type)).getNestedType();
- const auto * x_in_lambda = &lambda_actions_dag.addInput("x",
nested_type);
- const auto * has_in_lambda = toFunctionNode(lambda_actions_dag, "has",
{arr2_in_lambda, x_in_lambda});
- const auto * lambda_output = toFunctionNode(lambda_actions_dag, "not",
{has_in_lambda});
+ const auto * x_in_lambda = &lambda_actions_dag.addInput("x",
range_type.getNestedType());
+ const auto * elem_in_lambda =
&lambda_actions_dag.addInput(elem_arg->result_name, elem_arg->result_type);
+ const auto * lambda_output = elem_in_lambda;
lambda_actions_dag.getOutputs().push_back(lambda_output);
lambda_actions_dag.removeUnusedActions(Names(1,
lambda_output->result_name));
auto expression_actions_settings =
DB::ExpressionActionsSettings::fromContext(getContext(),
DB::CompileExpressions::yes);
auto lambda_actions =
std::make_shared<DB::ExpressionActions>(std::move(lambda_actions_dag),
expression_actions_settings);
- DB::Names captured_column_names{arr2_in_lambda->result_name};
+ DB::Names captured_column_names{elem_in_lambda->result_name};
NamesAndTypesList lambda_arguments_names_and_types;
lambda_arguments_names_and_types.emplace_back(x_in_lambda->result_name,
x_in_lambda->result_type);
DB::Names required_column_names = lambda_actions->getRequiredColumns();
@@ -80,29 +79,19 @@ public:
lambda_arguments_names_and_types,
lambda_output->result_type,
lambda_output->result_name);
- const auto * lambda_function =
&actions_dag.addFunction(function_capture, {arr2_not_null},
lambda_output->result_name);
-
- // Apply arrayFilter with the lambda function
- const auto * array_filter_node = toFunctionNode(actions_dag,
"arrayFilter", {lambda_function, arr1_not_null});
+ const auto * lambda_function =
&actions_dag.addFunction(function_capture, {elem_arg},
lambda_output->result_name);
- // Apply arrayDistinct to the result of arrayFilter
- const auto * array_distinct_node = toFunctionNode(actions_dag,
"arrayDistinct", {array_filter_node});
+ /// Apply arrayMap with the lambda function
+ const auto * array_map_node = toFunctionNode(actions_dag, "arrayMap",
{lambda_function, range_node});
- /// Return null if any of arr1 or arr2 is null
- const auto * arr1_is_null_node = toFunctionNode(actions_dag, "isNull",
{arr1_arg});
- const auto * arr2_is_null_node = toFunctionNode(actions_dag, "isNull",
{arr2_arg});
+ /// Return null if n is null
+ const auto * n_is_null_node = toFunctionNode(actions_dag, "isNull",
{n_arg});
const auto * null_array_node
- = addColumnToActionsDAG(actions_dag,
std::make_shared<DataTypeNullable>(array_distinct_node->result_type), {});
- const auto * multi_if_node = toFunctionNode(actions_dag, "multiIf", {
- arr1_is_null_node,
- null_array_node,
- arr2_is_null_node,
- null_array_node,
- array_distinct_node,
- });
- return convertNodeTypeIfNeeded(substrait_func, multi_if_node,
actions_dag);
+ = addColumnToActionsDAG(actions_dag,
std::make_shared<DataTypeNullable>(array_map_node->result_type), {});
+ const auto * if_node = toFunctionNode(actions_dag, "if",
{n_is_null_node, null_array_node, array_map_node});
+ return convertNodeTypeIfNeeded(substrait_func, if_node, actions_dag);
}
};
-static FunctionParserRegister<FunctionParserArrayExcept> register_array_except;
+static FunctionParserRegister<FunctionParserArrayRepeat> register_array_repeat;
}
diff --git
a/gluten-ut/spark32/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
b/gluten-ut/spark32/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
index a7ccef986..64fcaeea4 100644
---
a/gluten-ut/spark32/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
+++
b/gluten-ut/spark32/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
@@ -653,34 +653,13 @@ class ClickHouseTestSettings extends BackendTestSettings {
.exclude("cast from struct II")
.exclude("cast from struct III")
enableSuite[GlutenCollectionExpressionsSuite]
- .exclude("Array and Map Size")
- .exclude("MapEntries")
- .exclude("Map Concat")
- .exclude("MapFromEntries")
- .exclude("ArraysOverlap")
- .exclude("ArraysZip")
+ .exclude("ArraysZip") // wait for
https://github.com/ClickHouse/ClickHouse/pull/69576
.exclude("Sequence of numbers")
- .exclude("Sequence of timestamps")
- .exclude("Sequence on DST boundaries")
- .exclude("Sequence of dates")
- .exclude("SPARK-37544: Time zone should not affect date sequence with
month interval")
- .exclude("SPARK-35088: Accept ANSI intervals by the Sequence expression")
- .exclude("SPARK-36090: Support TimestampNTZType in expression Sequence")
- .exclude("Sequence with default step")
- .exclude("Reverse")
.exclude("elementAt")
- .exclude("ArrayRepeat")
- .exclude("Array remove")
- .exclude("Array Distinct")
.exclude("Shuffle")
- .exclude("Array Except")
- .exclude("Array Except - null handling")
- .exclude("SPARK-31980: Start and end equal in month range")
- .exclude("SPARK-36639: Start and end equal in month range with a negative
step")
.exclude("SPARK-33386: element_at ArrayIndexOutOfBoundsException")
.exclude("SPARK-33460: element_at NoSuchElementException")
.exclude("SPARK-36753: ArrayExcept should handle duplicated Double.NaN and
Float.Nan")
- .exclude("SPARK-36755: ArraysOverlap hould handle duplicated Double.NaN
and Float.Nan")
.exclude(
"SPARK-36740: ArrayMin/ArrayMax/SortArray should handle NaN greater then
non-NaN value")
.excludeGlutenTest("Shuffle")
diff --git
a/gluten-ut/spark33/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
b/gluten-ut/spark33/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
index 41ae3b00d..f40007957 100644
---
a/gluten-ut/spark33/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
+++
b/gluten-ut/spark33/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
@@ -681,37 +681,15 @@ class ClickHouseTestSettings extends BackendTestSettings {
.exclude("cast from struct II")
.exclude("cast from struct III")
enableSuite[GlutenCollectionExpressionsSuite]
- .exclude("Array and Map Size")
- .exclude("MapEntries")
- .exclude("Map Concat")
- .exclude("MapFromEntries")
- .exclude("ArraysOverlap")
- .exclude("ArraysZip")
+ .exclude("ArraysZip") // wait for
https://github.com/ClickHouse/ClickHouse/pull/69576
.exclude("Sequence of numbers")
- .exclude("Sequence of timestamps")
- .exclude("Sequence on DST boundaries")
- .exclude("Sequence of dates")
- .exclude("SPARK-37544: Time zone should not affect date sequence with
month interval")
- .exclude("SPARK-35088: Accept ANSI intervals by the Sequence expression")
- .exclude("SPARK-36090: Support TimestampNTZType in expression Sequence")
- .exclude("Sequence with default step")
- .exclude("Reverse")
.exclude("elementAt")
- .exclude("ArrayRepeat")
- .exclude("Array remove")
- .exclude("Array Distinct")
.exclude("Shuffle")
- .exclude("Array Except")
- .exclude("Array Except - null handling")
- .exclude("SPARK-31980: Start and end equal in month range")
- .exclude("SPARK-36639: Start and end equal in month range with a negative
step")
.exclude("SPARK-33386: element_at ArrayIndexOutOfBoundsException")
.exclude("SPARK-33460: element_at NoSuchElementException")
.exclude("SPARK-36753: ArrayExcept should handle duplicated Double.NaN and
Float.Nan")
- .exclude("SPARK-36755: ArraysOverlap hould handle duplicated Double.NaN
and Float.Nan")
.exclude(
"SPARK-36740: ArrayMin/ArrayMax/SortArray should handle NaN greater then
non-NaN value")
- .exclude("SPARK-39184: Avoid ArrayIndexOutOfBoundsException when crossing
DST boundary")
.excludeGlutenTest("Shuffle")
enableSuite[GlutenComplexTypeSuite]
.exclude("SPARK-33386: GetArrayItem ArrayIndexOutOfBoundsException")
diff --git
a/gluten-ut/spark34/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
b/gluten-ut/spark34/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
index b199d1dd8..76ca12b0a 100644
---
a/gluten-ut/spark34/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
+++
b/gluten-ut/spark34/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
@@ -570,38 +570,15 @@ class ClickHouseTestSettings extends BackendTestSettings {
.exclude("SPARK-36924: Cast YearMonthIntervalType to IntegralType")
.exclude("SPARK-36924: Cast IntegralType to YearMonthIntervalType")
enableSuite[GlutenCollectionExpressionsSuite]
- .exclude("Array and Map Size")
- .exclude("MapEntries")
- .exclude("Map Concat")
- .exclude("MapFromEntries")
- .exclude("ArraysOverlap")
- .exclude("ArraysZip")
+ .exclude("ArraysZip") // wait for
https://github.com/ClickHouse/ClickHouse/pull/69576
.exclude("Sequence of numbers")
- .exclude("Sequence of timestamps")
- .exclude("Sequence on DST boundaries")
- .exclude("Sequence of dates")
- .exclude("SPARK-37544: Time zone should not affect date sequence with
month interval")
- .exclude("SPARK-35088: Accept ANSI intervals by the Sequence expression")
- .exclude("SPARK-36090: Support TimestampNTZType in expression Sequence")
- .exclude("Sequence with default step")
- .exclude("Reverse")
.exclude("elementAt")
- .exclude("Flatten")
- .exclude("ArrayRepeat")
- .exclude("Array remove")
- .exclude("Array Distinct")
.exclude("Shuffle")
- .exclude("Array Except")
- .exclude("Array Except - null handling")
- .exclude("SPARK-31980: Start and end equal in month range")
- .exclude("SPARK-36639: Start and end equal in month range with a negative
step")
.exclude("SPARK-33386: element_at ArrayIndexOutOfBoundsException")
.exclude("SPARK-33460: element_at NoSuchElementException")
.exclude("SPARK-36753: ArrayExcept should handle duplicated Double.NaN and
Float.Nan")
- .exclude("SPARK-36755: ArraysOverlap hould handle duplicated Double.NaN
and Float.Nan")
.exclude(
"SPARK-36740: ArrayMin/ArrayMax/SortArray should handle NaN greater then
non-NaN value")
- .exclude("SPARK-39184: Avoid ArrayIndexOutOfBoundsException when crossing
DST boundary")
.excludeGlutenTest("Shuffle")
enableSuite[GlutenComplexTypeSuite]
.exclude("SPARK-33386: GetArrayItem ArrayIndexOutOfBoundsException")
diff --git
a/gluten-ut/spark35/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
b/gluten-ut/spark35/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
index f156bbc9c..a3553935a 100644
---
a/gluten-ut/spark35/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
+++
b/gluten-ut/spark35/src/test/scala/org/apache/gluten/utils/clickhouse/ClickHouseTestSettings.scala
@@ -570,38 +570,15 @@ class ClickHouseTestSettings extends BackendTestSettings {
.exclude("SPARK-36924: Cast YearMonthIntervalType to IntegralType")
.exclude("SPARK-36924: Cast IntegralType to YearMonthIntervalType")
enableSuite[GlutenCollectionExpressionsSuite]
- .exclude("Array and Map Size")
- .exclude("MapEntries")
- .exclude("Map Concat")
- .exclude("MapFromEntries")
- .exclude("ArraysOverlap")
- .exclude("ArraysZip")
+ .exclude("ArraysZip") // wait for
https://github.com/ClickHouse/ClickHouse/pull/69576
.exclude("Sequence of numbers")
- .exclude("Sequence of timestamps")
- .exclude("Sequence on DST boundaries")
- .exclude("Sequence of dates")
- .exclude("SPARK-37544: Time zone should not affect date sequence with
month interval")
- .exclude("SPARK-35088: Accept ANSI intervals by the Sequence expression")
- .exclude("SPARK-36090: Support TimestampNTZType in expression Sequence")
- .exclude("Sequence with default step")
- .exclude("Reverse")
.exclude("elementAt")
- .exclude("Flatten")
- .exclude("ArrayRepeat")
- .exclude("Array remove")
- .exclude("Array Distinct")
.exclude("Shuffle")
- .exclude("Array Except")
- .exclude("Array Except - null handling")
- .exclude("SPARK-31980: Start and end equal in month range")
- .exclude("SPARK-36639: Start and end equal in month range with a negative
step")
.exclude("SPARK-33386: element_at ArrayIndexOutOfBoundsException")
.exclude("SPARK-33460: element_at NoSuchElementException")
.exclude("SPARK-36753: ArrayExcept should handle duplicated Double.NaN and
Float.Nan")
- .exclude("SPARK-36755: ArraysOverlap hould handle duplicated Double.NaN
and Float.Nan")
.exclude(
"SPARK-36740: ArrayMin/ArrayMax/SortArray should handle NaN greater then
non-NaN value")
- .exclude("SPARK-39184: Avoid ArrayIndexOutOfBoundsException when crossing
DST boundary")
.excludeGlutenTest("Shuffle")
enableSuite[GlutenComplexTypeSuite]
.exclude("SPARK-33386: GetArrayItem ArrayIndexOutOfBoundsException")
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
