Files
lean4/tests/elab/finally.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

124 lines
2.5 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.
def checkM (b : IO Bool) : IO Unit :=
unless ( b) do throw $ IO.userError "failed"
abbrev M := ExceptT String $ StateRefT Nat IO
def f1 : M Nat :=
throw "error 1"
def f2 : M Nat :=
throwThe IO.Error $ IO.userError "error 2"
def tst1 : M Nat := do
try
try f1 finally set 100; IO.println "finisher executed"
catch _ =>
get
def checkE {α ε : Type} [BEq α] (x : IO (Except ε α)) (expected : α) : IO Unit := do
let r x;
match r with
| Except.ok a => unless a == expected do throw $ IO.userError "unexpected result"
| Except.error _ => throw $ IO.userError "unexpected error"
/--
info: finisher executed
---
info: Except.ok 100
-/
#guard_msgs in
#eval (tst1.run).run' 0
/-- info: finisher executed -/
#guard_msgs in
#eval checkE ((tst1.run).run' 0) 100
def tst2 : M Nat :=
tryCatchThe IO.Error
(tryFinally f2 (do set 100; IO.println "finisher executed"))
(fun _ => get)
/--
info: finisher executed
---
info: Except.ok 100
-/
#guard_msgs in
#eval (tst2.run).run' 0
/-- info: finisher executed -/
#guard_msgs in
#eval checkE ((tst2.run).run' 0) 100
def tst3 : M Nat :=
tryCatchThe IO.Error
(tryFinally
(tryFinally f1 (do set 100; IO.println "inner finisher executed"; discard $ f2; pure ()))
(do modify Nat.succ; IO.println "outer finisher executed"))
(fun _ => get)
/--
info: inner finisher executed
outer finisher executed
---
info: Except.ok 101
-/
#guard_msgs in
#eval (tst3.run).run' 0
/--
info: inner finisher executed
outer finisher executed
-/
#guard_msgs in
#eval checkE ((tst3.run).run' 0) 101
def tst4 : M Nat := do
let a tryFinally
(tryFinally (pure 42) (do set 100; IO.println "inner finisher executed"; pure ()))
(do modify Nat.succ; IO.println "outer finisher executed");
let s get;
pure (a + s)
/--
info: inner finisher executed
outer finisher executed
---
info: Except.ok 143
-/
#guard_msgs in
#eval (tst4.run).run' 0
/--
info: inner finisher executed
outer finisher executed
-/
#guard_msgs in
#eval checkE ((tst4.run).run' 0) 143
def tst5 : M Nat := do
let (a, _) tryFinally' (pure 42) (fun a? => do IO.println ("finalizer received: " ++ toString a?));
pure a
/--
info: finalizer received: (some 42)
---
info: Except.ok 42
-/
#guard_msgs in
#eval (tst5.run).run' 0
def tst6 : M Nat := do
let (a, _) tryFinally' f2 (fun a? => do IO.println ("finalizer received: " ++ toString a?));
pure a
def tst7 : IO Unit :=
tryCatchThe IO.Error (do discard $ (tst6.run).run' 0; pure ()) (fun _ => IO.println "failed as expected")
/--
info: finalizer received: none
failed as expected
-/
#guard_msgs in
#eval tst7