Intermediate changes

commit_hash:4b0cad6620396d4b5422b264f4aa162104d45e8d

1 files changed, 34 insertions, 3 deletions

diff --git a/yql/essentials/udfs/common/python/bindings/py_test_engine.h b/yql/essentials/udfs/common/python/bindings/py_test_engine.h
index ff0a61cbb8a..6809fc61cff 100644
--- a/yql/essentials/udfs/common/python/bindings/py_test_engine.h
+++ b/yql/essentials/udfs/common/python/bindings/py_test_engine.h
@@ -115,6 +115,32 @@ public:
         ToMiniKQLWithArg<TChecker>(type, argValue, script, std::move(checker));
     }
 
+    template <typename FunctionType,
+              typename TMiniKQLValueBuilder,
+              typename TChecker>
+    void UnsafeCall(TMiniKQLValueBuilder&& builder,
+                    const TStringBuf& script,
+                    TChecker&& checker)
+    {
+        TPyObjectPtr function = CompilePythonFunction(script);
+        const auto functionType = GetTypeBuilder().SimpleSignatureType<FunctionType>();
+        NUdf::TCallableTypeInspector inspector(*CastCtx_->PyCtx->TypeInfoHelper, functionType);
+        Y_ENSURE(inspector.GetArgsCount() == 1);
+        const TType* argType = static_cast<const TType*>(inspector.GetArgType(0));
+        NUdf::TUnboxedValue value = builder(argType, GetValueBuilder());
+        TPyObjectPtr pyArgs = ToPyArgs(CastCtx_, functionType, &value, inspector);
+        TPyObjectPtr resultObj = PyObject_CallObject(function.Get(), pyArgs.Get());
+
+        if (!resultObj) {
+            return checker(NUdf::TUnboxedValuePod::Invalid());
+        }
+
+        const auto returnType = inspector.GetReturnType();
+        Y_ENSURE(CastCtx_->PyCtx->TypeInfoHelper->GetTypeKind(returnType) != NUdf::ETypeKind::Stream);
+
+        checker(FromPyObject(CastCtx_, returnType, resultObj.Get()));
+    }
+
     template <typename TMiniKQLValueBuilder>
     TPyObjectPtr ToPython(
                 NUdf::TType* udfType,
@@ -189,9 +215,7 @@ public:
     }
 
 private:
-    TPyObjectPtr RunPythonFunction(
-            const TStringBuf& script, PyObject* args = nullptr)
-    {
+    TPyObjectPtr CompilePythonFunction(const TStringBuf& script) {
         TString filename(TStringBuf("embedded:test.py"));
         TPyObjectPtr code(Py_CompileString(script.data(), filename.data(), Py_file_input));
         if (!code) {
@@ -211,6 +235,13 @@ private:
             PyErr_Print();
             UNIT_FAIL("function 'Test' is not found in module");
         }
+        return function;
+    }
+
+    TPyObjectPtr RunPythonFunction(
+            const TStringBuf& script, PyObject* args = nullptr)
+    {
+        TPyObjectPtr function(CompilePythonFunction(script));
         return PyObject_CallObject(function.Get(), args);
     }