From 7554a4ac2ef157506415b7756bb6bb8e957174c6 Mon Sep 17 00:00:00 2001
From: Liang-Chi Hsieh <viirya@gmail.com>
Date: Tue, 9 Jul 2024 11:59:56 -0700
Subject: [PATCH] Disable q72-v2.7

---
 .../spark/sql/CometTPCDSQuerySuite.scala      | 47 ++++++++++++++++---
 1 file changed, 40 insertions(+), 7 deletions(-)

diff --git a/spark/src/test/scala/org/apache/spark/sql/CometTPCDSQuerySuite.scala b/spark/src/test/scala/org/apache/spark/sql/CometTPCDSQuerySuite.scala
index 24d1d4412..9bc494446 100644
--- a/spark/src/test/scala/org/apache/spark/sql/CometTPCDSQuerySuite.scala
+++ b/spark/src/test/scala/org/apache/spark/sql/CometTPCDSQuerySuite.scala
@@ -27,10 +27,6 @@ import org.apache.comet.CometConf
 
 class CometTPCDSQuerySuite
     extends {
-      override val excludedTpcdsQueries: Set[String] = Set()
-
-      // This is private in `TPCDSBase` and `excludedTpcdsQueries` is private too.
-      // So we cannot override `excludedTpcdsQueries` to exclude the queries.
       val tpcdsAllQueries: Seq[String] = Seq(
         "q1",
         "q2",
@@ -112,6 +108,7 @@ class CometTPCDSQuerySuite
         "q69",
         "q70",
         "q71",
+        // TODO: unknown failure in CI with q72 in https://github.com/apache/datafusion-comet/pull/613.
         // "q72",
         "q73",
         "q74",
@@ -141,9 +138,45 @@ class CometTPCDSQuerySuite
         "q98",
         "q99")
 
-      // TODO: enable the 3 queries after fixing the issues #1358.
-      override val tpcdsQueries: Seq[String] =
-        tpcdsAllQueries.filterNot(excludedTpcdsQueries.contains)
+      val tpcdsAllQueriesV2_7_0 = Seq(
+        "q5a",
+        "q6",
+        "q10a",
+        "q11",
+        "q12",
+        "q14",
+        "q14a",
+        "q18a",
+        "q20",
+        "q22",
+        "q22a",
+        "q24",
+        "q27a",
+        "q34",
+        "q35",
+        "q35a",
+        "q36a",
+        "q47",
+        "q49",
+        "q51a",
+        "q57",
+        "q64",
+        "q67a",
+        "q70a",
+        // TODO: unknown failure in CI with q72-v2.7 in
+        // https://github.com/apache/datafusion-comet/pull/613.
+        "q72",
+        "q74",
+        "q75",
+        "q77a",
+        "q78",
+        "q80a",
+        "q86a",
+        "q98")
+
+      override val tpcdsQueries: Seq[String] = tpcdsAllQueries
+
+      override val tpcdsQueriesV2_7_0: Seq[String] = tpcdsAllQueriesV2_7_0
     }
     with CometTPCDSQueryTestSuite
     with ShimCometTPCDSQuerySuite {