Merge pull request #11788 from ydb-platform/mergelibs-241120-1113

Library import 241120-1113

1 files changed, 23 insertions, 0 deletions

diff --git a/yql/essentials/tests/sql/suites/udf/python_struct.sql b/yql/essentials/tests/sql/suites/udf/python_struct.sql
new file mode 100644
index 0000000000..ed7fc8eeb3
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/udf/python_struct.sql
@@ -0,0 +1,23 @@
+/* postgres can not */
+use plato;
+
+$udf = YQL::@@(block '(
+    (let $udfScript (String '@@@@
+class Person:
+    def __init__(self, name, age):
+        self.name = name
+        self.age = age
+
+def NewPerson(name, age):
+    return Person(name, age)
+@@@@))
+    (let ui32 (DataType 'Uint32))
+    (let str (DataType 'String))
+    (let personType (StructType '('name str) '('age ui32)))
+    (let udfType (CallableType '() '(personType) '(str) '(ui32)))
+    (let udf (ScriptUdf 'Python 'NewPerson udfType $udfScript))
+    (return udf)
+))@@;
+
+$persons = (select $udf(value, 100) as val from Input);
+select val from $persons;