File tree Expand file tree Collapse file tree
resources/sql-tests/expressions/aggregate
scala/org/apache/comet/exec Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1515-- specific language governing permissions and limitations
1616-- under the License.
1717
18+ -- Config: spark.comet.expression.Corr.allowIncompatible=true
1819-- ConfigMatrix: parquet.enable.dictionary=false,true
1920
2021statement
Original file line number Diff line number Diff line change @@ -24,6 +24,7 @@ import scala.util.Random
2424import org .apache .hadoop .fs .Path
2525import org .apache .spark .sql .{CometTestBase , DataFrame , Row }
2626import org .apache .spark .sql .catalyst .expressions .Cast
27+ import org .apache .spark .sql .catalyst .expressions .aggregate .Corr
2728import org .apache .spark .sql .catalyst .optimizer .EliminateSorts
2829import org .apache .spark .sql .comet .CometHashAggregateExec
2930import org .apache .spark .sql .execution .adaptive .AdaptiveSparkPlanHelper
@@ -1319,7 +1320,9 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {
13191320 }
13201321
13211322 test(" covariance & correlation" ) {
1322- withSQLConf(CometConf .COMET_EXEC_SHUFFLE_ENABLED .key -> " true" ) {
1323+ withSQLConf(
1324+ CometConf .COMET_EXEC_SHUFFLE_ENABLED .key -> " true" ,
1325+ CometConf .getExprAllowIncompatConfigKey(classOf [Corr ]) -> " true" ) {
13231326 Seq (" jvm" , " native" ).foreach { cometShuffleMode =>
13241327 withSQLConf(CometConf .COMET_SHUFFLE_MODE .key -> cometShuffleMode) {
13251328 Seq (true , false ).foreach { dictionary =>
You can’t perform that action at this time.
0 commit comments