address comments

apache · Jan 4, 2018 · df75bff · df75bff
1 parent 661e5d9
commit df75bff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...yst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/FoldablePropagationSuite.scala b/...yst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/FoldablePropagationSuite.scala
@@ -147,8 +147,8 @@ class FoldablePropagationSuite extends PlanTest {
   test("Propagate in expand") {
     val c1 = Literal(1).as('a)
     val c2 = Literal(2).as('b)
-    val a1 = c1.toAttribute.withNullability(true)
-    val a2 = c2.toAttribute.withNullability(true)
+    val a1 = c1.toAttribute.newInstance().withNullability(true)
+    val a2 = c2.toAttribute.newInstance().withNullability(true)
     val expand = Expand(
       Seq(Seq(Literal(null), 'b), Seq('a, Literal(null))),
       Seq(a1, a2),
@@ -164,7 +164,7 @@ class FoldablePropagationSuite extends PlanTest {
 
   test("Propagate above outer join") {
     val left = LocalRelation('a.int).select('a, Literal(1).as('b))
-    val right = LocalRelation('c.int).select('c, Literal(1).as("d"))
+    val right = LocalRelation('c.int).select('c, Literal(1).as('d))
 
     val join = left.join(
       right,