Accumulate accept any type

Author: mowshon

accumulate.go

@@ -2,14 +2,14 @@ package iterium
 
 // Accumulate returns an iterator that sends the accumulated
 // result from the binary function to the channel.
-func Accumulate[N Number](iterable Iter[N], operator func(N, N) N) Iter[N] {
-	iter := Instance[N](iterable.Count(), iterable.IsInfinite())
+func Accumulate[T any](iterable Iter[T], operator func(T, T) T) Iter[T] {
+	iter := Instance[T](iterable.Count(), iterable.IsInfinite())
 
 	go func() {
 		defer iterRecover()
 		defer iter.Close()
 
-		var last N
+		var last T
 		var start bool
 		for true {
 			next, err := iterable.Next()

accumulate_test.go

@@ -1,6 +1,7 @@
 package iterium
 
 import (
+	"fmt"
 	"github.com/stretchr/testify/assert"
 	"testing"
 )
@@ -24,3 +25,15 @@ func TestAccumulate(t *testing.T) {
 		assert.Exactly(t, []int{1, 2, 6, 24, 120}, slice)
 	}
 }
+
+func mergeAcc(first, second string) string {
+	return fmt.Sprintf("%s-%s", first, second)
+}
+
+func TestAccumulateString(t *testing.T) {
+	data := New("A", "B", "C", "D")
+	accumulateString := Accumulate(data, mergeAcc)
+	if slice, err := accumulateString.Slice(); assert.Nil(t, err) {
+		assert.Exactly(t, []string{"A", "A-B", "A-B-C", "A-B-C-D"}, slice)
+	}
+}