41 lines
871 B
Go
41 lines
871 B
Go
package simsx
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCollect(t *testing.T) {
|
|
src := []int{1, 2, 3}
|
|
got := Collect(src, func(a int) int { return a * 2 })
|
|
assert.Equal(t, []int{2, 4, 6}, got)
|
|
gotStrings := Collect(src, strconv.Itoa)
|
|
assert.Equal(t, []string{"1", "2", "3"}, gotStrings)
|
|
}
|
|
|
|
func TestFirst(t *testing.T) {
|
|
src := []string{"a", "b"}
|
|
assert.Equal(t, &src[1], First(src, func(a string) bool { return a == "b" }))
|
|
assert.Nil(t, First(src, func(a string) bool { return false }))
|
|
}
|
|
|
|
func TestOneOf(t *testing.T) {
|
|
src := []string{"a", "b"}
|
|
got := OneOf(randMock{next: 1}, src)
|
|
assert.Equal(t, "b", got)
|
|
// test with other type
|
|
src2 := []int{1, 2, 3}
|
|
got2 := OneOf(randMock{next: 2}, src2)
|
|
assert.Equal(t, 3, got2)
|
|
}
|
|
|
|
type randMock struct {
|
|
next int
|
|
}
|
|
|
|
func (x randMock) Intn(n int) int {
|
|
return x.next
|
|
}
|