lean4-htt/tests/lean/run/trace.lean
Kim Morrison 3a457e6ad6
chore: use #guard_msgs in run tests (#4175)
Many of our tests in `tests/lean/run/` produce output from `#eval` (or
`#check`) statements, that is then ignored.

This PR tries to capture all the useful output using `#guard_msgs`. I've
only done a cursory check that the output is still sane --- there is a
chance that some "unchecked" tests have already accumulated regressions
and this just cements them!

In the other direction, I did identify two rotten tests:
* a minor one in `setStructInstNotation.lean`, where a comment says `Set
Nat`, but `#check` actually prints `?_`. Weird?
* `CompilerProbe.lean` is generating empty output, apparently indicating
that something is broken, but I don't know the signficance of this file.

In any case, I'll ask about these elsewhere.

(This started by noticing that a recent `grind` test file had an
untested `trace_state`, and then got carried away.)
2024-05-16 00:38:31 +00:00

89 lines
2.1 KiB
Text

import Lean.CoreM
open Lean
structure MyState :=
(trace_state : TraceState := {})
(s : Nat := 0)
abbrev M := CoreM
def tst1 : M Unit :=
do trace[module] (m!"hello" ++ MessageData.nest 9 (m!"\n" ++ "world"));
trace[module.aux] "another message";
pure ()
def tst2 (b : Bool) : M Unit :=
withTraceNode `module (fun _ => return "message") do
tst1;
trace[bughunt] "at test2";
if b then throwError "error";
tst1;
pure ()
partial def ack : Nat → Nat → Nat
| 0, n => n+1
| m+1, 0 => ack m 1
| m+1, n+1 => ack m (ack (m+1) n)
def slow (b : Bool) : Nat :=
ack 4 (cond b 0 1)
def tst3 (b : Bool) : M Unit :=
do withTraceNode `module.slow (fun _ => return m!"slow: {slow b}") do {
tst2 b;
tst1
};
trace[bughunt] "at end of tst3";
-- Messages are computed lazily. The following message will only be computed
-- if `trace.slow is active.
trace[slow] (m!"slow message: " ++ toString (slow b))
-- This is true even if it is a monad computation:
trace[slow] (m!"slow message: " ++ (← pure (toString (slow b))))
def run (x : M Unit) : M Unit :=
withReader
(fun ctx =>
-- Try commeting/uncommeting the following `setBool`s
let opts := ctx.options;
let opts := opts.setBool `trace.module true;
-- let opts := opts.setBool `trace.module.aux false;
let opts := opts.setBool `trace.bughunt true;
-- let opts := opts.setBool `trace.slow true;
{ ctx with options := opts })
(tryCatch (tryFinally x printTraces) (fun _ => IO.println "ERROR"))
/--
info: [module] message
[module] hello
world
[bughunt] at test2
ERROR
[module] message
[module] hello
world
[bughunt] at test2
-/
#guard_msgs in
#eval run (tst3 true)
/--
info: [module] message
[module] hello
world
[bughunt] at test2
[module] hello world
[module] hello
world
[bughunt] at end of tst3
[module] message
[module] hello
world
[bughunt] at test2
[module] hello world
[module] hello
world
[bughunt] at end of tst3
-/
#guard_msgs in
#eval run (tst3 false)