use consistent row count

Author: andygrove

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastBenchmark.scala

@@ -61,7 +61,7 @@ object CometCastBenchmark extends CometBenchmarkBase {
           case Compatible(notes) =>
             runBenchmarkWithTable(
               s"Running benchmark cast operation from : $LongType to : $toDataType",
-              1024 * 1024 * 10) { v =>
+              1024 * 1024) { v => // 1M rows
               castBenchmark(v, LongType, toDataType, isAnsiMode = ansiMode)
             }
           case Incompatible(notes) => None

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastBooleanBenchmark.scala

@@ -71,7 +71,7 @@ object CometCastBooleanBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc($colName AS BOOLEAN) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate boolean data for boolean-to-other casts
     runBenchmarkWithTable("Boolean to other types casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastNumericToNumericBenchmark.scala

@@ -92,7 +92,7 @@ object CometCastNumericToNumericBenchmark extends CometBenchmarkBase {
   }
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate input data once with all numeric types
     runBenchmarkWithTable("Numeric to Numeric casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastNumericToStringBenchmark.scala

@@ -54,7 +54,7 @@ object CometCastNumericToStringBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc($colName AS STRING) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate input data once with all numeric types
     runBenchmarkWithTable("Numeric to String casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastNumericToTemporalBenchmark.scala

@@ -52,7 +52,7 @@ object CometCastNumericToTemporalBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc(c_long AS TIMESTAMP) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate data once for INT to DATE conversions
     runBenchmarkWithTable("Int to Date casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastStringToTemporalBenchmark.scala

@@ -52,7 +52,7 @@ object CometCastStringToTemporalBenchmark extends CometBenchmarkBase {
       "SELECT TRY_CAST(c1 AS TIMESTAMP) FROM parquetV1Table"))
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 10 // 10M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate date data once with ~10% invalid values
     runBenchmarkWithTable("date data generation", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastTemporalToNumericBenchmark.scala

@@ -56,7 +56,7 @@ object CometCastTemporalToNumericBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc(c_timestamp AS $targetType) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate DATE data once for all date-to-numeric benchmarks
     runBenchmarkWithTable("Date to Numeric casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastTemporalToStringBenchmark.scala

@@ -49,7 +49,7 @@ object CometCastTemporalToStringBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc(c_timestamp AS STRING) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate temporal data once for date benchmarks
     runBenchmarkWithTable("Date to String casts", values) { v =>

spark/src/test/scala/org/apache/spark/sql/benchmark/CometCastTemporalToTemporalBenchmark.scala

@@ -51,7 +51,7 @@ object CometCastTemporalToTemporalBenchmark extends CometBenchmarkBase {
     s"SELECT $castFunc(c_timestamp AS DATE) FROM parquetV1Table")
 
   override def runCometBenchmark(mainArgs: Array[String]): Unit = {
-    val values = 1024 * 1024 * 5 // 5M rows
+    val values = 1024 * 1024 // 1M rows
 
     // Generate DATE data for Date -> Timestamp benchmarks
     runBenchmarkWithTable("Date to Timestamp casts", values) { v =>