add uts

bigo-sg · Dec 5, 2023 · ddf6b62 · ddf6b62
1 parent 862fa93
commit ddf6b62
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/...clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseHiveTableSuite.scala b/...clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseHiveTableSuite.scala
@@ -97,6 +97,7 @@ class GlutenClickHouseHiveTableSuite()
         "spark.sql.warehouse.dir",
         getClass.getResource("/").getPath + "unit-tests-working-home/spark-warehouse")
       .set("spark.hive.exec.dynamic.partition.mode", "nonstrict")
+      .set("spark.gluten.supported.hive.udfs", "my_add")
       .setMaster("local[*]")
   }
 
@@ -1060,4 +1061,14 @@ class GlutenClickHouseHiveTableSuite()
     compareResultsAgainstVanillaSpark(select_sql, compareResult = true, _ => {})
     spark.sql("DROP TABLE test_tbl_3548")
   }
+
+  test("test 'hive udf'") {
+    val jarPath = "backends-clickhouse/src/test/resources/udfs/hive-test-udfs.jar"
+    val jarUrl = s"file://${System.getProperty("user.dir")}/$jarPath"
+    spark.sql(
+      s"CREATE FUNCTION my_add as " +
+        "'org.apache.hadoop.hive.contrib.udf.example.UDFExampleAdd2' USING JAR '$jarUrl'")
+    runQueryAndCompare("select MY_ADD(id, id+1) from range(10)")(
+      checkOperatorMatch[ProjectExecTransformer])
+  }
 }
diff --git a/...ickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHParquetSuite.scala b/...ickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHParquetSuite.scala
@@ -48,7 +48,6 @@ class GlutenClickHouseTPCHParquetSuite extends GlutenClickHouseTPCHAbstractSuite
       .set("spark.sql.autoBroadcastJoinThreshold", "10MB")
       .set("spark.gluten.sql.columnar.backend.ch.use.v2", "false")
       .set("spark.gluten.supported.scala.udfs", "my_add")
-      .set("spark.gluten.supported.hive.udfs", "my_add")
   }
 
   override protected val createNullableTables = true
@@ -1319,16 +1318,6 @@ class GlutenClickHouseTPCHParquetSuite extends GlutenClickHouseTPCHAbstractSuite
       checkOperatorMatch[ProjectExecTransformer])
   }
 
-  ignore("test 'hive udf'") {
-    val jarPath = "backends-clickhouse/src/test/resources/udfs/hive-test-udfs.jar"
-    val jarUrl = s"file://${System.getProperty("user.dir")}/$jarPath"
-    spark.sql(
-      s"CREATE FUNCTION my_add as " +
-        "'org.apache.hadoop.hive.contrib.udf.example.UDFExampleAdd2' USING JAR '$jarUrl'")
-    runQueryAndCompare("select my_add(id, id+1) from range(10)")(
-      checkOperatorMatch[ProjectExecTransformer])
-  }
-
   override protected def runTPCHQuery(
       queryNum: Int,
       tpchQueries: String = tpchQueries,