add ut

zml1206 · Dec 8, 2023 · f295898 · f295898
1 parent a00649e
commit f295898
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameWindowFunctionsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameWindowFunctionsSuite.scala
@@ -1389,6 +1389,17 @@ class DataFrameWindowFunctionsSuite extends QueryTest
               )
             )
 
+            // Both RowFrame and RangeFrame exist
+            checkAnswer(
+              df.withColumn("sum_value1", sum("value")
+                .over(window.rowsBetween(Window.unboundedPreceding, Window.currentRow)))
+              .withColumn("sum_value2", sum("value")
+                .over(window.rangeBetween(Window.unboundedPreceding, Window.currentRow))).limit(1),
+              Seq(
+                Row("a", 4, "", 4, 8)
+              )
+            )
+
             // Choose LimitPushDownThroughWindow instead of WindowGroupLimit if the
             // window function is rank-like and Window partitionSpec is empty.
             val existWindowGroupLimit =