补充
This commit is contained in:
61
Test/IterTest/App/main.py
Normal file
61
Test/IterTest/App/main.py
Normal file
@@ -0,0 +1,61 @@
|
||||
import t
|
||||
import stdio
|
||||
import string
|
||||
|
||||
def test_for_in_string():
|
||||
stdio.printf("--- Test 1: for-in string iteration ---\n")
|
||||
s: t.CChar | t.CPtr = "Hello"
|
||||
result: t.CInt = 0
|
||||
for ch in s:
|
||||
result += ch
|
||||
# H=72, e=101, l=108, l=108, o=111 => 500
|
||||
if result == 500:
|
||||
stdio.printf("PASS: for-in string (sum=%d)\n", result)
|
||||
else:
|
||||
stdio.printf("FAIL: for-in string (sum=%d, expect 500)\n", result)
|
||||
|
||||
def test_for_in_char_ptr():
|
||||
stdio.printf("--- Test 2: for-in char* iteration ---\n")
|
||||
s: t.CChar | t.CPtr = "ABC"
|
||||
chars: list[t.CInt, 4] = [0]
|
||||
idx: t.CInt = 0
|
||||
for ch in s:
|
||||
if idx < 3:
|
||||
chars[idx] = ch
|
||||
idx += 1
|
||||
if chars[0] == 65 and chars[1] == 66 and chars[2] == 67:
|
||||
stdio.printf("PASS: for-in char* (A=%d B=%d C=%d)\n", chars[0], chars[1], chars[2])
|
||||
else:
|
||||
stdio.printf("FAIL: for-in char* (A=%d B=%d C=%d, expect 65 66 67)\n", chars[0], chars[1], chars[2])
|
||||
|
||||
def test_strcpy_for_in():
|
||||
stdio.printf("--- Test 3: strcpy using for-in ---\n")
|
||||
src: t.CChar | t.CPtr = "test"
|
||||
# 手动实现 strcpy 用 for-in
|
||||
import w32.win32memory
|
||||
buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4)
|
||||
if t.CUInt64T(buf) == 0:
|
||||
stdio.printf("SKIP: VirtualAlloc returned NULL\n")
|
||||
return
|
||||
string.memset(buf, 0, 64)
|
||||
# 用 for-in 复制
|
||||
p: t.CChar | t.CPtr = buf
|
||||
for ch in src:
|
||||
p[0] = ch
|
||||
p += 1
|
||||
p[0] = 0
|
||||
cmp_result: t.CInt = string.strcmp(buf, "test")
|
||||
if cmp_result == 0:
|
||||
stdio.printf("PASS: strcpy for-in (result=%s)\n", buf)
|
||||
else:
|
||||
stdio.printf("FAIL: strcpy for-in (cmp=%d)\n", cmp_result)
|
||||
|
||||
import w32.win32memory
|
||||
|
||||
def main() -> t.CInt:
|
||||
stdio.printf("=== IterTest: 指针迭代测试 ===\n\n")
|
||||
test_for_in_string()
|
||||
test_for_in_char_ptr()
|
||||
test_strcpy_for_in()
|
||||
stdio.printf("\n=== IterTest Complete ===\n")
|
||||
return 0
|
||||
Reference in New Issue
Block a user