diff options
| author | Alexander Smirnov <alex@ydb.tech> | 2024-11-20 11:14:58 +0000 |
|---|---|---|
| committer | Alexander Smirnov <alex@ydb.tech> | 2024-11-20 11:14:58 +0000 |
| commit | 31773f157bf8164364649b5f470f52dece0a4317 (patch) | |
| tree | 33d0f7eef45303ab68cf08ab381ce5e5e36c5240 /yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterLReduce.sql | |
| parent | 2c7938962d8689e175574fc1e817c05049f27905 (diff) | |
| parent | eff600952d5dfe17942f38f510a8ac2b203bb3a5 (diff) | |
| download | ydb-31773f157bf8164364649b5f470f52dece0a4317.tar.gz | |
Merge branch 'rightlib' into mergelibs-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); |