max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2
lean4-htt/tests/lean/run/hashmap.lean
Markus Himmel 3e0ea762b8
feat: Std.HashMap (#4583) 
### Preliminary PRs:

- [x] #4597 
- [x] #4599
- [x] #4600
- [x] #4602
- [x] #4603
- [x] #4604
- [x] #4605
- [x] #4607
- [x] #4627
- [x] #4629 

### Quick overview over API/naming changes compared to `Lean.HashMap`
and `Batteries.HashMap`:
#### Lean

* `find?` -> `get?`/`getElem?`
* `find!` -> `get!`/`gtetElem!`
* `findD` -> `getD`
* `findEntry?` -> not implemented for now
* `insert'` -> `containsThenInsert` (order reversed in result)
* `insertIfNew` -> `getThenInsertIfNew?` (order reversed in result)
* `numBuckets` -> `Internal.numBuckets`
* `ofListWith` -> not implemented for now
* `Array.groupByKey` -> not implemented for now
* `merge` -> not implemented for now, but you can use `insertMany`

#### Batteries

* `modify` -> not implemented for now
* `mergeWith` -> not implemented for now
* `mergeWithM` -> not implemented for now
2024-07-05 10:14:20 +00:00

443 lines
9.7 KiB
Text
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import Std.Data.HashSet.Basic
import Std.Data.HashSet.Raw
import Std.Data.HashMap.AdditionalOperations
open Std
/-! Basic tests for (DHashMap|HashMap|HashSet)(.Raw)? functions which do not (yet) have proofs -/
namespace DHashMap.Raw
def m : DHashMap.Raw Nat (fun _ => Nat) :=
.ofList [⟨1, 2⟩, ⟨2, 4⟩, ⟨3, 6⟩]
/-- info: [⟨2, 8⟩] -/
#guard_msgs in
#eval (m.filterMap fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [⟨3, none⟩, ⟨2, some 8⟩, ⟨1, none⟩] -/
#guard_msgs in
#eval (m.map fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [⟨3, 6⟩] -/
#guard_msgs in
#eval (m.filter fun _ v => v > 4).toList
def addValueToStateAndAddKey (acc : Nat) (k : Nat) (v : Nat) : StateM Nat Nat := do
modify (fun st => st + v)
return acc + k
/-- info: (18, 25) -/
#guard_msgs in
#eval (m.foldM (init := 12) addValueToStateAndAddKey).run 13
/-- info: 18 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k _ => acc + k
def addValueToState (_ : Nat) (v : Nat) : StateM Nat PUnit := do
modify (fun st => st + v)
/-- info: ((), 25) -/
#guard_msgs in
#eval (m.forM addValueToState).run 13
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval Id.run do
let mut ans : List (Nat × Nat) := []
for ⟨k, v⟩ in m do
ans := (k, v) :: ans
return ans
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩] -/
#guard_msgs in
#eval m.toList
/-- info: #[⟨1, 2⟩, ⟨2, 4⟩, ⟨3, 6⟩] -/
#guard_msgs in
#eval m.toArray
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval DHashMap.Raw.Const.toList m
/-- info: #[(1, 2), (2, 4), (3, 6)] -/
#guard_msgs in
#eval DHashMap.Raw.Const.toArray m
/-- info: [3, 2, 1] -/
#guard_msgs in
#eval m.keys
/-- info: #[1, 2, 3] -/
#guard_msgs in
#eval m.keysArray
/-- info: [6, 4, 2] -/
#guard_msgs in
#eval m.values
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩, ⟨16, 9⟩] -/
#guard_msgs in
#eval (m.insertMany [⟨16, 8⟩, ⟨16, 9⟩]).toList
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩, ⟨16, 9⟩] -/
#guard_msgs in
#eval (DHashMap.Raw.Const.insertMany m [(16, 8), (16, 9)]).toList
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩] -/
#guard_msgs in
#eval (DHashMap.Raw.Const.ofList [(1, 2), (2, 4), (3, 6)]).toList
/-- info: Std.DHashMap.Raw.ofList [⟨1, 4⟩] -/
#guard_msgs in
#eval DHashMap.Raw.Const.ofList [(1, 2), (1, 4)]
end DHashMap.Raw
namespace DHashMap
def m : DHashMap Nat (fun _ => Nat) :=
.ofList [⟨1, 2⟩, ⟨2, 4⟩, ⟨3, 6⟩]
/-- info: [⟨2, 8⟩] -/
#guard_msgs in
#eval (m.filterMap fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [⟨3, none⟩, ⟨2, some 8⟩, ⟨1, none⟩] -/
#guard_msgs in
#eval (m.map fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [⟨3, 6⟩] -/
#guard_msgs in
#eval (m.filter fun _ v => v > 4).toList
def addValueToStateAndAddKey (acc : Nat) (k : Nat) (v : Nat) : StateM Nat Nat := do
modify (fun st => st + v)
return acc + k
/-- info: (18, 25) -/
#guard_msgs in
#eval (m.foldM (init := 12) addValueToStateAndAddKey).run 13
/-- info: 18 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k _ => acc + k
def addValueToState (_ : Nat) (v : Nat) : StateM Nat PUnit := do
modify (fun st => st + v)
/-- info: ((), 25) -/
#guard_msgs in
#eval (m.forM addValueToState).run 13
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval Id.run do
let mut ans : List (Nat × Nat) := []
for ⟨k, v⟩ in m do
ans := (k, v) :: ans
return ans
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩] -/
#guard_msgs in
#eval m.toList
/-- info: #[⟨1, 2⟩, ⟨2, 4⟩, ⟨3, 6⟩] -/
#guard_msgs in
#eval m.toArray
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval DHashMap.Const.toList m
/-- info: #[(1, 2), (2, 4), (3, 6)] -/
#guard_msgs in
#eval DHashMap.Const.toArray m
/-- info: [3, 2, 1] -/
#guard_msgs in
#eval m.keys
/-- info: #[1, 2, 3] -/
#guard_msgs in
#eval m.keysArray
/-- info: [6, 4, 2] -/
#guard_msgs in
#eval m.values
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩, ⟨16, 9⟩] -/
#guard_msgs in
#eval (m.insertMany [⟨16, 8⟩, ⟨16, 9⟩]).toList
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩, ⟨16, 9⟩] -/
#guard_msgs in
#eval (DHashMap.Const.insertMany m [(16, 8), (16, 9)]).toList
/-- info: [⟨3, 6⟩, ⟨2, 4⟩, ⟨1, 2⟩] -/
#guard_msgs in
#eval (DHashMap.Const.ofList [(1, 2), (2, 4), (3, 6)]).toList
/-- info: Std.DHashMap.ofList [⟨1, 4⟩] -/
#guard_msgs in
#eval DHashMap.Const.ofList [(1, 2), (1, 4)]
end DHashMap
namespace HashMap.Raw
def m : HashMap.Raw Nat Nat :=
.ofList [(1, 2), (2, 4), (3, 6)]
/-- info: [(2, 8)] -/
#guard_msgs in
#eval (m.filterMap fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [(3, none), (2, some 8), (1, none)] -/
#guard_msgs in
#eval (m.map fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [(3, 6)] -/
#guard_msgs in
#eval (m.filter fun _ v => v > 4).toList
def addValueToStateAndAddKey (acc : Nat) (k : Nat) (v : Nat) : StateM Nat Nat := do
modify (fun st => st + v)
return acc + k
/-- info: (18, 25) -/
#guard_msgs in
#eval (m.foldM (init := 12) addValueToStateAndAddKey).run 13
/-- info: 18 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k _ => acc + k
def addValueToState (_ : Nat) (v : Nat) : StateM Nat PUnit := do
modify (fun st => st + v)
/-- info: ((), 25) -/
#guard_msgs in
#eval (m.forM addValueToState).run 13
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval Id.run do
let mut ans : List (Nat × Nat) := []
for (k, v) in m do
ans := (k, v) :: ans
return ans
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval m.toList
/-- info: #[(1, 2), (2, 4), (3, 6)] -/
#guard_msgs in
#eval m.toArray
/-- info: [3, 2, 1] -/
#guard_msgs in
#eval m.keys
/-- info: #[1, 2, 3] -/
#guard_msgs in
#eval m.keysArray
/-- info: [6, 4, 2] -/
#guard_msgs in
#eval m.values
/-- info: [(3, 6), (2, 4), (1, 2), (16, 9)] -/
#guard_msgs in
#eval (m.insertMany [(16, 8), (16, 9)]).toList
/-- info: Std.HashMap.Raw.ofList [(1, 4)] -/
#guard_msgs in
#eval HashMap.Raw.ofList [(1, 2), (1, 4)]
end HashMap.Raw
namespace HashMap
def m : HashMap Nat Nat :=
.ofList [(1, 2), (2, 4), (3, 6)]
/-- info: [(2, 8)] -/
#guard_msgs in
#eval (m.filterMap fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [(3, none), (2, some 8), (1, none)] -/
#guard_msgs in
#eval (m.map fun k v => if k % 2 == 0 then some (2 * v) else none).toList
/-- info: [(3, 6)] -/
#guard_msgs in
#eval (m.filter fun _ v => v > 4).toList
def addValueToStateAndAddKey (acc : Nat) (k : Nat) (v : Nat) : StateM Nat Nat := do
modify (fun st => st + v)
return acc + k
/-- info: (18, 25) -/
#guard_msgs in
#eval (m.foldM (init := 12) addValueToStateAndAddKey).run 13
/-- info: 18 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k _ => acc + k
def addValueToState (_ : Nat) (v : Nat) : StateM Nat PUnit := do
modify (fun st => st + v)
/-- info: ((), 25) -/
#guard_msgs in
#eval (m.forM addValueToState).run 13
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval Id.run do
let mut ans : List (Nat × Nat) := []
for (k, v) in m do
ans := (k, v) :: ans
return ans
/-- info: [(3, 6), (2, 4), (1, 2)] -/
#guard_msgs in
#eval m.toList
/-- info: #[(1, 2), (2, 4), (3, 6)] -/
#guard_msgs in
#eval m.toArray
/-- info: [3, 2, 1] -/
#guard_msgs in
#eval m.keys
/-- info: #[1, 2, 3] -/
#guard_msgs in
#eval m.keysArray
/-- info: [6, 4, 2] -/
#guard_msgs in
#eval m.values
/-- info: [(3, 6), (2, 4), (1, 2), (16, 9)] -/
#guard_msgs in
#eval (m.insertMany [(16, 8), (16, 9)]).toList
/-- info: Std.HashMap.ofList [(1, 4)] -/
#guard_msgs in
#eval HashMap.ofList [(1, 2), (1, 4)]
end HashMap
namespace HashSet.Raw
def m : HashSet.Raw Nat :=
.ofList [1, 2, 1000000000]
/-- info: [1000000000] -/
#guard_msgs in
#eval (m.filter fun v => v > 4).toList
def addKeyToStateAndAddKey (acc : Nat) (k : Nat) : StateM Nat Nat := do
modify (fun st => st + k)
return acc + k
/-- info: (1000000015, 1000000016) -/
#guard_msgs in
#eval (m.foldM (init := 12) addKeyToStateAndAddKey).run 13
/-- info: 1000000015 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k => acc + k
def addKeyToState (k : Nat) : StateM Nat PUnit := do
modify (fun st => st + k)
/-- info: ((), 1000000016) -/
#guard_msgs in
#eval (m.forM addKeyToState).run 13
/-- info: [1000000000, 2, 1] -/
#guard_msgs in
#eval Id.run do
let mut ans : List Nat := []
for k in m do
ans := k :: ans
return ans
/-- info: [1000000000, 2, 1] -/
#guard_msgs in
#eval m.toList
/-- info: #[1, 2, 1000000000] -/
#guard_msgs in
#eval m.toArray
/-- info: [1000000000, 2, 1, 16] -/
#guard_msgs in
#eval (m.insertMany [16, 16]).toList
/-- info: Std.HashSet.Raw.ofList [100, 1] -/
#guard_msgs in
#eval HashSet.Raw.ofList [1, 100]
end HashSet.Raw
namespace HashSet
def m : HashSet Nat :=
.ofList [1, 2, 1000000000]
/-- info: [1000000000] -/
#guard_msgs in
#eval (m.filter fun v => v > 4).toList
def addKeyToStateAndAddKey (acc : Nat) (k : Nat) : StateM Nat Nat := do
modify (fun st => st + k)
return acc + k
/-- info: (1000000015, 1000000016) -/
#guard_msgs in
#eval (m.foldM (init := 12) addKeyToStateAndAddKey).run 13
/-- info: 1000000015 -/
#guard_msgs in
#eval m.fold (init := 12) fun acc k => acc + k
def addKeyToState (k : Nat) : StateM Nat PUnit := do
modify (fun st => st + k)
/-- info: ((), 1000000016) -/
#guard_msgs in
#eval (m.forM addKeyToState).run 13
/-- info: [1000000000, 2, 1] -/
#guard_msgs in
#eval Id.run do
let mut ans : List Nat := []
for k in m do
ans := k :: ans
return ans
/-- info: [1000000000, 2, 1] -/
#guard_msgs in
#eval m.toList
/-- info: #[1, 2, 1000000000] -/
#guard_msgs in
#eval m.toArray
/-- info: [1000000000, 2, 1, 16] -/
#guard_msgs in
#eval (m.insertMany [16, 16]).toList
/-- info: Std.HashSet.ofList [100, 1] -/
#guard_msgs in
#eval HashSet.ofList [1, 100]
end HashSet
Reference in a new issue View git blame Copy permalink