PyPantograph/src/Test/Library.lean

import Pantograph.Library
import Test.Common

namespace Pantograph.Test.Library

open Lean

def runTermElabM { α } (termElabM: Elab.TermElabM α): CoreM α :=
  termElabM.run' (ctx := defaultElabContext) |>.run'

def test_expr_echo (env: Environment): IO LSpec.TestSeq := do
  let inner: CoreM LSpec.TestSeq := do
    let prop_and_proof := "⟨∀ (x: Prop), x → x, λ (x: Prop) (h: x) => h⟩"
    let tests := LSpec.TestSeq.done
    let echoResult ← runTermElabM $ exprEcho prop_and_proof (options := {})
    let tests := tests.append (LSpec.test "fail" (echoResult.toOption == .some {
      type := { pp? := "?m.1" }, expr := { pp? := "?m.2" }
    }))
    let echoResult ← runTermElabM $ exprEcho prop_and_proof (expectedType? := .some "Σ' p:Prop, p") (options := { printExprAST := true })
    let tests := tests.append (LSpec.test "fail" (echoResult.toOption == .some {
      type := {
        pp? := "(p : Prop) ×' p",
        sexp? := "((:c PSigma) (:sort 0) (:lambda p (:sort 0) 0))",
      },
      expr := {
        pp? := "⟨∀ (x : Prop), x → x, fun x h => h⟩",
        sexp? := "((:c PSigma.mk) (:sort 0) (:lambda p (:sort 0) 0) (:forall x (:sort 0) (:forall a 0 1)) (:lambda x (:sort 0) (:lambda h 0 0)))",
      }
    }))
    return tests
  runCoreMSeq env (options := #["pp.proofs.threshold=100"]) inner

def test_core_context (env : Environment) : IO LSpec.TestSeq := runTest do
  let coreM := runTermElabM $ runTest do
    let goal := (← Meta.mkFreshExprSyntheticOpaqueMVar (.const `Nat [])).mvarId!
    let (_, _) ← (Tactic.collatz 27).run { elaborator := .anonymous} |>.run { goals := [goal] }
    fail "should fail"
  let coreContext ← createCoreContext (options := #["maxRecDepth=10"])
  match ← (coreM.run' coreContext { env }).toBaseIO with
  | .error exception =>
    let message ← exception.toMessageData.toString
    checkEq "exception" message "maximum recursion depth has been reached\nuse `set_option maxRecDepth <num>` to increase limit\nuse `set_option diagnostics true` to get diagnostic information"
  | .ok _ =>
    fail "macRecDepth set should fail"
  let coreContext ← createCoreContext (options := #["maxRecDepth=200"])
  match ← (coreM.run' coreContext { env }).toBaseIO with
  | .error exception =>
    let message ← exception.toMessageData.toString
    fail s!"Exception: {message}"
  | .ok _ =>
    pure ()

def suite (env: Environment): List (String × IO LSpec.TestSeq) :=
  [
    ("expr_echo", test_expr_echo env),
    ("core.context", test_core_context env)
  ]