better benchmark

sql/core/src/test/scala/org/apache/spark/sql/CollationStringExpressionsSuite.scala

@@ -18,8 +18,8 @@
 package org.apache.spark.sql
 
 import org.apache.spark.SparkConf
-import org.apache.spark.sql.catalyst.expressions.aggregate.Mode
 import org.apache.spark.sql.catalyst.expressions.{ExpressionEvalHelper, Literal, StringTrim, StringTrimLeft, StringTrimRight}
+import org.apache.spark.sql.catalyst.expressions.aggregate.Mode
 import org.apache.spark.sql.catalyst.util.CollationFactory
 import org.apache.spark.sql.internal.SQLConf
 import org.apache.spark.sql.test.SharedSparkSession
@@ -880,7 +880,7 @@ class CollationStringExpressionsSuite
       assert(myMode.eval(buffer).toString.toLowerCase() == t.result.toLowerCase())
     })
   }
-   
+
   test("StringTrim* functions - unit tests for both paths (codegen and eval)") {
     // Without trimString param.
     checkEvaluation(StringTrim(Literal.create( "  asd  ", StringType("UTF8_BINARY"))), "asd")

sql/core/src/test/scala/org/apache/spark/sql/execution/benchmark/CollationBenchmark.scala

@@ -22,7 +22,7 @@ import org.apache.spark.benchmark.{Benchmark, BenchmarkBase}
 import org.apache.spark.sql.catalyst.expressions.Literal
 import org.apache.spark.sql.catalyst.expressions.aggregate.Mode
 import org.apache.spark.sql.catalyst.util.{CollationFactory, CollationSupport}
-import org.apache.spark.sql.types.{IntegerType, StringType}
+import org.apache.spark.sql.types.StringType
 import org.apache.spark.unsafe.types.UTF8String
 import org.apache.spark.util.collection.OpenHashMap
 
@@ -249,7 +249,7 @@ object CollationBenchmark extends CollationBenchmarkBase {
   Lowercase and some repeated strings to test the performance of the collation functions.
    */
   def generateBaseInputStringswithUniqueGroupNumber(n: Long): Seq[UTF8String] = {
-    (0 to n / baseInputStrings.size).flatMap(k => baseInputStrings.map(
+    (0 to n.toInt / baseInputStrings.size).flatMap(k => baseInputStrings.map(
       x => UTF8String.fromString(x + "_" + k)))
       .flatMap(
         x => Seq(x, x.repeat(4), x.repeat(8))) // Variable Lengths...