add ut

apache · Sep 9, 2024 · 171b7a4 · 171b7a4
1 parent 5426936
commit 171b7a4
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/backends-velox/src/test/scala/org/apache/gluten/execution/WindowFunctionsValidateSuite.scala b/backends-velox/src/test/scala/org/apache/gluten/execution/WindowFunctionsValidateSuite.scala
@@ -19,14 +19,26 @@ package org.apache.gluten.execution
 class WindowFunctionsValidateSuite extends FunctionsValidateSuite {
 
   test("lag/lead window function with negative input offset") {
+    runQueryAndCompare(
+      "select l_suppkey,lag(l_orderkey, -2) over" +
+        " (partition by l_suppkey order by l_orderkey) from lineitem") {
+      checkGlutenOperatorMatch[WindowExecTransformer]
+    }
+
+    runQueryAndCompare(
+      "select l_suppkey, lead(l_orderkey, -2) over" +
+        " (partition by l_suppkey order by l_orderkey) from lineitem") {
+      checkGlutenOperatorMatch[WindowExecTransformer]
+    }
+
     runQueryAndCompare(
       "select lag(l_orderkey, -2) over" +
         " (partition by l_suppkey order by l_orderkey) from lineitem") {
       checkGlutenOperatorMatch[WindowExecTransformer]
     }
 
     runQueryAndCompare(
-      "select lead(l_orderkey, -2) over" +
+      "select lag(l_orderkey, 2) over" +
         " (partition by l_suppkey order by l_orderkey) from lineitem") {
       checkGlutenOperatorMatch[WindowExecTransformer]
     }