def eval_expr(exp, env, out): return conde( ( eq([quote, out], exp), no_item(out, is_internal), missing(quote, env), ), fresh(lambda lst: conjp( eq([list_, lst, ...], exp), missing(list_, env), eval_list(lst, env, out), )), fresh( 5, lambda fn, arg, var, body, cenv: conj( eval_list(exp, env, [fn, arg]), conde( eval_builtin(fn, arg, out), ( eq(Closure(var, body, cenv), fn), eval_expr(body, Env(var, arg, cenv), out), ), ))), fresh( 2, lambda var, body: conjp( eq([lambda_, [var], body], exp), eq(Closure(var, body, env), out), eqt(var, Symbol), missing(lambda_, env), )), (eqt(exp, Symbol), lookup(exp, env, out)), (builtin(exp, out), missing(exp, env)), )
def eval_list(lst, env, out): return conde((eq([], lst), eq([], out)), fresh( 4, lambda h, t, oh, ot: conjp( eq([h, t, ...], lst), eq([oh, ot, ...], out), eval_expr(h, env, oh), eval_list(t, env, ot), )))
def missing(var, env): return disj( eq((), env), fresh( 3, lambda rest, sym, val: conjp( eq(Env(sym, val, rest), env), neq(sym, var), missing(var, rest), )), )
def jugs(states): return disj( eq([(0, 0, "")], states), fresh( 8, lambda big, small, act, prev_big, prev_small, tail, _, __: conjp( eq([(big, small, act), tail, ...], states), eq([(prev_big, prev_small, _), __, ...], tail), conde( [ conde( [ eq(small, prev_small), conde( [eq(big, BIG), eq(act, "fill big")], [eq(big, 0), eq(act, "empty big")])], [ fresh( lambda total: conjp( conde( [eq(big, BIG), eq(act, "to big")], [ eq(small, SMALL), eq(act, "to small")], [ eq(small, 0), neq(big, BIG), eq(act, "to big")], [ eq(big, 0), neq(small, SMALL), eq(act, "to small")]), add(big, small, total), add(prev_big, prev_small, total)))]), neq(big, prev_big)], [ eq(big, prev_big), conde( [eq(small, SMALL), eq(act, "fill small")], [eq(small, 0), eq(act, "empty small")]), neq(small, prev_small)]), gte(big, 0), lte(big, BIG), gte(small, 0), lte(small, SMALL), jugs(tail), )))
def lookup(o, v, x): return fresh( 3, lambda a, b, t: conj( eq(o, ((a, b), t)), conde( [eq(a, v), eq(b, TMono(x))], fresh( 3, lambda po, pb, pm: conjp( eq(a, v), eq(b, TPoly(po, pb)), infer(pb, po, x, pm), )), [neq(a, v), lookup(t, v, x)], )))
def main(): states = Var() big = Var() small = Var() _ = Var() __ = Var() for i in range(1, BIG + 1): p = conjp(eq([(big, small, _), __, ...], states), disj( eq(big, i), conj(neq(big, i), eq(small, i)), ), jugs(states)) for answer in run(1, states, p): print("{}:".format(i)) for b, s, a in reversed(answer): print(b, s, a) print()
def infer(expr, env, typ, ann): return conde( fresh(lambda n: conjp( eq(expr, Sym(n)), lookup(env, n, typ), eq(ann, Ann(expr, typ)), )), fresh( 6, lambda f, ft, fa, a, at, aa: conjp( eq(expr, App(f, a)), infer(a, env, at, aa), infer(f, env, TFunc(at, typ), fa), eq(ann, App(fa, aa)), )), fresh( 5, lambda v, vt, b, bt, ba: conjp( eq(expr, Abs(Sym(v), b)), infer(b, ((v, TMono(vt)), env), bt, ba), eq(typ, TFunc(vt, bt)), eq(ann, Abs(Ann(Sym(v), vt), ba)), )), fresh( 4, lambda n, v, b, bm: conjp( eq(expr, Let(Sym(n), v, b)), infer(b, ((n, TPoly(env, v)), env), typ, bm), eq(ann, Let(Sym(n), v, bm)), )), fresh( 3, lambda n, b, bm: conjp( eq(expr, Rec(Sym(n), b)), infer(b, ((n, TMono(typ)), env), typ, bm), eq(ann, Rec(Ann(Sym(n), typ), bm)), )), fresh( 6, lambda c, cm, a, am, b, bm: conjp( eq(expr, If(c, a, b)), infer(c, env, TTerm("bool"), cm), infer(a, env, typ, am), infer(b, env, typ, bm), eq(ann, If(cm, am, bm)), )), [eqt(expr, int), eq(typ, TTerm("int")), eq(ann, expr)], [eqt(expr, bool), eq(typ, TTerm("bool")), eq(ann, expr)], )
def solve(): solutions = {} states = Var() big = Var() small = Var() _ = Var() __ = Var() for i in range(1, BIG + 1): p = conjp( eq([(big, small, _), __, ...], states), disj( eq(big, i), conj(neq(big, i), eq(small, i)), ), jugs(states) ) for answer in run(1, states, p): solutions[i] = list(reversed(answer)) return solutions
from mk.ext.tuples import no_item from mk.run import initial, reify from mk.stream import unfold from mk.unify import Var a = Var() b = Var() Tuple = namedtuple("Tuple", "x, y") GOALS_DATA = [ (eq(a, ()), a, [()]), (eq(a, (1, )), a, [(1, )]), (conj(eq(a, (b, )), eq(b, 1)), a, [(1, )]), (eq(a, (a, )), a, []), (conjp(eq(a, (1, )), eq(b, (1, )), eq(a, b)), a, [(1, )]), (conjp(eq(a, (1, )), eq(b, (2, )), eq(a, b)), a, []), (conjp(eq(a, (1, )), eq(b, (1, 2)), eq(a, b)), a, []), (conjp(eq(a, (1, )), eq(b, ()), eq(a, b)), a, []), (conj(eq(a, Tuple(1, 2)), eq(a, Tuple(1, 2))), a, [Tuple(1, 2)]), (conj(eq(a, Tuple(1, 1)), eq(a, Tuple(1, 2))), a, []), (conj(eq(a, Tuple(1, 2)), eq(a, (1, 2))), a, []), (no_item((1, 2, 3), lambda v: v == 0), None, [None]), (no_item((1, 2, 3), lambda v: v == 1), None, []), (conj(no_item(a, lambda v: v == 0), eq(a, (1, 2, 3))), None, [None]), (conj(no_item(a, lambda v: v == 1), eq(a, (1, 2, 3))), None, []), (no_item((1, (1, 2), (1, 2, (1, 2, 3))), lambda v: v == 0), None, [None]), (no_item((1, (1, 2), (1, 2, (1, 2, 3))), lambda v: v == 3), None, []), ]
a = Var() b = Var() GOALS_DATA = [ (eq(a, []), a, [[]]), (eq(a, [1]), a, [[1]]), (conj(eq(a, [b]), eq(b, 1)), a, [[1]]), (eq(a, [a]), a, []), (conj(eq(a, [b]), eq(b, [1, 2])), a, [[[1, 2]]]), (conj(eq(a, [0, b]), eq(b, [1, 2])), a, [[0, [1, 2]]]), (conj(eq(a, [b, ...]), eq(b, [1, 2])), a, [[1, 2]]), (conj(eq(a, [0, b, ...]), eq(b, [1, 2])), a, [[0, 1, 2]]), (conjp(eq(a, [1]), eq(b, [1]), eq(a, b)), a, [[1]]), (conjp(eq(a, [1]), eq(b, [2]), eq(a, b)), a, []), (conjp(eq(a, [1]), eq(b, [1, 2]), eq(a, b)), a, []), (conjp(eq(a, [1]), eq(b, []), eq(a, b)), a, []), (eq([1], [1]), None, [None]), (conj(eq(a, [1]), eq(a, [1])), a, [[1]]), (conj(eq(a, [1]), eq([1], a)), a, [[1]]), (conj(eq(a, 1), eq([1], a)), a, []), (conj(eq(a, []), eq([1], a)), a, []), (conj(eq(a, []), eq([], a)), a, [[]]), (conj(eq(a, [1]), eq([], a)), a, []), (conj(eq(a, [1, b, ...]), eq(b, [2, ...])), a, [[1, 2, ...]]),
(eq(a, 1), delay_goal(2, b)), (eq(a, 2), delay_goal(3, b), eq(a, b)) ) GOALS_DATA = [ (conj(eq(a, 1), fresh(lambda x: conj(eq(a, x), eq(x, c)))), c, [1]), ( conj( eq(a, 1), fresh(2, lambda x, y: conj(eq((x, y), (a, 2)), eq(c, (x, y)))) ), c, [(1, 2)] ), (conjp(eq(a, 1), eq(b, 2), eq(c, (a, b))), c, [(1, 2)]), (conjp(disjp(eq(a, 1), eq(a, 2), eq(a, 3)), eq(c, a)), c, [1, 3, 2]), ( conde( (eq(a, 1), eq(b, 2), eq(c, a)), (eq(a, 2), eq(b, 3), eq(c, b)), eq(c, 5), ), c, [1, 5, 3] ), (zzz_goal(1, c), c, [(1, 1), (1, 2)]), (delay_goal(1, c), c, [(1, 1), (1, 2)]), ]
from mk.unify import Var a = Var() b = Var() GOALS_DATA = [ (conj(eq(a, 1), predicate(a, lambda a: a > 0)), [{ a: 1 }]), (conj(predicate(a, lambda a: a > 0), eq(a, 1)), [{ a: 1 }]), (conj(eq(a, 0), predicate(a, lambda a: a > 0)), []), (conj(predicate(a, lambda a: a > 0), eq(a, 0)), []), (predicate(a, lambda a: a > 0), [{}]), (conjp(eq(a, 1), eq(b, 0), relation(a, b, lambda a, b: a > b)), [{ a: 1, b: 0 }]), (conjp(relation(a, b, lambda a, b: a > b), eq(a, 1), eq(b, 0)), [{ a: 1, b: 0 }]), (conjp(eq(a, 1), relation(a, b, lambda a, b: a > b), eq(b, 0)), [{ a: 1, b: 0 }]), (conjp(eq(a, 1), eq(b, 1), relation(a, b, lambda a, b: a > b)), []), (conjp(relation(a, b, lambda a, b: a > b), eq(a, 1), eq(b, 1)), []), (conjp(eq(a, 1), relation(a, b, lambda a, b: a > b), eq(b, 1)), []), (conj(eq(a, 1), relation(a, b, lambda a, b: a > b)), [{
from collections import namedtuple import pytest from mk.core import eq, eqt from mk.dsl import conjp from mk.run import run from mk.unify import Var a = Var() b = Var() RUN_DATA = [ (eq(a, 1), a, 1), (conjp(eq(a, (1, b)), eq(b, 2)), a, (1, 2)), (conjp(eq(a, (1, b)), eq(b, 2)), (a, b), ((1, 2), 2)), ] @pytest.mark.parametrize("goal, query, expected", RUN_DATA) def test_run(goal, query, expected): assert next(run(0, query, goal)) == expected Tuple = namedtuple("Tuple", "a, b") REIFY_DATA = [ (eq(a, a), a, '_0:_0'), (eq(a, (1, b)), a, '(1, _0:_0)'), (eq(a, Tuple(1, b)), a, 'Tuple(a=1, b=_0:_0)'), (eq(a, [1, b]), a, '[1, _0:_0]'), (eq(a, [1, b, ...]), a, '[1, _0:_0, Ellipsis]'),