Undo part of a SparkPlanTest change in apache#7162 that broke my test.

JoshRosen · Jul 6, 2015 · 7c3c864 · 7c3c864
1 parent 9969c14
commit 7c3c864
Showing 1 changed file with 3 additions and 1 deletion.
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
@@ -301,7 +301,9 @@ object SparkPlanTest {
     val resolvedPlan = TestSQLContext.prepareForExecution.execute(
       outputPlan transform {
         case plan: SparkPlan =>
-          val inputMap = plan.children.flatMap(_.output).map(a => (a.name, a)).toMap
+          val inputMap = plan.children.flatMap(_.output).zipWithIndex.map { case (a, i) =>
+            (a.name, BoundReference(i, a.dataType, a.nullable))
+          }.toMap
           plan.transformExpressions {
             case UnresolvedAttribute(Seq(u)) =>
               inputMap.getOrElse(u,