Add first_value, last_value, nth_value and ntile window function test…

… cases
snower · Mar 14, 2024 · c230c45 · c230c45
1 parent ebe8e88
commit c230c45
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/examples/window_aggregate/window.sql b/examples/window_aggregate/window.sql
@@ -68,4 +68,8 @@ select order_id, uid, goods_id, row_number() over(order by order_id) as rn, lag(
  lead(order_id, 2, 0) over(order by order_id desc) as lead from `data/orders.json` where status=0 order by order_id;
 
 select order_id, uid, goods_id, row_number() over(order by order_id) + 2 as rn, (lag(order_id, 2, 0) over(order by order_id)
-    + lead(order_id, 2, 0) over(order by order_id desc)) * 3 as lead from `data/orders.json` where status=0 order by order_id;
+    + lead(order_id, 2, 0) over(order by order_id desc)) * 3 as lead from `data/orders.json` where status=0 order by order_id;
+
+select order_id, uid, goods_id, first_value(uid) over w as fuid, last_value(uid) over w as luid,
+    nth_value(uid, 2) over w as nuid, ntile(4) over w as bi
+    from `data/orders.json` window w as (partition by goods_id order by order_id) order by order_id;
diff --git a/tests/test_example_window_aggregate.py b/tests/test_example_window_aggregate.py
@@ -145,4 +145,12 @@ def test_window(self):
                               {'order_id': 3, 'uid': 2, 'goods_id': 2, 'rn': 5, 'lead': 6},
                               {'order_id': 4, 'uid': 1, 'goods_id': 1, 'rn': 6, 'lead': 12},
                               {'order_id': 5, 'uid': 2, 'goods_id': 1, 'rn': 7, 'lead': 18},
-                              {'order_id': 6, 'uid': 2, 'goods_id': 2, 'rn': 8, 'lead': 24}], "data error")
+                              {'order_id': 6, 'uid': 2, 'goods_id': 2, 'rn': 8, 'lead': 24}], "data error")
+
+        self.assert_data(73, [{'order_id': 1, 'uid': 2, 'goods_id': 1, 'fuid': 2, 'luid': 2, 'nuid': 1, 'bi': 1},
+                              {'order_id': 2, 'uid': 1, 'goods_id': 1, 'fuid': 2, 'luid': 2, 'nuid': 1, 'bi': 2},
+                              {'order_id': 3, 'uid': 2, 'goods_id': 2, 'fuid': 2, 'luid': 2, 'nuid': 2, 'bi': 1},
+                              {'order_id': 4, 'uid': 1, 'goods_id': 1, 'fuid': 2, 'luid': 2, 'nuid': 1, 'bi': 3},
+                              {'order_id': 5, 'uid': 2, 'goods_id': 1, 'fuid': 2, 'luid': 2, 'nuid': 1, 'bi': 4},
+                              {'order_id': 6, 'uid': 2, 'goods_id': 2, 'fuid': 2, 'luid': 2, 'nuid': 2, 'bi': 3}],
+                         "data error")