lean4-htt/tests/lean/getElem.lean
Joachim Breitner 684f32fabe
feat: let get_elem_tactic_trivial handle [a]'h.2 (#3132)
The pattern
```
    for h : i in [:xs.size] do
      let x := xs[i]'h.2
```
is occassionally useful to iterate over an array with the index in
hand. This PR extends the `get_elem_tactic_trivial` so that one can
simply write
```
    for h : i in [:xs.size] do
      let x := xs[i]
```

fixes #3032.
2024-01-08 16:23:09 +00:00

44 lines
835 B
Text

def f1 (a : Array Nat) (i : Nat) :=
a[i]
def f2 (a : Array Nat) (i : Fin a.size) :=
a[i] -- Ok
def f3 (a : Array Nat) (h : n ≤ a.size) (i : Fin n) :=
a[i] -- Ok
opaque a : Array Nat
opaque n : Nat
axiom n_lt_a_size : n < a.size
def f4 (i : Nat) (h : i < n) :=
have : i < a.size := Nat.lt_trans h n_lt_a_size
a[i]
def f5 (i : Nat) (h : i < n) :=
a[i]'(Nat.lt_trans h n_lt_a_size)
def f6 (i : Nat) :=
a[i]!
def f7 (i : Nat) :=
a[i]?
#print f2
#print f3
#print f5
#print f6
#print f7
def withRange (xs : Array Nat) : Option Nat := Id.run do
for h : i in [:xs.size] do
if i == xs[i] then
return i
return none
def withTwoRanges (xs : Array Nat) : Option Nat := Id.run do
for h1 : i in [:xs.size] do
for h2 : j in [i + 1:xs.size] do
if xs[i] == xs[j] then
return i
return none