diff options
author | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-11-19 14:58:38 +0300 |
---|---|---|
committer | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-11-19 15:16:27 +0300 |
commit | 24521403b1c44303e043ba540c09b1fe991c7474 (patch) | |
tree | 341d1e7206bc7c143d04d2d96f05b6dc0655606d /yql/essentials/tests/sql/suites/optimizers/test_lmap_opts.sql | |
parent | 72b3cd51dc3fb9d16975d353ea82fd85701393cc (diff) | |
download | ydb-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.sql | 47 |
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; |