Merge branch 'rightlib' into mergelibs-241120-1113

1 files changed, 53 insertions, 0 deletions

diff --git a/yql/essentials/tests/s-expressions/suites/View/DsvCountPartitionBy.yql b/yql/essentials/tests/s-expressions/suites/View/DsvCountPartitionBy.yql
new file mode 100644
index 0000000000..f186cb7101
--- /dev/null
+++ b/yql/essentials/tests/s-expressions/suites/View/DsvCountPartitionBy.yql
@@ -0,0 +1,53 @@
+(
+(let x (Read! world (DataSource 'yt 'plato) (Key '('table (String 'Input))) '('dict 'key) '()))
+(let world (Left! x))
+(let Input (Right! x))
+(let select1_input12 Input)
+(let output13 (block '(
+  (let select1_output select1_input12)
+  (let select1_output (Filter select1_output (lambda '(row) (block '(
+    (let res (Coalesce (!= (Not (Exists (Lookup (Member row 'dict) (String '"b")))) (Bool 'true)) (Bool 'false)))
+    (return res)
+)
+))))
+  (let select1_output (PartitionByKey select1_output (lambda '(row) (block '(
+    (let res (Member row 'key))
+    (return res)
+)
+)) (Void) (Void) (lambda '(stream) (FlatMap stream (lambda '(pair) (block '(
+    (let key (Nth pair '0))
+    (let group (ForwardList (Nth pair '1)))
+    (let aggr (Fold1 group (lambda '(row) (block '(
+      (let res (Struct))
+      (let res (AddMember res 'key (Member row 'key)))
+      (let res (AddMember res 'Count7 (Convert (Exists (Lookup (Member row 'dict) (String '"a"))) 'Uint64)))
+      (return res)
+)
+)) (lambda '(row state) (block '(
+      (let res (Struct))
+      (let res (AddMember res 'key (Member row 'key)))
+      (let res (AddMember res 'Count7 (OptionalReduce (Member state 'Count7) (Convert (Exists (Lookup (Member row 'dict) (String '"a"))) 'Uint64) (lambda '(a b) (+ a b)))))
+      (return res)
+)
+))))
+    (let res (FlatMap aggr (lambda '(row) (block '(
+      (let res (Struct))
+      (let res (AddMember res 'key (Member row 'key)))
+      (let res (AddMember res 'a (Member row 'Count7)))
+      (let res (AsList res))
+      (return res)
+)
+))))
+    (return res)
+)))
+))))
+  (return select1_output)
+)
+))
+(let result_sink (DataSink 'result))
+(let world (Write! world result_sink (Key) output13 '('('type) '('autoref) '('columns '('key 'a)))))
+(let plato_sink (DataSink 'yt 'plato))
+(let world (Commit! world plato_sink))
+(let world (Commit! world result_sink))
+(return world)
+)