1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#include "ut3/py_test_engine.h"
#include <library/cpp/testing/unittest/registar.h>
using namespace NPython;
Y_UNIT_TEST_SUITE(TPyCastTest) {
Y_UNIT_TEST(FromPyStrToInt) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<i32>(
"def Test():\n"
" return '123a'",
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
}),
yexception, "str");
}
Y_UNIT_TEST(FromPyTupleToLong) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<ui64>(
"def Test():\n"
" return 1, 1",
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
}),
yexception, "tuple");
}
Y_UNIT_TEST(FromPyFuncToString) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<char*>(
"def f():\n"
" return 42\n"
"def Test():\n"
" return f",
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
}),
yexception, "function");
}
Y_UNIT_TEST(FromPyNoneToString) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<char*>(
"def Test():\n"
" return None",
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
}),
yexception, "None");
}
Y_UNIT_TEST(BadFromPythonFloat) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<float>(
"def Test():\n"
" return '3 <dot> 1415926'",
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
Y_UNREACHABLE();
}),
yexception, "Cast error object '3 <dot> 1415926' to Float");
}
#if PY_MAJOR_VERSION >= 3
# define RETVAL "-1"
#else
# define RETVAL "-18446744073709551616L"
#endif
Y_UNIT_TEST(BadFromPythonLong) {
TPythonTestEngine engine;
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<ui64>(
"def Test():\n"
" return " RETVAL,
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
Y_UNREACHABLE();
}),
yexception, "Cast error object " RETVAL " to Long");
}
}
|
