Files
TransPyC/Test/ListTest/App/main.py
2026-07-18 19:25:40 +08:00

91 lines
2.8 KiB
Python

from stdint import *
import w32.win32console
import t, c
from t import CInt, CExport
import stdlib
import stdio
import memhub
import testcheck
def main() -> CInt | CExport:
w32.win32console.SetConsoleCP(65001)
w32.win32console.SetConsoleOutputCP(65001)
testcheck.begin("ListTest: list[T] 堆容器测试")
# 创建 mbuddy arena
arena: bytes = stdlib.malloc(65536)
bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536)
# === Test 1: 创建和 append ===
testcheck.section("Test 1: 创建和 append")
nums = list[int](bd)
nums.append(10)
nums.append(20)
nums.append(30)
v0: CInt = nums.get(0)
v1: CInt = nums.get(1)
v2: CInt = nums.get(2)
stdio.printf("v0=%d v1=%d v2=%d\n", v0, v1, v2)
testcheck.check(v0 == 10 and v1 == 20 and v2 == 30,
"append+get OK (10,20,30)", "append+get FAILED")
# === Test 2: __len__ ===
testcheck.section("Test 2: __len__")
n: t.CSizeT = nums.__len__()
stdio.printf("len=%lu\n", n)
testcheck.check(n == 3, "__len__ OK (3)", "__len__ FAILED expect 3")
# === Test 3: set ===
testcheck.section("Test 3: set")
nums.set(1, 99)
v1b: CInt = nums.get(1)
stdio.printf("after set(1,99): v1=%d\n", v1b)
testcheck.check(v1b == 99, "set OK (idx1=99)", "set FAILED")
# === Test 4: pop ===
testcheck.section("Test 4: pop")
popped: CInt = nums.pop()
stdio.printf("popped=%d\n", popped)
testcheck.check(popped == 30, "pop OK (30)", "pop FAILED expect 30")
n2: t.CSizeT = nums.__len__()
testcheck.check(n2 == 2, "pop len OK (2)", "pop len FAILED expect 2")
# === Test 5: clear ===
testcheck.section("Test 5: clear")
nums.clear()
n3: t.CSizeT = nums.__len__()
testcheck.check(n3 == 0, "clear OK (0)", "clear FAILED expect 0")
# === Test 6: 容量增长 (append 超过初始容量 8) ===
testcheck.section("Test 6: 容量增长")
nums2 = list[int](bd)
i: CInt
for i in range(20):
nums2.append(i * 5)
n4: t.CSizeT = nums2.__len__()
stdio.printf("after 20 appends: len=%lu\n", n4)
testcheck.check(n4 == 20, "grow len OK (20)", "grow len FAILED")
ok: CInt = 1
for i in range(20):
v: CInt = nums2.get(i)
if v != i * 5:
stdio.printf("MISMATCH at %d: got %d expect %d\n", i, v, i * 5)
ok = 0
break
testcheck.check(ok == 1, "grow data OK", "grow data FAILED")
# === Test 7: 重新 append 后验证 ===
testcheck.section("Test 7: 重新 append")
nums2.clear()
nums2.append(100)
nums2.append(200)
va: CInt = nums2.get(0)
vb: CInt = nums2.get(1)
stdio.printf("re-append: va=%d vb=%d\n", va, vb)
testcheck.check(va == 100 and vb == 200, "re-append OK", "re-append FAILED")
stdlib.free(arena)
return testcheck.end()