aboutsummaryrefslogtreecommitdiffstats
path: root/yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql
diff options
context:
space:
mode:
authorudovichenko-r <udovichenko-r@yandex-team.com>2024-11-19 14:58:38 +0300
committerudovichenko-r <udovichenko-r@yandex-team.com>2024-11-19 15:16:27 +0300
commit24521403b1c44303e043ba540c09b1fe991c7474 (patch)
tree341d1e7206bc7c143d04d2d96f05b6dc0655606d /yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql
parent72b3cd51dc3fb9d16975d353ea82fd85701393cc (diff)
downloadydb-24521403b1c44303e043ba540c09b1fe991c7474.tar.gz
YQL-19206 Move contrib/ydb/library/yql/tests/sql/suites -> yql/essentials/tests/sql/suites
commit_hash:d0ef1f92b09c94db7c2408f946d2a4c62b603f00
Diffstat (limited to 'yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql')
-rw-r--r--yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql47
1 files changed, 47 insertions, 0 deletions
diff --git a/yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql b/yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql
new file mode 100644
index 0000000000..90c0f8a2a7
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql
@@ -0,0 +1,47 @@
+/* syntax version 1 */
+-- not supported on windows
+/* postgres can not */
+$udfScript = @@
+
+import collections;
+
+def processRows(prefix, rowList, separator):
+ ResultRow = collections.namedtuple("ResultRow", ["Result"]);
+
+ result = [];
+ for row in rowList:
+ resultValue = prefix + row.Name + separator + row.Value;
+ resultItem = ResultRow(Result=resultValue);
+ result.append(resultItem);
+
+ return result;
+@@;
+
+$udf = Python::processRows(
+ Callable<(String, List<Struct<Name:String, Value:String>>, String)->List<Struct<Result:String>>>,
+ $udfScript
+);
+
+$data = (
+ SELECT key AS Name, value AS Value FROM plato.Input0
+);
+
+$prefix = ">>";
+
+$p1 = (
+ PROCESS $data USING $udf($prefix, TableRows(), "=") WHERE Name != "foo"
+);
+
+$p2 = (
+ SELECT Result AS Data FROM $p1
+);
+
+$p3 = (
+ PROCESS $p2 USING Streaming::Process(TableRows(), "grep", AsList("180"))
+);
+
+$p4 = (
+ SELECT Data AS FinalResult FROM $p3
+);
+
+SELECT Avg(Length(FinalResult)) AS AvgResultLength FROM $p4;