lean4-htt/tests/compiler/array_test.lean
Leonardo de Moura 0f43c2e2d9 feat(library/init/data/array/basic): efficient heterogeneous Array.map
This commit also removes Array.hmap.
Motivation: I wanted to use Array.hmap as an example in the paper, but
I found it would be too distracting to explain why we had `Array.hmap`
and `Array.map`.

cc @kha
2019-05-25 16:32:59 -07:00

38 lines
1.2 KiB
Text

def foo (a : Array Nat) : Array Nat :=
let a := a.push 0 in
let a := a.push 1 in
let a := a.push 2 in
let a := a.push 3 in
a
def main : IO UInt32 :=
do
let a : Array Nat := Array.empty,
IO.println (toString a),
IO.println (toString a.sz),
let a := foo a,
IO.println (toString a),
let a := a.map (+10),
IO.println (toString a),
IO.println (toString a.sz),
let a1 := a.pop,
let a2 := a.push 100,
IO.println (toString a1),
IO.println (toString a2),
let a2 := a.pop,
IO.println a2,
IO.println $ (([1, 2, 3, 4].toArray).map (+2)).map toString,
IO.println $ ([1, 2, 3, 4].toArray.extract 1 3),
IO.println $ ([1, 2, 3, 4].toArray.extract 0 100),
IO.println $ ([1, 2, 3, 4].toArray.extract 1 1),
IO.println $ ([1, 2, 3, 4].toArray.extract 2 4),
IO.println [1,2,3,4].toArray.reverse,
IO.println ([] : List Nat).toArray.reverse,
IO.println [1,2,3].toArray.reverse,
IO.println $ [1,2,3,4].toArray.filter (λ a, a % 2 == 0),
IO.println $ [1,2,3,4,5].toArray.filter (λ a, a % 2 == 0),
IO.println $ [1,2,3,4,5].toArray.filter (λ a, a % 2 == 1),
IO.println $ [1,2,3,4].toArray.filter (>2),
IO.println $ [1,2,3,4].toArray.filter (>10),
IO.println $ [1,2,3,4].toArray.filter (>0),
pure 0