yt/yql/tests/sql/suites/produce/reduce_with_python_few_keys.yql


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

/* postgres can not */
/* syntax version 1 */
$udfScript = @@
import functools
def Len(val_key, input):
    return {"zuza": {val_key[0] + b"-" + str(val_key[1]).encode('utf-8'): functools.reduce(lambda x,y: x + 1, input, 0)}}
@@;

$udf = Python::Len(Callable<(Tuple<String,Uint32>, Stream<String>)->Struct<zuza:Dict<String, Uint32>>>, $udfScript);

$data = (select Cast(value as uint32) ?? 0 as kk, value as ss, key as val from plato.Input1);

$res = (reduce $data on val, kk using $udf(ss));

select * from $res order by Yql::ToOptional(Yql::DictKeys(zuza));