go/fp-go
1
0
Fork 0
mirror of https://github.com/IBM/fp-go.git synced 2025-07-15 01:24:23 +02:00
Code Issues Releases Activity

fix: add Ap for iterators

Browse Source 
Signed-off-by: Dr. Carsten Leue <carsten.leue@de.ibm.com>
This commit is contained in:
Dr. Carsten Leue
2023-07-25 17:34:32 +02:00
parent 41b792b23a
commit d2346b016e
3 changed files with 82 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
iterator/stateless/iterator_test.go
View File

@ -17,11 +17,13 @@ package stateless
import (
"fmt"
"math"
"testing"
A "github.com/IBM/fp-go/array"
F "github.com/IBM/fp-go/function"
"github.com/IBM/fp-go/internal/utils"
O "github.com/IBM/fp-go/option"
"github.com/stretchr/testify/assert"
)
@ -38,7 +40,7 @@ func TestIterator(t *testing.T) {
func TestChain(t *testing.T) {
outer := FromArray[int](A.From(1, 2, 3))
outer := From(1, 2, 3)
inner := func(data int) Iterator[string] {
return F.Pipe2(
@ -58,3 +60,27 @@ func TestChain(t *testing.T) {
assert.Equal(t, A.From("item[1][0]", "item[1][1]", "item[2][0]", "item[2][1]", "item[3][0]", "item[3][1]"), total)
}
func isPrimeNumber(num int) bool {
if num <= 2 {
return true
}
sq_root := int(math.Sqrt(float64(num)))
for i := 2; i <= sq_root; i++ {
if num%i == 0 {
return false
}
}
return true
}
func TestFilterMap(t *testing.T) {
it := F.Pipe2(
MakeBy(100, utils.Inc),
FilterMap(O.FromPredicate(isPrimeNumber)),
ToArray[int],
)
assert.Equal(t, A.From(1, 2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97), it)
}