mirror of
https://github.com/IBM/fp-go.git
synced 2025-06-17 00:07:49 +02:00
29 lines
518 B
Go
29 lines
518 B
Go
|
package array
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
O "github.com/ibm/fp-go/option"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ArrayType = []int
|
||
|
|
|
||
|
|
func TestTraverse(t *testing.T) {
|
||
|
|
|
||
|
|
traverse := Traverse(
|
||
|
|
O.Of[ArrayType],
|
||
|
|
O.MonadMap[ArrayType, func(int) ArrayType],
|
||
|
|
O.MonadAp[int, ArrayType],
|
||
|
|
|
||
|
|
func(n int) O.Option[int] {
|
||
|
|
if n%2 == 0 {
|
||
|
|
return O.None[int]()
|
||
|
|
}
|
||
|
|
return O.Of(n)
|
||
|
|
})
|
||
|
|
|
||
|
|
assert.Equal(t, O.None[[]int](), traverse(ArrayType{1, 2}))
|
||
|
|
assert.Equal(t, O.Of(ArrayType{1, 3}), traverse(ArrayType{1, 3}))
|
||
|
|
}
|