mirror of https://github.com/golang/go.git
117 lines
1.9 KiB
Go
117 lines
1.9 KiB
Go
// 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)
|
|
}
|
|
}
|