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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#include "py_test_engine.h"
#include <library/cpp/testing/unittest/registar.h>
#include <util/string/strip.h>
using namespace NPython;
namespace {
template <typename TType>
void TestBadUtf8Encode() {
#if PY_MAJOR_VERSION == 2
// In Python 2, strings can encode single surrogate pairs, so this issue does not occur there.
return;
#endif // PY_MAJOR_VERSION == 2
TPythonTestEngine engine;
constexpr char programToRun[] = R"(
def Test():
return "\uDC00"
)";
constexpr char expectedError[] = R"(
Failed to convert the string to UTF-8 format. Original message is:
UnicodeEncodeError: 'utf-8' codec can't encode character '\udc00' in position 0: surrogates not allowed
)";
UNIT_ASSERT_EXCEPTION_CONTAINS(
engine.ToMiniKQL<TType>(
StripString(TString(programToRun)),
[](const NUdf::TUnboxedValuePod& value) {
Y_UNUSED(value);
}),
yexception, StripString(TString(expectedError)));
}
} // namespace
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");
}
Y_UNIT_TEST(BadFromPythonUtf8) {
TestBadUtf8Encode<NUdf::TUtf8>();
}
Y_UNIT_TEST(BadFromPythonJson) {
TestBadUtf8Encode<NUdf::TJson>();
}
} // Y_UNIT_TEST_SUITE(TPyCastTest)
|
