Use withGpuSparkSession to customize SparkConf (#9769)

Fixes #9768 withGpuSparkSession for (Ansi)CastOpSuite @ (ansi) cast decimal to string --------- Signed-off-by: Gera Shegalov <[email protected]>
NVIDIA · Nov 20, 2023 · c6b0a50 · c6b0a50
1 parent 0e6fda6
commit c6b0a50
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 15 deletions.
diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/AnsiCastOpSuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/AnsiCastOpSuite.scala
@@ -458,14 +458,14 @@ class AnsiCastOpSuite extends GpuExpressionTestSuite {
   }
 
   test("ansi_cast decimal to string") {
-    val sqlCtx = SparkSession.getActiveSession.get.sqlContext
-    sqlCtx.setConf("spark.sql.legacy.allowNegativeScaleOfDecimal", "true")
-
-    Seq(10, 15, 18).foreach { precision =>
-      Seq(-precision, -5, 0, 5, precision).foreach { scale =>
-        testCastToString(DataTypes.createDecimalType(precision, scale),
-          ansiMode = true,
-          comparisonFunc = None)
+    withGpuSparkSession { spark =>
+      spark.conf.set("spark.sql.legacy.allowNegativeScaleOfDecimal", true.toString)
+      Seq(10, 15, 18).foreach { precision =>
+        Seq(-precision, -5, 0, 5, precision).foreach { scale =>
+          testCastToString(DataTypes.createDecimalType(precision, scale),
+            ansiMode = true,
+            comparisonFunc = None)
+        }
       }
     }
   }

diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/CastOpSuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/CastOpSuite.scala
@@ -445,13 +445,13 @@ class CastOpSuite extends GpuExpressionTestSuite {
   }
 
   test("cast decimal to string") {
-    val sqlCtx = SparkSession.getActiveSession.get.sqlContext
-    sqlCtx.setConf("spark.sql.legacy.allowNegativeScaleOfDecimal", "true")
-
-    Seq(10, 15, 28).foreach { precision =>
-      Seq(-precision, -5, 0, 5, precision).foreach { scale =>
-        testCastToString(DataTypes.createDecimalType(precision, scale),
-          comparisonFunc = None)
+    withGpuSparkSession { spark =>
+      spark.conf.set("spark.sql.legacy.allowNegativeScaleOfDecimal", true.toString)
+      Seq(10, 15, 28).foreach { precision =>
+        Seq(-precision, -5, 0, 5, precision).foreach { scale =>
+          testCastToString(DataTypes.createDecimalType(precision, scale),
+            comparisonFunc = None)
+        }
       }
     }
   }