add test

apache · Jul 8, 2020 · 8314a04 · 8314a04
1 parent bb71e83
commit 8314a04
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/UDFSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/UDFSuite.scala
@@ -589,6 +589,13 @@ class UDFSuite extends QueryTest with SharedSparkSession {
       Row(null))
   }
 
+  test("SPARK-32154: return null without explicit type is not allowed") {
+    // without explicit type
+    intercept[ClassNotFoundException](spark.udf.register("returnNull", udf((i: Int) => null)))
+    // with explicit type
+    spark.udf.register("returnNull", udf((i: Int) => null.asInstanceOf[Int]))
+  }
+
   test("SPARK-28321 0-args Java UDF should not be called only once") {
     val nonDeterministicJavaUDF = udf(
       new UDF0[Int] {