Update AutoTuner unit tests to Spark version 3.2.0

Signed-off-by: Partho Sarthi <[email protected]>
NVIDIA · Dec 31, 2024 · 3f31738 · 3f31738
1 parent 5380342
commit 3f31738
Show file tree

Hide file tree

Showing 2 changed files with 117 additions and 88 deletions.
diff --git a/core/src/test/scala/com/nvidia/spark/rapids/tool/tuning/BaseAutoTunerSuite.scala b/core/src/test/scala/com/nvidia/spark/rapids/tool/tuning/BaseAutoTunerSuite.scala
@@ -68,7 +68,10 @@ class AppInfoProviderMockTest(val maxInput: Double,
  */
 abstract class BaseAutoTunerSuite extends FunSuite with BeforeAndAfterEach with Logging {
 
-  val defaultSparkVersion = "3.1.1"
+  // Default Spark version
+  val defaultSparkVersion = "3.2.0"
+  // Default RapidsShuffleManager version
+  val defaultSmVersion: String = defaultSparkVersion.filterNot(_ == '.')
 
   val defaultDataprocProps: mutable.Map[String, String] = {
     mutable.LinkedHashMap[String, String](