360 lines
14 KiB
Python
360 lines
14 KiB
Python
import t, c
|
|
from t import CInt, CLong, CDouble, CPtr, CSizeT
|
|
from stdio import printf
|
|
import cpython
|
|
import testcheck
|
|
|
|
|
|
def test_init_finalize():
|
|
testcheck.section("Py_Initialize / Py_Finalize")
|
|
cpython.Py_Initialize()
|
|
testcheck.check(cpython.Py_IsInitialized() != 0, "Py_Initialize succeeded", "Py_Initialize succeeded")
|
|
cpython.Py_Finalize()
|
|
testcheck.check(cpython.Py_IsInitialized() == 0, "Py_Finalize succeeded", "Py_Finalize succeeded")
|
|
|
|
|
|
def test_long_operations():
|
|
testcheck.section("PyLong operations")
|
|
cpython.Py_Initialize()
|
|
|
|
# 创建整数
|
|
py_num: cpython.PyObject | CPtr = cpython.PyLong_FromLong(12345)
|
|
testcheck.check(py_num != 0, "PyLong_FromLong != NULL", "PyLong_FromLong != NULL")
|
|
|
|
# 整数转字符串
|
|
py_str: cpython.PyObject | CPtr = cpython.PyObject_Str(py_num)
|
|
testcheck.check(py_str != 0, "PyObject_Str != NULL", "PyObject_Str != NULL")
|
|
|
|
# 获取 C 字符串
|
|
c_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_str)
|
|
testcheck.check(c_str != 0, "PyUnicode_AsUTF8 != NULL", "PyUnicode_AsUTF8 != NULL")
|
|
printf(" PyLong_FromLong(12345) -> str = %s\n", c_str)
|
|
|
|
# 读取回来
|
|
val: CLong = cpython.PyLong_AsLong(py_num)
|
|
testcheck.check(val == 12345, "PyLong_AsLong == 12345", "PyLong_AsLong == 12345")
|
|
|
|
# 负数
|
|
py_neg: cpython.PyObject | CPtr = cpython.PyLong_FromLong(-999)
|
|
neg_val: CLong = cpython.PyLong_AsLong(py_neg)
|
|
testcheck.check(neg_val == -999, "PyLong_FromLong(-999) == -999", "PyLong_FromLong(-999) == -999")
|
|
|
|
# 浮点数转整数
|
|
py_dbl: cpython.PyObject | CPtr = cpython.PyFloat_FromDouble(3.14159)
|
|
testcheck.check(py_dbl != 0, "PyFloat_FromDouble != NULL", "PyFloat_FromDouble != NULL")
|
|
dbl_val: CDouble = cpython.PyFloat_AsDouble(py_dbl)
|
|
testcheck.check(dbl_val > 3.14 and dbl_val < 3.15, "PyFloat_AsDouble ~= 3.14", "PyFloat_AsDouble ~= 3.14")
|
|
|
|
# 清理
|
|
cpython.Py_DecRef(py_dbl)
|
|
cpython.Py_DecRef(py_neg)
|
|
cpython.Py_DecRef(py_str)
|
|
cpython.Py_DecRef(py_num)
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_string_operations():
|
|
testcheck.section("PyUnicode operations")
|
|
cpython.Py_Initialize()
|
|
|
|
# 从 C 字符串创建
|
|
py_s1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Hello, Viper!")
|
|
testcheck.check(py_s1 != 0, "PyUnicode_FromString != NULL", "PyUnicode_FromString != NULL")
|
|
|
|
# 获取 UTF8
|
|
utf8: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_s1)
|
|
testcheck.check(utf8 != 0, "AsUTF8 != NULL", "AsUTF8 != NULL")
|
|
printf(" PyUnicode_FromString -> %s\n", utf8)
|
|
|
|
# 字符串长度
|
|
length: CLong = cpython.PyUnicode_GetLength(py_s1)
|
|
testcheck.check(length == 13, "GetLength == 13", "GetLength == 13")
|
|
|
|
# 字符串拼接
|
|
py_s2: cpython.PyObject | CPtr = cpython.PyUnicode_FromString(" World")
|
|
py_concat: cpython.PyObject | CPtr = cpython.PyUnicode_Concat(py_s1, py_s2)
|
|
concat_utf8: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_concat)
|
|
testcheck.check(py_concat != 0, "Concat != NULL", "Concat != NULL")
|
|
printf(" Concat -> %s\n", concat_utf8)
|
|
|
|
# 清理
|
|
cpython.Py_DecRef(py_concat)
|
|
cpython.Py_DecRef(py_s2)
|
|
cpython.Py_DecRef(py_s1)
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_list_operations():
|
|
testcheck.section("PyList operations")
|
|
cpython.Py_Initialize()
|
|
|
|
# 创建列表
|
|
py_list: cpython.PyObject | CPtr = cpython.PyList_New(0)
|
|
testcheck.check(py_list != 0, "PyList_New != NULL", "PyList_New != NULL")
|
|
|
|
# Append 元素
|
|
py_i1: cpython.PyObject | CPtr = cpython.PyLong_FromLong(10)
|
|
py_i2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(20)
|
|
py_i3: cpython.PyObject | CPtr = cpython.PyLong_FromLong(30)
|
|
cpython.PyList_Append(py_list, py_i1)
|
|
cpython.PyList_Append(py_list, py_i2)
|
|
cpython.PyList_Append(py_list, py_i3)
|
|
|
|
# 检查大小
|
|
list_size: CSizeT = cpython.PyList_Size(py_list)
|
|
testcheck.check(list_size == 3, "PyList_Size == 3", "PyList_Size == 3")
|
|
|
|
# 获取元素
|
|
item0: cpython.PyObject | CPtr = cpython.PyList_GetItem(py_list, 0)
|
|
val0: CLong = cpython.PyLong_AsLong(item0)
|
|
testcheck.check(val0 == 10, "list[0] == 10", "list[0] == 10")
|
|
|
|
item2: cpython.PyObject | CPtr = cpython.PyList_GetItem(py_list, 2)
|
|
val2: CLong = cpython.PyLong_AsLong(item2)
|
|
testcheck.check(val2 == 30, "list[2] == 30", "list[2] == 30")
|
|
|
|
# 清理
|
|
cpython.Py_DecRef(py_i3)
|
|
cpython.Py_DecRef(py_i2)
|
|
cpython.Py_DecRef(py_i1)
|
|
cpython.Py_DecRef(py_list)
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_dict_operations():
|
|
testcheck.section("PyDict operations")
|
|
cpython.Py_Initialize()
|
|
|
|
# 创建字典
|
|
py_dict: cpython.PyObject | CPtr = cpython.PyDict_New()
|
|
testcheck.check(py_dict != 0, "PyDict_New != NULL", "PyDict_New != NULL")
|
|
|
|
# 设置键值对
|
|
py_key1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("name")
|
|
py_val1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Viper")
|
|
cpython.PyDict_SetItem(py_dict, py_key1, py_val1)
|
|
|
|
py_key2: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("version")
|
|
py_val2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(1)
|
|
cpython.PyDict_SetItem(py_dict, py_key2, py_val2)
|
|
|
|
# 检查大小
|
|
dict_size: CSizeT = cpython.PyDict_Size(py_dict)
|
|
testcheck.check(dict_size == 2, "PyDict_Size == 2", "PyDict_Size == 2")
|
|
|
|
# 获取值
|
|
got_val: cpython.PyObject | CPtr = cpython.PyDict_GetItem(py_dict, py_key1)
|
|
testcheck.check(got_val != 0, "GetItem(name) != NULL", "GetItem(name) != NULL")
|
|
got_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(got_val)
|
|
testcheck.check(got_str != 0, "name == Viper", "name == Viper")
|
|
printf(" dict['name'] = %s\n", got_str)
|
|
|
|
# 使用 GetItemString
|
|
got_ver: cpython.PyObject | CPtr = cpython.PyDict_GetItemString(py_dict, "version")
|
|
testcheck.check(got_ver != 0, "GetItemString(version) != NULL", "GetItemString(version) != NULL")
|
|
ver_val: CLong = cpython.PyLong_AsLong(got_ver)
|
|
testcheck.check(ver_val == 1, "version == 1", "version == 1")
|
|
|
|
# 清理
|
|
cpython.Py_DecRef(py_val2)
|
|
cpython.Py_DecRef(py_key2)
|
|
cpython.Py_DecRef(py_val1)
|
|
cpython.Py_DecRef(py_key1)
|
|
cpython.Py_DecRef(py_dict)
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_pyrun_string():
|
|
testcheck.section("PyRun_SimpleString")
|
|
cpython.Py_Initialize()
|
|
|
|
# 执行简单 Python 代码
|
|
rc: CInt = cpython.PyRun_SimpleString("print('Hello from embedded Python!')")
|
|
testcheck.check(rc == 0, "PyRun_SimpleString == 0", "PyRun_SimpleString == 0")
|
|
|
|
# 执行赋值
|
|
rc2: CInt = cpython.PyRun_SimpleString("x = 6 * 7")
|
|
testcheck.check(rc2 == 0, "PyRun_SimpleString(assign) == 0", "PyRun_SimpleString(assign) == 0")
|
|
|
|
# 读取结果
|
|
rc3: CInt = cpython.PyRun_SimpleString("print(f'The answer is {x}')")
|
|
testcheck.check(rc3 == 0, "PyRun_SimpleString(print) == 0", "PyRun_SimpleString(print) == 0")
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_import_module():
|
|
testcheck.section("PyImport_ImportModule")
|
|
cpython.Py_Initialize()
|
|
|
|
# 导入 math 模块
|
|
math_mod: cpython.PyObject | CPtr = cpython.PyImport_ImportModule("math")
|
|
testcheck.check(math_mod != 0, "import math != NULL", "import math != NULL")
|
|
|
|
if math_mod != 0:
|
|
# 获取 math.pi
|
|
pi_val: cpython.PyObject | CPtr = cpython.PyObject_GetAttrString(math_mod, "pi")
|
|
testcheck.check(pi_val != 0, "math.pi != NULL", "math.pi != NULL")
|
|
if pi_val != 0:
|
|
pi_dbl: CDouble = cpython.PyFloat_AsDouble(pi_val)
|
|
testcheck.check(pi_dbl > 3.14 and pi_dbl < 3.15, "math.pi ~= 3.14", "math.pi ~= 3.14")
|
|
printf(" math.pi = %f\n", pi_dbl)
|
|
cpython.Py_DecRef(pi_val)
|
|
|
|
# 获取 math.sqrt 函数
|
|
sqrt_func: cpython.PyObject | CPtr = cpython.PyObject_GetAttrString(math_mod, "sqrt")
|
|
testcheck.check(sqrt_func != 0, "math.sqrt != NULL", "math.sqrt != NULL")
|
|
if sqrt_func != 0:
|
|
cpython.Py_DecRef(sqrt_func)
|
|
|
|
cpython.Py_DecRef(math_mod)
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def test_operator_overload():
|
|
testcheck.section("Operator Overload (PyLongObj / PyFloatObj / PyUnicodeObj / PyListObj / PyDictObj)")
|
|
cpython.Py_Initialize()
|
|
|
|
# --- PyLongObj: from_long, to_long, +, -, *, - (neg) ---
|
|
a: cpython.PyLongObj | CPtr = cpython.PyLongObj.from_long(10)
|
|
b: cpython.PyLongObj | CPtr = cpython.PyLongObj.from_long(3)
|
|
testcheck.check(a != 0 and b != 0, "PyLongObj.from_long != NULL", "PyLongObj.from_long != NULL")
|
|
|
|
add_r: cpython.PyLongObj | CPtr = a + b
|
|
testcheck.check(add_r.to_long() == 13, "PyLongObj 10 + 3 == 13", "PyLongObj 10 + 3 == 13")
|
|
|
|
sub_r: cpython.PyLongObj | CPtr = a - b
|
|
testcheck.check(sub_r.to_long() == 7, "PyLongObj 10 - 3 == 7", "PyLongObj 10 - 3 == 7")
|
|
|
|
mul_r: cpython.PyLongObj | CPtr = a * b
|
|
testcheck.check(mul_r.to_long() == 30, "PyLongObj 10 * 3 == 30", "PyLongObj 10 * 3 == 30")
|
|
|
|
neg_r: cpython.PyLongObj | CPtr = -b
|
|
testcheck.check(neg_r.to_long() == -3, "PyLongObj -3 == -3", "PyLongObj -3 == -3")
|
|
|
|
mul_r.dec_ref()
|
|
sub_r.dec_ref()
|
|
add_r.dec_ref()
|
|
neg_r.dec_ref()
|
|
b.dec_ref()
|
|
a.dec_ref()
|
|
|
|
# --- PyFloatObj: from_double, to_double, +, -, *, /, - (neg) ---
|
|
fa: cpython.PyFloatObj | CPtr = cpython.PyFloatObj.from_double(10.0)
|
|
fb: cpython.PyFloatObj | CPtr = cpython.PyFloatObj.from_double(4.0)
|
|
testcheck.check(fa != 0 and fb != 0, "PyFloatObj.from_double != NULL", "PyFloatObj.from_double != NULL")
|
|
|
|
fadd: cpython.PyFloatObj | CPtr = fa + fb
|
|
testcheck.check(fadd.to_double() == 14.0, "PyFloatObj 10.0 + 4.0 == 14.0", "PyFloatObj 10.0 + 4.0 == 14.0")
|
|
|
|
fsub: cpython.PyFloatObj | CPtr = fa - fb
|
|
testcheck.check(fsub.to_double() == 6.0, "PyFloatObj 10.0 - 4.0 == 6.0", "PyFloatObj 10.0 - 4.0 == 6.0")
|
|
|
|
fmul: cpython.PyFloatObj | CPtr = fa * fb
|
|
testcheck.check(fmul.to_double() == 40.0, "PyFloatObj 10.0 * 4.0 == 40.0", "PyFloatObj 10.0 * 4.0 == 40.0")
|
|
|
|
fdiv: cpython.PyFloatObj | CPtr = fa / fb
|
|
testcheck.check(fdiv.to_double() == 2.5, "PyFloatObj 10.0 / 4.0 == 2.5", "PyFloatObj 10.0 / 4.0 == 2.5")
|
|
|
|
fneg: cpython.PyFloatObj | CPtr = -fb
|
|
testcheck.check(fneg.to_double() == -4.0, "PyFloatObj -4.0 == -4.0", "PyFloatObj -4.0 == -4.0")
|
|
|
|
fadd.dec_ref()
|
|
fsub.dec_ref()
|
|
fmul.dec_ref()
|
|
fdiv.dec_ref()
|
|
fneg.dec_ref()
|
|
fb.dec_ref()
|
|
fa.dec_ref()
|
|
|
|
# --- PyUnicodeObj: from_string, as_utf8, len, + (concat) ---
|
|
sa: cpython.PyUnicodeObj | CPtr = cpython.PyUnicodeObj.from_string("Hello")
|
|
sb: cpython.PyUnicodeObj | CPtr = cpython.PyUnicodeObj.from_string(" World")
|
|
testcheck.check(sa != 0 and sb != 0, "PyUnicodeObj.from_string != NULL", "PyUnicodeObj.from_string != NULL")
|
|
|
|
sc: cpython.PyUnicodeObj | CPtr = sa + sb
|
|
sc_utf8: t.CChar | CPtr = sc.as_utf8()
|
|
testcheck.check(sc_utf8 != 0, "PyUnicodeObj concat != NULL", "PyUnicodeObj concat != NULL")
|
|
if sc_utf8 != 0:
|
|
printf(" PyUnicodeObj 'Hello' + ' World' = %s\n", sc_utf8)
|
|
|
|
sa_len: t.CPtrDiffT = len(sa)
|
|
testcheck.check(sa_len == 5, "PyUnicodeObj len('Hello') == 5", "PyUnicodeObj len('Hello') == 5")
|
|
|
|
sc.dec_ref()
|
|
sb.dec_ref()
|
|
sa.dec_ref()
|
|
|
|
# --- PyListObj: new, append, len, [], []= ---
|
|
lst: cpython.PyListObj | CPtr = cpython.PyListObj.new()
|
|
testcheck.check(lst != 0, "PyListObj.new != NULL", "PyListObj.new != NULL")
|
|
if lst != 0:
|
|
e1: cpython.PyObject | CPtr = cpython.PyLong_FromLong(100)
|
|
e2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(200)
|
|
e3: cpython.PyObject | CPtr = cpython.PyLong_FromLong(300)
|
|
lst.append(e1)
|
|
lst.append(e2)
|
|
lst.append(e3)
|
|
testcheck.check(len(lst) == 3, "PyListObj len == 3", "PyListObj len == 3")
|
|
|
|
item0: cpython.PyObject | CPtr = lst[t.CSizeT(0)]
|
|
testcheck.check(cpython.PyLong_AsLong(item0) == 100, "PyListObj lst[0] == 100", "PyListObj lst[0] == 100")
|
|
|
|
# __setitem__
|
|
e4: cpython.PyObject | CPtr = cpython.PyLong_FromLong(999)
|
|
lst[t.CSizeT(1)] = e4
|
|
item1: cpython.PyObject | CPtr = lst[t.CSizeT(1)]
|
|
testcheck.check(cpython.PyLong_AsLong(item1) == 999, "PyListObj lst[1] = 999", "PyListObj lst[1] = 999")
|
|
|
|
cpython.Py_DecRef(e4)
|
|
cpython.Py_DecRef(e3)
|
|
cpython.Py_DecRef(e2)
|
|
cpython.Py_DecRef(e1)
|
|
lst.dec_ref()
|
|
|
|
# --- PyDictObj: new, [], []=, len, set_string, get_string ---
|
|
dct: cpython.PyDictObj | CPtr = cpython.PyDictObj.new()
|
|
testcheck.check(dct != 0, "PyDictObj.new != NULL", "PyDictObj.new != NULL")
|
|
if dct != 0:
|
|
dk1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("lang")
|
|
dv1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Viper")
|
|
dct[dk1] = dv1
|
|
testcheck.check(len(dct) == 1, "PyDictObj len == 1", "PyDictObj len == 1")
|
|
|
|
got_lang: cpython.PyObject | CPtr = dct[dk1]
|
|
got_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(got_lang)
|
|
testcheck.check(got_str != 0, "PyDictObj dct['lang'] == Viper", "PyDictObj dct['lang'] == Viper")
|
|
if got_str != 0:
|
|
printf(" PyDictObj dct['lang'] = %s\n", got_str)
|
|
|
|
# get_string / set_string
|
|
dct.set_string("ver", cpython.PyLong_FromLong(3))
|
|
got_ver: cpython.PyObject | CPtr = dct.get_string("ver")
|
|
testcheck.check(got_ver != 0 and cpython.PyLong_AsLong(got_ver) == 3, "PyDictObj get_string('ver') == 3", "PyDictObj get_string('ver') == 3")
|
|
|
|
cpython.Py_DecRef(dv1)
|
|
cpython.Py_DecRef(dk1)
|
|
dct.dec_ref()
|
|
|
|
cpython.Py_Finalize()
|
|
|
|
|
|
def main() -> CInt | t.CExport:
|
|
testcheck.begin("CPython Integration Test")
|
|
|
|
test_init_finalize()
|
|
test_long_operations()
|
|
test_string_operations()
|
|
test_list_operations()
|
|
test_dict_operations()
|
|
test_pyrun_string()
|
|
test_import_module()
|
|
test_operator_overload()
|
|
|
|
return testcheck.end()
|