pikapython/test/class-test.cpp
2022-10-15 13:58:05 +08:00

143 lines
3.7 KiB
C

#include "test_common.h"
#if !PIKA_NANO_ENABLE
TEST(class, class_par) {
/* init */
pikaMemInfo.heapUsedMax = 0;
/* run */
PikaObj* self = newRootObj("pikaMain", New_PikaMain);
__platform_printf("BEGIN\r\n");
obj_run(self, "pi = PikaMath.Math.pi\n");
/* assert */
pika_float pi = obj_getFloat(self, "pi");
#if PIKA_FLOAT_TYPE_DOUBLE
EXPECT_DOUBLE_EQ(pi, 3.141592653589793115997963468544185161590576171875l);
#else
EXPECT_FLOAT_EQ(pi, 3.141592653589793115997963468544185161590576171875l);
#endif
/* deinit */
obj_deinit(self);
EXPECT_EQ(pikaMemNow(), 0);
}
TEST(class, classpar1) {
/* init */
pikaMemInfo.heapUsedMax = 0;
/* run */
PikaObj* self = newRootObj("pikaMain", New_PikaMain);
__platform_printf("BEGIN\r\n");
pikaVM_runSingleFile(self, "../../examples/Class/classpar1.py");
/* assert */
EXPECT_STREQ(log_buff[4], "BEGIN\r\n");
EXPECT_STREQ(log_buff[3], "1\r\n");
EXPECT_STREQ(log_buff[2], "test\r\n");
EXPECT_STREQ(log_buff[1], "2\r\n");
EXPECT_STREQ(log_buff[0], "pewq\r\n");
/* deinit */
obj_deinit(self);
EXPECT_EQ(pikaMemNow(), 0);
}
TEST(class, static_method) {
/* init */
pikaMemInfo.heapUsedMax = 0;
/* run */
PikaObj* self = newRootObj("pikaMain", New_PikaMain);
__platform_printf("BEGIN\r\n");
obj_run(self,
"import PikaMath\n"
"a = PikaMath.Operator.plusInt(10, 20)\n");
/* collect */
int a = obj_getInt(self, "a");
/* assert */
EXPECT_EQ(a, 30);
/* deinit */
obj_deinit(self);
EXPECT_EQ(pikaMemNow(), 0);
}
#endif
TEST(class, dir_) {
/* init */
pikaMemInfo.heapUsedMax = 0;
/* run */
PikaObj* self = newRootObj("pikaMain", New_PikaMain);
__platform_printf("BEGIN\r\n");
obj_run(self,
"class T:\n"
" def a(self):\n"
" return 1\n"
"t = T()\n"
"l = dir(t)\n"
"print(l)\n");
/* collect */
/* assert */
/* deinit */
obj_deinit(self);
EXPECT_EQ(pikaMemNow(), 0);
}
extern "C" {
void PikaStdLib_SysObj_intMethod(PikaObj* self, Args* args);
void PikaStdLib_SysObj_floatMethod(PikaObj* self, Args* args);
const MethodProp floatMethod = {
.ptr = (void*)PikaStdLib_SysObj_floatMethod,
.bytecode_frame = NULL,
.def_context = NULL,
.declareation = "float(arg)",
.type_list = "arg",
.name = "float",
};
const MethodProp intMethod = {
.ptr = (void*)PikaStdLib_SysObj_intMethod,
.bytecode_frame = NULL,
.def_context = NULL,
.declareation = "int(arg)",
.type_list = "arg",
.name = "int",
};
const Arg methods[] = {
{._ = {.buffer = (uint8_t*)&floatMethod},
.size = sizeof(MethodProp),
#if PIKA_ARG_CACHE_ENABLE
.heap_size = 0,
#endif
.type = ARG_TYPE_METHOD_NATIVE,
.flag = 0,
.name_hash = 259121563},
{._ = {.buffer = (uint8_t*)&intMethod},
.size = sizeof(MethodProp),
#if PIKA_ARG_CACHE_ENABLE
.heap_size = 0,
#endif
.type = ARG_TYPE_METHOD_NATIVE,
.flag = 0,
.name_hash = hash_time33("int")},
};
const NativeProperty NativePropertyProp = {
.super = &TinyObj_prop,
.methodGroup = methods,
.methodGroupCount = sizeof(methods) / sizeof(methods[0]),
};
PikaObj* New_NativeMethodBase(Args* args) {
PikaObj* self = New_TinyObj(NULL);
obj_setPtr(self, "@p", (void*)&NativePropertyProp);
return self;
}
}
TEST(class, native_class1) {
PikaObj* native_obj = newNormalObj(New_NativeMethodBase);
obj_run(native_obj,
"int(123.0000)\n"
"float(123)\n");
obj_deinit(native_obj);
EXPECT_STREQ(log_buff[1], "123\r\n");
EXPECT_STREQ(log_buff[0], "123.000000\r\n");
EXPECT_EQ(pikaMemNow(), 0);
}