diff options
Diffstat (limited to 'tests/lean/External/Funs.lean')
-rw-r--r-- | tests/lean/External/Funs.lean | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/tests/lean/External/Funs.lean b/tests/lean/External/Funs.lean new file mode 100644 index 00000000..674aaebd --- /dev/null +++ b/tests/lean/External/Funs.lean @@ -0,0 +1,88 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [external]: function definitions +import Base +import External.Types +import External.FunsExternal +open Primitives +namespace external + +/- [external::swap]: forward function -/ +def swap (T : Type) (x : T) (y : T) (st : State) : Result (State × Unit) := + do + let (st0, _) ← core.mem.swap T x y st + let (st1, _) ← core.mem.swap_back0 T x y st st0 + let (st2, _) ← core.mem.swap_back1 T x y st st1 + Result.ret (st2, ()) + +/- [external::swap]: backward function 0 -/ +def swap_back + (T : Type) (x : T) (y : T) (st : State) (st0 : State) : + Result (State × (T × T)) + := + do + let (st1, _) ← core.mem.swap T x y st + let (st2, x0) ← core.mem.swap_back0 T x y st st1 + let (_, y0) ← core.mem.swap_back1 T x y st st2 + Result.ret (st0, (x0, y0)) + +/- [external::test_new_non_zero_u32]: forward function -/ +def test_new_non_zero_u32 + (x : U32) (st : State) : Result (State × core.num.nonzero.NonZeroU32) := + do + let (st0, opt) ← core.num.nonzero.NonZeroU32.new x st + core.option.Option.unwrap core.num.nonzero.NonZeroU32 opt st0 + +/- [external::test_vec]: forward function -/ +def test_vec : Result Unit := + do + let v := Vec.new U32 + let _ ← Vec.push U32 v (U32.ofInt 0) + Result.ret () + +/- Unit test for [external::test_vec] -/ +#assert (test_vec == .ret ()) + +/- [external::custom_swap]: forward function -/ +def custom_swap + (T : Type) (x : T) (y : T) (st : State) : Result (State × T) := + do + let (st0, _) ← core.mem.swap T x y st + let (st1, x0) ← core.mem.swap_back0 T x y st st0 + let (st2, _) ← core.mem.swap_back1 T x y st st1 + Result.ret (st2, x0) + +/- [external::custom_swap]: backward function 0 -/ +def custom_swap_back + (T : Type) (x : T) (y : T) (st : State) (ret0 : T) (st0 : State) : + Result (State × (T × T)) + := + do + let (st1, _) ← core.mem.swap T x y st + let (st2, _) ← core.mem.swap_back0 T x y st st1 + let (_, y0) ← core.mem.swap_back1 T x y st st2 + Result.ret (st0, (ret0, y0)) + +/- [external::test_custom_swap]: forward function -/ +def test_custom_swap + (x : U32) (y : U32) (st : State) : Result (State × Unit) := + do + let (st0, _) ← custom_swap U32 x y st + Result.ret (st0, ()) + +/- [external::test_custom_swap]: backward function 0 -/ +def test_custom_swap_back + (x : U32) (y : U32) (st : State) (st0 : State) : + Result (State × (U32 × U32)) + := + custom_swap_back U32 x y st (U32.ofInt 1) st0 + +/- [external::test_swap_non_zero]: forward function -/ +def test_swap_non_zero (x : U32) (st : State) : Result (State × U32) := + do + let (st0, _) ← swap U32 x (U32.ofInt 0) st + let (st1, (x0, _)) ← swap_back U32 x (U32.ofInt 0) st st0 + if x0 = (U32.ofInt 0) + then Result.fail Error.panic + else Result.ret (st1, x0) + +end external |