File size: 1,922 Bytes
e36aeda | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | // Copyright 2024 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package maps
import (
"slices"
"testing"
)
func TestAll(t *testing.T) {
for size := 0; size < 10; size++ {
m := make(map[int]int)
for i := range size {
m[i] = i
}
cnt := 0
for i, v := range All(m) {
v1, ok := m[i]
if !ok || v != v1 {
t.Errorf("at iteration %d got %d, %d want %d, %d", cnt, i, v, i, v1)
}
cnt++
}
if cnt != size {
t.Errorf("read %d values expected %d", cnt, size)
}
}
}
func TestKeys(t *testing.T) {
for size := 0; size < 10; size++ {
var want []int
m := make(map[int]int)
for i := range size {
m[i] = i
want = append(want, i)
}
var got []int
for k := range Keys(m) {
got = append(got, k)
}
slices.Sort(got)
if !slices.Equal(got, want) {
t.Errorf("Keys(%v) = %v, want %v", m, got, want)
}
}
}
func TestValues(t *testing.T) {
for size := 0; size < 10; size++ {
var want []int
m := make(map[int]int)
for i := range size {
m[i] = i
want = append(want, i)
}
var got []int
for v := range Values(m) {
got = append(got, v)
}
slices.Sort(got)
if !slices.Equal(got, want) {
t.Errorf("Values(%v) = %v, want %v", m, got, want)
}
}
}
func TestInsert(t *testing.T) {
got := map[int]int{
1: 1,
2: 1,
}
Insert(got, func(yield func(int, int) bool) {
for i := 0; i < 10; i += 2 {
if !yield(i, i+1) {
return
}
}
})
want := map[int]int{
1: 1,
2: 1,
}
for i, v := range map[int]int{
0: 1,
2: 3,
4: 5,
6: 7,
8: 9,
} {
want[i] = v
}
if !Equal(got, want) {
t.Errorf("Insert got: %v, want: %v", got, want)
}
}
func TestCollect(t *testing.T) {
m := map[int]int{
0: 1,
2: 3,
4: 5,
6: 7,
8: 9,
}
got := Collect(All(m))
if !Equal(got, m) {
t.Errorf("Collect got: %v, want: %v", got, m)
}
}
|