add test in SQLMetricsSuite

sameeragarwal · sameeragarwal · commit fa7c991a66d8 · 2016-02-24T18:07:47.000-08:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/metric/SQLMetricsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/metric/SQLMetricsSuite.scala
@@ -153,6 +153,13 @@ class SQLMetricsSuite extends SparkFunSuite with SharedSQLContext {
     )
   }
 
+  test("Sort metrics") {
+    // Assume the execution plan is
+    // WholeStageCodegen(nodeId = 0, Range(nodeId = 2) -> Sort(nodeId = 1))
+    val df = sqlContext.range(10).sort('id)
+    testSparkPlanMetrics(df, 2, Map.empty)
+  }
+
   test("SortMergeJoin metrics") {
     // Because SortMergeJoin may skip different rows if the number of partitions is different, this
     // test should use the deterministic number of partitions.

Original file line number	Diff line number	Diff line change
`@@ -153,6 +153,13 @@ class SQLMetricsSuite extends SparkFunSuite with SharedSQLContext {`
`153`	`153`	`)`
`154`	`154`	`}`
`155`	`155`
	`156`	`+ test("Sort metrics") {`
	`157`	`+ // Assume the execution plan is`
	`158`	`+ // WholeStageCodegen(nodeId = 0, Range(nodeId = 2) -> Sort(nodeId = 1))`
	`159`	`+ val df = sqlContext.range(10).sort('id)`
	`160`	`+ testSparkPlanMetrics(df, 2, Map.empty)`
	`161`	`+ }`
	`162`	`+`
`156`	`163`	`test("SortMergeJoin metrics") {`
`157`	`164`	`// Because SortMergeJoin may skip different rows if the number of partitions is different, this`
`158`	`165`	`// test should use the deterministic number of partitions.`