diff options
| author | Maxim Yurchuk <maxim-yurchuk@ydb.tech> | 2024-11-20 17:37:57 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-20 17:37:57 +0000 |
| commit | f76323e9b295c15751e51e3443aa47a36bee8023 (patch) | |
| tree | 4113c8cad473a33e0f746966e0cf087252fa1d7a /yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql | |
| parent | 753ecb8d410a4cb459c26f3a0082fb2d1724fe63 (diff) | |
| parent | a7b9a6afea2a9d7a7bfac4c5eb4c1a8e60adb9e6 (diff) | |
| download | ydb-f76323e9b295c15751e51e3443aa47a36bee8023.tar.gz | |
Merge pull request #11788 from ydb-platform/mergelibs-241120-1113
Library import 241120-1113
Diffstat (limited to 'yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql')
| -rw-r--r-- | yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql b/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql new file mode 100644 index 0000000000..0f5cd0b0c6 --- /dev/null +++ b/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql @@ -0,0 +1,31 @@ +USE plato; + +$a = (SELECT + skey,count(*) as cnt +FROM Input +GROUP BY Length(key) as skey); + +$udfScript = @@ +def f(input,x): + for i in list(input): + d = i.__dict__ + d["pass"] = x + yield d +@@; + +$udf = Python::f(@@ +(Stream<Struct<skey:Uint32,cnt:Uint64>>,Int32) +-> +Stream<Struct<skey:Uint32,cnt:Uint64,pass:Int32>> +@@, $udfScript); + +$udf2 = Python::f(@@ +(Stream<Struct<skey:Uint32,cnt:Uint64,pass:Int32>>,Int32) +-> +Stream<Struct<skey:Uint32,cnt:Uint64,pass:Int32>> +@@, $udfScript); + + +$b = (PROCESS $a USING $udf($ROWS,1)); + +PROCESS $b USING $udf2($ROWS,2); |