Update spark/src/test/scala/org/apache/comet/CometCastSuite.scala

Co-authored-by: Liang-Chi Hsieh <viirya@gmail.com>
apache · Apr 27, 2024 · 9a51239 · 9a51239
1 parent b283711
commit 9a51239
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark/src/test/scala/org/apache/comet/CometCastSuite.scala b/spark/src/test/scala/org/apache/comet/CometCastSuite.scala
@@ -120,7 +120,7 @@ class CometCastSuite extends CometTestBase with AdaptiveSparkPlanHelper {
 
   test("cast string to int") {
     // test with hand-picked values
-    castTest(castStringToIntegralInputs.toDF("a"), DataTypes.ByteType)
+    castTest(castStringToIntegralInputs.toDF("a"), DataTypes.IntegerType)
     // fuzz test
     castTest(generateStrings(numericPattern, 5).toDF("a"), DataTypes.ByteType)
   }