Merge branch 'rightlib' into mergelibs-241120-1113

1 files changed, 22 insertions, 0 deletions

diff --git a/yql/essentials/tests/sql/suites/pg/pg_iterate.sql b/yql/essentials/tests/sql/suites/pg/pg_iterate.sql
new file mode 100644
index 0000000000..2c58f3957b
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/pg/pg_iterate.sql
@@ -0,0 +1,22 @@
+pragma warning("disable","4510");
+$init = ListCreate(Struct<n:Int32>);
+$transform = ($value)->{
+    return ListMap(ListFilter($value, ($r)->($r.n<5)), ($r)->(<|n:$r.n + 1|>));
+};
+
+select * from AS_TABLE(Yql::PgIterateAll($init,$transform)) order by n;
+
+$init = [<|n:1|>];
+$transform = ($value)->{
+    return ListMap(ListFilter($value, ($r)->($r.n<5)), ($r)->(<|n:$r.n + 1|>));
+};
+
+select * from AS_TABLE(Yql::PgIterateAll($init,$transform)) order by n;
+
+$init = [<|n:1|>, <|n:1|>, <|n:2|>];
+$transform = ($value)->{
+    return ListFlatMap($value, ($_r)->([<|n:1|>,<|n:2|>,<|n:2|>]));
+};
+
+select * from AS_TABLE(Yql::PgIterate($init,$transform)) order by n;
+