Files
lean4/tests/elab/generalizeTelescope.lean
Garmelon 08eb78a5b2 chore: switch to new test/bench suite (#12590)
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.
2026-02-25 13:51:53 +00:00

69 lines
1.6 KiB
Lean4
Raw Permalink Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
import Lean
open Lean
open Lean.Meta
inductive Vec (α : Type) : Nat Type
| nil : Vec α 0
| cons {n} : α Vec α n Vec α (n+1)
set_option trace.Meta.debug true
def nat := mkConst `Nat
def succ := mkConst `Nat.succ
def tst1 : MetaM Unit :=
withLocalDeclD `n nat fun n => do
let n1 := mkApp succ n
let vecN1 := mkApp2 (mkConst `Vec) nat n1
withLocalDeclD `xs vecN1 fun xs => do
generalizeTelescope #[n1, xs] fun ys => do
let t mkLambdaFVars ys ys.back!
trace[Meta.debug] t
pure ()
/-- trace: [Meta.debug] fun x x_1 => x_1 -/
#guard_msgs in
#eval tst1
set_option pp.all true
def tst2 : MetaM Unit :=
withLocalDeclD `n nat fun n => do
let n1 := mkApp succ n
let vecN1 := mkApp2 (mkConst `Vec) nat n1
withLocalDeclD `xs vecN1 $ fun xs => do
let e mkEqRefl xs
generalizeTelescope #[n1, xs, e] fun ys => do
let t mkLambdaFVars ys ys.back!
trace[Meta.debug] t
pure ()
/--
trace: [Meta.debug] fun (x : Nat) (x_1 : Vec Nat x) (x_2 : @Eq.{1} (Vec Nat x) x_1 x_1) => x_2
-/
#guard_msgs in
#eval tst2
def failIfSuccess (x : MetaM Unit) : MetaM Unit := do
let worked try x; pure true catch _ => pure false
if worked then
throwError "unexpected success"
def tst3 : MetaM Unit :=
withLocalDeclD `n nat $ fun n => do
let n1 := mkApp succ n
let vecN1 := mkApp2 (mkConst `Vec) nat n1
withLocalDeclD `xs vecN1 $ fun xs => do
let e mkEqRefl xs
failIfSuccess do
generalizeTelescope #[n1, e] fun ys => do
let t mkLambdaFVars ys ys.back!
trace[Meta.debug] t
pure ()
trace[Meta.debug] "failed as expected"
/-- trace: [Meta.debug] failed as expected -/
#guard_msgs in
#eval tst3