62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
from type_def import *
|
|
from inference import *
|
|
from helper import *
|
|
|
|
types = {
|
|
'int32': PrimitiveType('int32'),
|
|
'int64': PrimitiveType('int64'),
|
|
'str': PrimitiveType('str'),
|
|
}
|
|
|
|
i32 = types['int32']
|
|
i64 = types['int64']
|
|
s = types['str']
|
|
|
|
variables = {
|
|
'X': TypeVariable('X', []),
|
|
'Y': TypeVariable('Y', []),
|
|
'I': TypeVariable('I', [i32, i64]),
|
|
'A': TypeVariable('A', [i32, i64, s]),
|
|
}
|
|
|
|
X = variables['X']
|
|
Y = variables['Y']
|
|
I = variables['I']
|
|
A = variables['A']
|
|
|
|
|
|
def stringify_subst(subst):
|
|
if isinstance(subst, str):
|
|
return subst
|
|
elements = [f"{key}: {str(value)}" for key, value in subst.items()]
|
|
return "{" + ', '.join(elements) + "}"
|
|
|
|
def try_case(a, b, ctx):
|
|
result = find_subst(ctx, {}, a, b)
|
|
print(f"{a} <- {b} w.r.t. {stringify_subst(ctx)}\n {stringify_subst(result)}\n")
|
|
|
|
|
|
try_case(i32, i32, {})
|
|
try_case(i32, i64, {})
|
|
try_case(i32, X, {})
|
|
try_case(i32, X, {'X': i32})
|
|
try_case(i32, X, {'X': i64})
|
|
try_case(X, X, {'X': i64})
|
|
try_case(X, Y, {'Y': i64})
|
|
try_case(X, Y, {'X': i64})
|
|
try_case(I, X, {})
|
|
try_case(I, A, {})
|
|
try_case(A, I, {})
|
|
try_case(X, I, {})
|
|
try_case(ListType(i32), TupleType([i32]), {})
|
|
try_case(TupleType([i32]), ListType(i32), {})
|
|
try_case(TupleType([i32, i32]), TupleType([i32]), {})
|
|
try_case(ListType(i32), ListType(i32), {})
|
|
try_case(TupleType([X, X]), TupleType([X, Y]), {})
|
|
try_case(TupleType([X, X]), TupleType([Y, Y]), {})
|
|
try_case(TupleType([X, Y]), TupleType([X, X]), {})
|
|
try_case(TupleType([X, X]), TupleType([X, X]), {})
|
|
try_case(TupleType([X, Y]), X, {})
|
|
try_case(TupleType([i32, Y]), X, {})
|
|
|