From c6b0a508a5d8a1bb7f5712d433d37bea9bed41f4 Mon Sep 17 00:00:00 2001 From: Gera Shegalov Date: Mon, 20 Nov 2023 13:44:43 -0800 Subject: [PATCH] Use withGpuSparkSession to customize SparkConf (#9769) Fixes #9768 withGpuSparkSession for (Ansi)CastOpSuite @ (ansi) cast decimal to string --------- Signed-off-by: Gera Shegalov --- .../nvidia/spark/rapids/AnsiCastOpSuite.scala | 16 ++++++++-------- .../com/nvidia/spark/rapids/CastOpSuite.scala | 14 +++++++------- 2 files changed, 15 insertions(+), 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 index d99b79bb0ad..00683d5bc65 100644 --- 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 index 8e7eb00579c..5fe42a6539c 100644 --- 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) + } } } }