Fixed Failing ToPrettyStringSuite Test for 3.5.0 (#10059)

* Fixed failing ToPrettyStringSuite test * Signing off Signed-off-by: Raza Jafri <[email protected]> --------- Signed-off-by: Raza Jafri <[email protected]>
NVIDIA · Dec 15, 2023 · 97ab6ab · 97ab6ab
1 parent ea43d89
commit 97ab6ab
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/tests/src/test/spark341db/scala/com/nvidia/spark/rapids/ToPrettyStringSuite.scala b/tests/src/test/spark341db/scala/com/nvidia/spark/rapids/ToPrettyStringSuite.scala
@@ -24,7 +24,6 @@ import ai.rapids.cudf.ColumnVector
 import com.nvidia.spark.rapids.Arm._
 import com.nvidia.spark.rapids.GpuColumnVector.GpuColumnarBatchBuilder
 import com.nvidia.spark.rapids.shims.GpuToPrettyString
-import org.scalatest.exceptions.TestFailedException
 
 import org.apache.spark.sql.catalyst.expressions.{BoundReference, NamedExpression, ToPrettyString}
 import org.apache.spark.sql.types.{ArrayType, DataType, DataTypes, DecimalType, MapType, StructField, StructType}
@@ -78,15 +77,11 @@ class ToPrettyStringSuite extends GpuUnitTests {
   }
 
   test("test show() on floats") {
-    // This test is expected to fail until https://github.com/NVIDIA/spark-rapids/issues/4204
-    // is resolved
-    assertThrows[TestFailedException](testDataType(DataTypes.FloatType))
+    testDataType(DataTypes.FloatType)
   }
 
   test("test show() on doubles") {
-    // This test is expected to fail until https://github.com/NVIDIA/spark-rapids/issues/4204
-    // is resolved
-    assertThrows[TestFailedException](testDataType(DataTypes.DoubleType))
+    testDataType(DataTypes.DoubleType)
   }
 
   test("test show() on strings") {