Add test for UDF

xinyunh · xinyunh · commit 9b227f78b1ea · 2015-08-02T22:59:16.000-07:00
diff --git a/src/test/scala/org/apache/spark/sql/hbase/HBaseAdditionalQuerySuite.scala b/src/test/scala/org/apache/spark/sql/hbase/HBaseAdditionalQuerySuite.scala
@@ -182,6 +182,13 @@ class HBaseAdditionalQuerySuite extends TestBase {
     teachers.orderBy(Column("grade").asc, Column("class").asc).show(3)
   }
 
+  test("UDF Test") {
+    def myFilter(date: String) = date contains "_1_2"
+    TestHbase.udf.register("myFilter", myFilter _)
+    val result = TestHbase.sql("Select * from spark_teacher_3key WHERE myFilter(teacher_name)")
+    result.foreach(println)
+  }
+
   test("group test for presplit table with coprocessor but without codegen") {
     aggregationTest()
   }

Original file line number	Diff line number	Diff line change
`@@ -182,6 +182,13 @@ class HBaseAdditionalQuerySuite extends TestBase {`
`182`	`182`	`teachers.orderBy(Column("grade").asc, Column("class").asc).show(3)`
`183`	`183`	`}`
`184`	`184`
	`185`	`+ test("UDF Test") {`
	`186`	`+ def myFilter(date: String) = date contains "_1_2"`
	`187`	`+ TestHbase.udf.register("myFilter", myFilter _)`
	`188`	`+ val result = TestHbase.sql("Select * from spark_teacher_3key WHERE myFilter(teacher_name)")`
	`189`	`+ result.foreach(println)`
	`190`	`+ }`
	`191`	`+`
`185`	`192`	`test("group test for presplit table with coprocessor but without codegen") {`
`186`	`193`	`aggregationTest()`
`187`	`194`	`}`