Fix test.

apache · Oct 2, 2019 · c138227 · c138227
1 parent 06541fa
commit c138227
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...st/src/test/scala/org/apache/spark/sql/catalyst/expressions/UnsafeRowConverterSuite.scala b/...st/src/test/scala/org/apache/spark/sql/catalyst/expressions/UnsafeRowConverterSuite.scala
@@ -604,6 +604,8 @@ class UnsafeRowConverterSuite extends SparkFunSuite with Matchers with PlanTestB
     val udt = new TestUDT.MyDenseVectorUDT()
     val udtRow = InternalRow.fromSeq(Seq(udt.serialize(vector)))
     val fields5 = Array[DataType](udt)
-    assert(convertBackToInternalRow(udtRow, fields5) === udtRow)
+    val safeProj = SafeProjection.create(fields5)
+    val safeRow = safeProj(udtRow)
+    assert(convertBackToInternalRow(udtRow, fields5) === safeRow)
   }
 }