fix: snowflake test

apache · Jul 16, 2024 · 3c9d48b · 3c9d48b
1 parent 6e2af80
commit 3c9d48b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctions.java b/snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctions.java
@@ -401,6 +401,8 @@ public void test_ST_GeneratePoints() {
     verifySqlSingleRes(
         "select sedona.ST_NumGeometries(sedona.ST_GeneratePoints(sedona.ST_GeomFromWKT('POLYGON ((1 0, 1 1, 2 1, 2 0, 1 0))'), 15))",
         15);
+
+    registerUDF("ST_GeneratePoints", byte[].class, int.class, long.class);
     verifySqlSingleRes(
         "select sedona.ST_NumGeometries(sedona.ST_GeneratePoints(sedona.ST_GeomFromWKT('POLYGON ((1 0, 1 1, 2 1, 2 0, 1 0))'), 15, 100))",
         15);

diff --git a/snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctionsV2.java b/snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctionsV2.java
@@ -383,6 +383,8 @@ public void test_ST_GeneratePoints() {
     verifySqlSingleRes(
         "select sedona.ST_NumGeometries(sedona.ST_GeneratePoints(ST_GeomFromWKT('POLYGON ((1 0, 1 1, 2 1, 2 0, 1 0))'), 15))",
         15);
+
+    registerUDFV2("ST_GeneratePoints", String.class, int.class, long.class);
     verifySqlSingleRes(
         "select sedona.ST_NumGeometries(sedona.ST_GeneratePoints(ST_GeomFromWKT('POLYGON ((1 0, 1 1, 2 1, 2 0, 1 0))'), 15, 100))",
         15);