58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
import t
|
|
import stdio
|
|
import testcheck
|
|
|
|
# ============================================================
|
|
# Test 1: 泛型函数手动传类型参数 - 基本调用
|
|
# def xxx[T](x: int) 然后调用 xxx[T](x)
|
|
# ============================================================
|
|
def double[T](x: int) -> int:
|
|
return x * 2
|
|
|
|
def test_explicit_type_arg():
|
|
testcheck.section("Test 1: explicit type arg xxx[T](x)")
|
|
r: t.CInt = double[t.CInt](21)
|
|
testcheck.check(r == 42, "double[t.CInt](21) = 42", "expect 42")
|
|
|
|
# ============================================================
|
|
# Test 2: 泛型函数手动传类型参数 - 多次调用不同类型
|
|
# ============================================================
|
|
def identity[T](x: int) -> int:
|
|
return x
|
|
|
|
def test_identity_explicit():
|
|
testcheck.section("Test 2: identity[T](x)")
|
|
r1: t.CInt = identity[t.CInt](10)
|
|
r2: t.CInt = identity[t.CInt](20)
|
|
testcheck.check(r1 == 10 and r2 == 20, "identity[t.CInt](10)=10 (20)=20", "expect 10 20")
|
|
|
|
# ============================================================
|
|
# Test 3: 泛型函数手动传类型参数 - 用 T 作为参数类型
|
|
# ============================================================
|
|
def add_one[T](x: T) -> T:
|
|
return x + T(1)
|
|
|
|
def test_add_one_explicit():
|
|
testcheck.section("Test 3: add_one[T](x) with T param")
|
|
r: t.CInt = add_one[t.CInt](41)
|
|
testcheck.check(r == 42, "add_one[t.CInt](41) = 42", "expect 42")
|
|
|
|
# ============================================================
|
|
# Test 4: 泛型函数手动传类型参数 - 混合自动推导和手动传
|
|
# ============================================================
|
|
def test_mixed_inference_explicit():
|
|
testcheck.section("Test 4: mixed inference and explicit")
|
|
# 自动推导
|
|
r1: t.CInt = add_one(99)
|
|
# 手动传
|
|
r2: t.CInt = add_one[t.CInt](99)
|
|
testcheck.check(r1 == 100 and r2 == 100, "add_one(99)=100 add_one[t.CInt](99)=100", "expect 100 100")
|
|
|
|
def main() -> t.CInt:
|
|
testcheck.begin("GenericTest3: 手动传类型参数 xxx[T](x)")
|
|
test_explicit_type_arg()
|
|
test_identity_explicit()
|
|
test_add_one_explicit()
|
|
test_mixed_inference_explicit()
|
|
return testcheck.end()
|