Files
lean4/tests/elab/grind_map.lean
Garmelon 08eb78a5b2 chore: switch to new test/bench suite (#12590)
This PR sets up the new integrated test/bench suite. It then migrates
all benchmarks and some related tests to the new suite. There's also
some documentation and some linting.

For now, a lot of the old tests are left alone so this PR doesn't become
even larger than it already is. Eventually, all tests should be migrated
to the new suite though so there isn't a confusing mix of two systems.
2026-02-25 13:51:53 +00:00

62 lines
2.2 KiB
Lean4
Raw Permalink 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.
module
import Std.Data.HashMap
import Std.Data.DHashMap
import Std.Data.ExtHashMap
import Std.Data.HashSet
import Std.Data.TreeMap
open Std
section
variable [BEq α] [LawfulBEq α] [Hashable α] [LawfulHashable α ]
example : ( : DHashMap α β).isEmpty := by grind
example (m : DHashMap α β) (h : m = ) : m.isEmpty := by grind
example : ((( : HashMap Nat Nat).insert 3 6).insert 4 7).contains 3 := by grind
example : ((( : HashMap Nat Nat).insert 3 6).insert 4 7).contains 9 == false := by grind
example (m : HashMap Nat Nat) (h : m.contains 3) : (m.erase 2).contains 3 := by grind
example (m : HashMap Nat Nat) (h : (m.erase 2).contains 3) : m.contains 3 := by grind
example (m : HashMap Nat Nat) : (m.erase 3).contains 3 = false := by grind
example (m : HashMap Nat Nat) (h : m.contains 3 = false) : (m.erase 2).contains 3 = false := by grind
-- Insert twice
example (m : HashMap Nat Nat) : m.size ((m.insert 1 2).insert 3 4).size := by grind
example (m : HashMap Nat Nat) : ((m.insert 1 2).insert 3 4).size m.size + 2 := by grind
-- Insert the same key twice
example (m : HashMap Nat Nat) : m.size ((m.insert 1 2).insert 1 4).size := by grind
example (m : HashMap Nat Nat) : ((m.insert 1 2).insert 1 4).size m.size + 1 := by grind
example : ((( : HashMap Nat Nat).insert 3 6).erase 4)[3]? = some 6 := by grind
open scoped HashMap in
example (m : HashMap Nat Nat) :
(m.insert 1 2).filter (fun k _ => k > 1000) ~m m.filter fun k _ => k > 1000 := by
apply HashMap.Equiv.of_forall_getElem?_eq
grind
example [BEq α] [LawfulBEq α] [Hashable α] [LawfulHashable α]
{m : HashMap α β} {f : α β γ} {k : α} :
(m.map f)[k]? = m[k]?.map (f k) := by
grind
example (m : Std.TreeMap Nat Bool) : (m.insert 37 true)[32]? = m[32]? := by
grind
example (m : HashMap Nat Nat) : ((m.alter 5 id).erase 7).size m.size - 1 := by grind
example (m : ExtHashMap Nat Nat) :
(m.insert 1 2).filter (fun k _ => k > 1000) = (m.insert 1 3).filter fun k _ => k > 1000 := by
ext1 k
grind
example (m : ExtHashMap Nat Nat) :
(((m.insert 1 2).insert 3 4).insert 5 6).filter (fun k _ => k > 6) = m.filter fun k _ => k > 6 := by
ext1 k
grind
end