Add a test for the case of WholeStageCodegen enabled.

apache · Sep 8, 2020 · c506803 · c506803
1 parent fdc858e
commit c506803
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/basicPhysicalOperators.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/basicPhysicalOperators.scala
@@ -396,8 +396,6 @@ case class RangeExec(range: org.apache.spark.sql.catalyst.plans.logical.Range)
     RangeExec(range.canonicalized.asInstanceOf[org.apache.spark.sql.catalyst.plans.logical.Range])
   }
 
-
-
   override def inputRDDs(): Seq[RDD[InternalRow]] = {
     val rdd = if (isEmptyRange) {
       new EmptyRDD[InternalRow](sqlContext.sparkContext)

diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/PlannerSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/PlannerSuite.scala
@@ -996,6 +996,12 @@ class PlannerSuite extends SharedSparkSession with AdaptiveSparkPlanHelper {
   }
 
   test("Change the number of partitions to zero when a range is empty") {
+    withSQLConf(SQLConf.WHOLESTAGE_CODEGEN_ENABLED.key -> "true") {
+      val range = spark.range(1, 1, 1, 1000)
+      val numPartitions = range.rdd.getNumPartitions
+      assert(numPartitions == 0)
+    }
+
     withSQLConf(SQLConf.WHOLESTAGE_CODEGEN_ENABLED.key -> "false") {
       val range = spark.range(1, 1, 1, 1000)
       val numPartitions = range.rdd.getNumPartitions