27 lines
966 B
Lua
27 lines
966 B
Lua
|
function test_unify(env, m, lhs, rhs, num_s)
|
||
|
|
print(tostring(lhs) .. " =?= " .. tostring(rhs) .. ", expected: " .. tostring(num_s))
|
||
|
|
local ss = unify(env, lhs, rhs)
|
||
|
|
local n = 0
|
||
|
|
for s in ss do
|
||
|
|
print("solution: " .. tostring(s:instantiate(m)))
|
||
|
|
n = n + 1
|
||
|
|
end
|
||
|
|
if num_s ~= n then print("n: " .. n) end
|
||
|
|
assert(num_s == n)
|
||
|
|
end
|
||
|
|
|
||
|
|
local env = environment()
|
||
|
|
env = add_decl(env, mk_var_decl("N", Type))
|
||
|
|
local N = Const("N")
|
||
|
|
env = add_decl(env, mk_var_decl("f", mk_arrow(N, N, N)))
|
||
|
|
env = add_decl(env, mk_var_decl("a", N))
|
||
|
|
local f = Const("f")
|
||
|
|
local a = Const("a")
|
||
|
|
local l1 = mk_local("l1", "x", N)
|
||
|
|
local l2 = mk_local("l2", "y", N)
|
||
|
|
local l3 = mk_local("l3", "z", N)
|
||
|
|
local m = mk_metavar("m", mk_arrow(N, N, mk_metavar("m_type", mk_arrow(N, N, mk_sort(mk_meta_univ("u"))))(Var(1), Var(0))))
|
||
|
|
test_unify(env, m, m(l1, l1), f(f(a, l1), l1), 4)
|
||
|
|
print("-----------------")
|
||
|
|
test_unify(env, m, m(l1, l1), mk_lambda("z", Bool, f(l1, f(Var(0), a))), 2)
|