ccfa57d657
Before we start writing a lot of Expr code. - `Expr.pi` does not make sense anymore. - `Expr.elet` is weird. `«let»` is too inconvenient to write. So, I used `letE` short for `letExpr`. GHC avoids this issue because keywords are lowercase and constructors are capitalized.
97 lines
2.9 KiB
Text
97 lines
2.9 KiB
Text
import Init.Lean.Expr
|
||
open Lean
|
||
|
||
def exprType : Expr := Expr.sort Level.one
|
||
def biDef := BinderInfo.default
|
||
def exprNat := Expr.const `Nat []
|
||
-- Type -> Type
|
||
def TypeArrowType := Expr.forallE `α biDef exprType exprType
|
||
-- Type -> Type
|
||
def TypeArrowType2 := Expr.forallE `β biDef exprType exprType
|
||
-- fun (x : Nat), x
|
||
def exprT1 := Expr.lam `x biDef exprNat (Expr.bvar 0)
|
||
-- fun (y : Nat), x
|
||
def exprT2 := Expr.lam `y biDef exprNat (Expr.bvar 0)
|
||
-- fun (x : Nat), (f x)
|
||
def exprT3 := Expr.lam `x biDef exprNat (Expr.app (Expr.const `f []) (Expr.bvar 0))
|
||
-- fun (x : Nat), (f x)
|
||
def exprT4 := Expr.lam `x BinderInfo.implicit exprNat (Expr.app (Expr.const `f []) (Expr.bvar 0))
|
||
|
||
def check (b : Bool) : IO Unit :=
|
||
unless b (throw "failed")
|
||
|
||
def tst1 : IO Unit :=
|
||
do IO.println TypeArrowType;
|
||
IO.println exprT1;
|
||
IO.println exprT2;
|
||
IO.println exprT3;
|
||
IO.println exprT4;
|
||
check (TypeArrowType == TypeArrowType2);
|
||
check (ExprStructEq.mk TypeArrowType != ExprStructEq.mk TypeArrowType2);
|
||
check (!Expr.equal TypeArrowType TypeArrowType2);
|
||
check (exprT1 == exprT2);
|
||
check (ExprStructEq.mk exprT1 != ExprStructEq.mk exprT2);
|
||
check (ExprStructEq.mk exprT1 == ExprStructEq.mk exprT1);
|
||
check (exprT3 == exprT4);
|
||
check (ExprStructEq.mk exprT3 != ExprStructEq.mk exprT4);
|
||
pure ()
|
||
|
||
#eval tst1
|
||
|
||
def tst2 : IO Unit :=
|
||
do let m1 : ExprMap Nat := {};
|
||
let m1 := m1.insert exprT1 10;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT2 == some 10);
|
||
check (m1.find exprT3 == none);
|
||
let m1 := m1.insert exprT4 20;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT3 == some 20);
|
||
IO.println (m1.find exprT1);
|
||
pure ()
|
||
|
||
#eval tst2
|
||
|
||
def tst3 : IO Unit :=
|
||
do let m1 : ExprStructMap Nat := {};
|
||
let m1 := m1.insert exprT1 10;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT2 == none);
|
||
check (m1.find exprT3 == none);
|
||
let m1 := m1.insert exprT4 20;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT4 == some 20);
|
||
check (m1.find exprT3 == none);
|
||
IO.println (m1.find exprT1);
|
||
pure ()
|
||
|
||
#eval tst3
|
||
|
||
def tst4 : IO Unit :=
|
||
do let m1 : PersistentExprMap Nat := {};
|
||
let m1 := m1.insert exprT1 10;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT2 == some 10);
|
||
check (m1.find exprT3 == none);
|
||
let m1 := m1.insert exprT4 20;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT3 == some 20);
|
||
IO.println (m1.find exprT1);
|
||
pure ()
|
||
|
||
#eval tst4
|
||
|
||
def tst5 : IO Unit :=
|
||
do let m1 : PersistentExprStructMap Nat := {};
|
||
let m1 := m1.insert exprT1 10;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT2 == none);
|
||
check (m1.find exprT3 == none);
|
||
let m1 := m1.insert exprT4 20;
|
||
check (m1.find exprT1 == some 10);
|
||
check (m1.find exprT4 == some 20);
|
||
check (m1.find exprT3 == none);
|
||
IO.println (m1.find exprT1);
|
||
pure ()
|
||
|
||
#eval tst5
|