Skip to content

Commit 49e4904

Browse files
committed
Revert "removal of the unit test for the now deprecated callUdf"
1 parent bbdeaf3 commit 49e4904

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,15 @@ class DataFrameSuite extends QueryTest {
301301
)
302302
}
303303

304+
test("deprecated callUdf in SQLContext") {
305+
val df = Seq(("id1", 1), ("id2", 4), ("id3", 5)).toDF("id", "value")
306+
val sqlctx = df.sqlContext
307+
sqlctx.udf.register("simpleUdf", (v: Int) => v * v)
308+
checkAnswer(
309+
df.select($"id", callUdf("simpleUdf", $"value")),
310+
Row("id1", 1) :: Row("id2", 16) :: Row("id3", 25) :: Nil)
311+
}
312+
304313
test("callUDF in SQLContext") {
305314
val df = Seq(("id1", 1), ("id2", 4), ("id3", 5)).toDF("id", "value")
306315
val sqlctx = df.sqlContext

0 commit comments

Comments
 (0)