08eb78a5b2
This PR sets up the new integrated test/bench suite. It then migrates all benchmarks and some related tests to the new suite. There's also some documentation and some linting. For now, a lot of the old tests are left alone so this PR doesn't become even larger than it already is. Eventually, all tests should be migrated to the new suite though so there isn't a confusing mix of two systems.
70 lines
2 KiB
Text
70 lines
2 KiB
Text
import Lean.Meta
|
|
|
|
open Lean
|
|
open Lean.Meta
|
|
|
|
def dbgOpt : Options :=
|
|
let opt : Options := {};
|
|
let opt := opt.set `trace.Meta true;
|
|
-- let opt := opt.set `trace.Meta.check false;
|
|
opt
|
|
|
|
def print (msg : MessageData) : MetaM Unit := do
|
|
trace[Meta.debug] msg
|
|
|
|
def check (x : MetaM Bool) : MetaM Unit :=
|
|
unless (← x) do throwError "check failed"
|
|
|
|
def getAssignment (m : Expr) : MetaM Expr :=
|
|
do let v? ← getExprMVarAssignment? m.mvarId!;
|
|
(match v? with
|
|
| some v => pure v
|
|
| none => throwError "metavariable is not assigned")
|
|
|
|
def nat := mkConst `Nat
|
|
def succ := mkConst `Nat.succ
|
|
def add := mkAppN (mkConst `Add.add [levelZero]) #[nat, mkConst `Nat.add]
|
|
|
|
def tst1 : MetaM Unit :=
|
|
do let d : DiscrTree Nat := {};
|
|
let mvar ← mkFreshExprMVar nat;
|
|
let d ← d.insert (mkAppN add #[mvar, mkNatLit 10]) 1
|
|
let d ← d.insert (mkAppN add #[mkNatLit 0, mkNatLit 10]) 2
|
|
let d ← d.insert (mkAppN (mkConst `Nat.add) #[mkNatLit 0, mkNatLit 20]) 3
|
|
let d ← d.insert (mkAppN add #[mvar, mkNatLit 20]) 4
|
|
let d ← d.insert mvar 5
|
|
print (format d);
|
|
let vs ← d.getMatch (mkAppN add #[mkNatLit 1, mkNatLit 10]);
|
|
print (format vs);
|
|
let t := mkAppN add #[mvar, mvar];
|
|
print t;
|
|
let vs ← d.getMatch t
|
|
print (format vs);
|
|
let vs ← d.getUnify t;
|
|
print (format vs);
|
|
let vs ← d.getUnify mvar;
|
|
print (format vs);
|
|
let vs ← d.getUnify (mkAppN add #[mkNatLit 0, mvar]);
|
|
print (format vs);
|
|
let vs ← d.getUnify (mkAppN add #[mvar, mkNatLit 20]);
|
|
print (format vs);
|
|
pure ()
|
|
|
|
set_option trace.Meta.debug true in
|
|
set_option pp.mvars false in
|
|
/--
|
|
trace: [Meta.debug] (Add.add => (node
|
|
(Nat => (node
|
|
(* => (node (* => (node (10 => (node #[1])) (20 => (node #[4])))) (0 => (node (10 => (node #[2]))))))))))
|
|
(* => (node #[5]))
|
|
(Nat.add => (node (0 => (node (20 => (node #[3]))))))
|
|
[Meta.debug] #[5, 1]
|
|
[Meta.debug] Add.add ?_ ?_
|
|
[Meta.debug] #[5]
|
|
[Meta.debug] #[5, 1, 4, 2]
|
|
[Meta.debug] #[1, 4, 2, 5, 3]
|
|
[Meta.debug] #[5, 1, 4, 2]
|
|
[Meta.debug] #[5, 4]
|
|
-/
|
|
#guard_msgs in
|
|
run_meta tst1
|