Skip to content

Commit 8f9d276

Browse files
committed
fix unit tests
1 parent 66008db commit 8f9d276

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

spark/src/test/scala/org/apache/comet/exec/CometAggregateSuite.scala

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -518,13 +518,12 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {
518518
(-0.0.asInstanceOf[Float], 2),
519519
(0.0.asInstanceOf[Float], 3),
520520
(Float.NaN, 4))
521-
withSQLConf(CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "false") {
522-
withParquetTable(data, "tbl", dictionaryEnabled) {
523-
checkSparkAnswer("SELECT SUM(_2), MIN(_2), MAX(_2), _1 FROM tbl GROUP BY _1")
524-
checkSparkAnswer("SELECT MIN(_1), MAX(_1), MIN(_2), MAX(_2) FROM tbl")
525-
checkSparkAnswer("SELECT AVG(_2), _1 FROM tbl GROUP BY _1")
526-
checkSparkAnswer("SELECT AVG(_1), AVG(_2) FROM tbl")
527-
}
521+
withParquetTable(data, "tbl", dictionaryEnabled) {
522+
checkSparkAnswer("SELECT SUM(_2), MIN(_2), MAX(_2), _1 FROM tbl GROUP BY _1")
523+
// FIXME: Add MIN(_1) once https://github.com/apache/datafusion-comet/issues/2448 is fixed
524+
checkSparkAnswer("SELECT MAX(_1), MIN(_2), MAX(_2) FROM tbl")
525+
checkSparkAnswer("SELECT AVG(_2), _1 FROM tbl GROUP BY _1")
526+
checkSparkAnswer("SELECT AVG(_1), AVG(_2) FROM tbl")
528527
}
529528
}
530529
}

0 commit comments

Comments
 (0)