mirror of
https://github.com/duke-git/lancet.git
synced 2026-02-04 12:52:28 +08:00
test: add unit test for Keys and Values func
This commit is contained in:
@@ -6,7 +6,7 @@ package maputil
|
||||
|
||||
// Keys returns a slice of the map's keys
|
||||
func Keys[K comparable, V any](m map[K]V) []K {
|
||||
keys := make([]K, len(m))
|
||||
keys := make([]K, 0, len(m))
|
||||
|
||||
for k := range m {
|
||||
keys = append(keys, k)
|
||||
@@ -17,7 +17,7 @@ func Keys[K comparable, V any](m map[K]V) []K {
|
||||
|
||||
// Values returns a slice of the map's values
|
||||
func Values[K comparable, V any](m map[K]V) []V {
|
||||
values := make([]V, len(m))
|
||||
values := make([]V, 0, len(m))
|
||||
|
||||
for _, v := range m {
|
||||
values = append(values, v)
|
||||
|
||||
42
maputil/map_test.go
Normal file
42
maputil/map_test.go
Normal file
@@ -0,0 +1,42 @@
|
||||
package maputil
|
||||
|
||||
import (
|
||||
"sort"
|
||||
"testing"
|
||||
|
||||
"github.com/duke-git/lancet/v2/internal"
|
||||
)
|
||||
|
||||
func TestKeys(t *testing.T) {
|
||||
assert := internal.NewAssert(t, "TestKeys")
|
||||
|
||||
m := map[int]string{
|
||||
1: "a",
|
||||
2: "a",
|
||||
3: "b",
|
||||
4: "c",
|
||||
5: "d",
|
||||
}
|
||||
|
||||
keys := Keys(m)
|
||||
sort.Ints(keys)
|
||||
|
||||
assert.Equal([]int{1, 2, 3, 4, 5}, keys)
|
||||
}
|
||||
|
||||
func TestValues(t *testing.T) {
|
||||
assert := internal.NewAssert(t, "TestValues")
|
||||
|
||||
m := map[int]string{
|
||||
1: "a",
|
||||
2: "a",
|
||||
3: "b",
|
||||
4: "c",
|
||||
5: "d",
|
||||
}
|
||||
|
||||
values := Values(m)
|
||||
sort.Strings(values)
|
||||
|
||||
assert.Equal([]string{"a", "a", "b", "c", "d"}, values)
|
||||
}
|
||||
Reference in New Issue
Block a user