Merge branch 'rightlib' into mergelibs-241120-1113

1 files changed, 41 insertions, 0 deletions

diff --git a/yql/essentials/tests/s-expressions/suites/Scheme/PartitionByValueAndMap.yql b/yql/essentials/tests/s-expressions/suites/Scheme/PartitionByValueAndMap.yql
new file mode 100644
index 0000000000..bc7222e7e1
--- /dev/null
+++ b/yql/essentials/tests/s-expressions/suites/Scheme/PartitionByValueAndMap.yql
@@ -0,0 +1,41 @@
+(
+(let x (Read! world (DataSource 'yt 'plato) (Key '('table (String 'Input))) '('key 'subkey 'value) '()))
+(let world (Left! x))
+(let yt.plato.Input (Right! x))
+(let select@0:0_input yt.plato.Input)
+(let output (block '(
+  (let select@0:0_output (Map select@0:0_input (lambda '(row) (block '(
+    (let res (Struct))
+    (let res (AddMember res 'foo (Member row 'value)))
+    (let res (AddMember res 'bar (Member row 'key)))
+    (return res)
+)
+))))
+  (let select@0:0_output (PartitionByKey select@0:0_output (lambda '(row) (block '(
+    (let res (Member row 'foo))
+    (return res)
+)
+)) (Void) (Void) (lambda '(stream) (FlatMap stream (lambda '(pair) (block '(
+    (let key (Nth pair '0))
+    (let group (Collect (Nth pair '1)))
+    (let res (FlatMap group (lambda '(row) (block '(
+      (let res (Struct))
+      (let res (AddMember res 'subkey (ToString (Length group))))
+      (let res (AddMember res 'value (Member row 'bar)))
+      (let res (AddMember res 'key (Member row 'foo)))
+      (let res (AsList res))
+      (return res)
+)
+))))
+    (return res)
+)))))
+))
+  (return select@0:0_output)
+)
+))
+(let output (Sort output (Bool 'true) (lambda '(x) (Concat (Member x 'subkey) (Member x 'value)))))
+(let sink (DataSink 'result))
+(let world (Write! world sink (Key) output '()))
+(let world (Commit! world sink))
+(return world)
+)