53 lines
1.6 KiB
Plaintext
53 lines
1.6 KiB
Plaintext
import t
|
|
import stdio
|
|
import string
|
|
import testcheck
|
|
|
|
def test_for_in_string():
|
|
testcheck.section("Test 1: for-in string iteration")
|
|
s: char* = "Hello"
|
|
result: int = 0
|
|
for ch in s:
|
|
result += ch
|
|
# H=72, e=101, l=108, l=108, o=111 => 500
|
|
testcheck.check(result == 500, "for-in string (sum=500)", "for-in string (sum expect 500)")
|
|
|
|
def test_for_in_char_ptr():
|
|
testcheck.section("Test 2: for-in char* iteration")
|
|
s: char* = "ABC"
|
|
chars: t.CArray[int, 4] = [0]
|
|
idx: int = 0
|
|
for ch in s:
|
|
if idx < 3:
|
|
chars[idx] = ch
|
|
idx += 1
|
|
testcheck.check(chars[0] == 65 and chars[1] == 66 and chars[2] == 67, "for-in char* (A=65 B=66 C=67)", "for-in char* (expect A=65 B=66 C=67)")
|
|
|
|
def test_strcpy_for_in():
|
|
testcheck.section("Test 3: strcpy using for-in")
|
|
src: char* = "test"
|
|
# Implement strcpy by hand with for-in.
|
|
import win32.win32memory
|
|
buf: char* = win32.win32memory.VirtualAlloc(None, 64, 12288, 4)
|
|
if uint64(buf) == 0:
|
|
stdio.printf("SKIP: VirtualAlloc returned NULL\n")
|
|
return
|
|
string.memset(buf, 0, 64)
|
|
# Copy with for-in.
|
|
p: char* = buf
|
|
for ch in src:
|
|
p[0] = ch
|
|
p += 1
|
|
p[0] = 0
|
|
# '==' on pointers compares addresses (C-like); use strcmp for contents.
|
|
testcheck.check(string.strcmp(buf, "test") == 0, "strcpy for-in (result=test)", "strcpy for-in (cmp mismatch)")
|
|
|
|
import win32.win32memory
|
|
|
|
def main() -> int:
|
|
testcheck.begin("IterTest: 指针迭代测试")
|
|
test_for_in_string()
|
|
test_for_in_char_ptr()
|
|
test_strcpy_for_in()
|
|
return testcheck.end()
|