Manually bind references

marmbrus · Jun 18, 2015 · c60a44d · c60a44d
1 parent 996332a
commit c60a44d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/SortSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/SortSuite.scala
@@ -17,8 +17,8 @@
 
 package org.apache.spark.sql.execution
 
-import org.apache.spark.sql.catalyst.dsl.expressions._
-import org.apache.spark.sql.catalyst.expressions.{Ascending, SortOrder}
+import org.apache.spark.sql.catalyst.expressions.{BoundReference, Ascending, SortOrder}
+import org.apache.spark.sql.types.{IntegerType, StringType}
 
 class SortSuite extends SparkPlanTest {
 
@@ -31,8 +31,8 @@ class SortSuite extends SparkPlanTest {
     )
 
     val sortOrder = Seq(
-      SortOrder('_1, Ascending),
-      SortOrder('_2, Ascending)
+      SortOrder(BoundReference(0, StringType, nullable = false), Ascending),
+      SortOrder(BoundReference(1, IntegerType, nullable = false), Ascending)
     )
 
     checkAnswer(

diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/SparkPlanTest.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/SparkPlanTest.scala
@@ -60,7 +60,7 @@ class SparkPlanTest extends SparkFunSuite {
       planFunction: SparkPlan => SparkPlan,
       expectedAnswer: Seq[A]): Unit = {
     val inputDf = TestSQLContext.createDataFrame(input)
-    val expectedRows = expectedAnswer.map(t => Row.apply(t))
+    val expectedRows = expectedAnswer.map(Row.fromTuple)
     SparkPlanTest.checkAnswer(inputDf, planFunction, expectedRows) match {
       case Some(errorMessage) => fail(errorMessage)
       case None =>