add uts for element_at

apache · Dec 6, 2023 · 7708c19 · 7708c19
1 parent 72186b8
commit 7708c19
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/...ds-clickhouse/src/test/scala/io/glutenproject/execution/GlutenFunctionValidateSuite.scala b/...ds-clickhouse/src/test/scala/io/glutenproject/execution/GlutenFunctionValidateSuite.scala
@@ -485,4 +485,31 @@ class GlutenFunctionValidateSuite extends GlutenClickHouseWholeStageTransformerS
       "select * from date_table where to_date(from_unixtime(ts)) = '2019-01-01'",
       noFallBack = true) { _ => }
   }
+
+  test("test element_at function") {
+    withSQLConf(
+      SQLConf.OPTIMIZER_EXCLUDED_RULES.key ->
+        (ConstantFolding.ruleName + "," + NullPropagation.ruleName)) {
+      // input type is array<array<int>>
+      runQueryAndCompare(
+        "SELECT array(array(1,2,3), array(4,5,6))[1], " +
+          "array(array(id,id+1,id+2), array(id+3,id+4,id+5)) from range(100)",
+        noFallBack = false
+      )(checkOperatorMatch[ProjectExecTransformer])
+
+      // input type is array<array<string>>
+      runQueryAndCompare(
+        "SELECT array(array('1','2','3'), array('4','5','6'))[1], " +
+          "array(array('1','2',cast(id as string)), array('4','5',cast(id as string)))[1] " +
+          "from range(100)",
+        noFallBack = false
+      )(checkOperatorMatch[ProjectExecTransformer])
+
+      // input type is array<map<string, int>>
+      runQueryAndCompare(
+        "SELECT array(map(cast(id as string), id), map(cast(id+1 as string), id+1))[1] " +
+          "from range(100)" noFallBack = false
+      )(checkOperatorMatch[ProjectExecTransformer])
+    }
+  }
 }