Skip to content

Commit 5472225

Browse files
committed
Add example specs for Take and Drop
1 parent 31476e7 commit 5472225

File tree

2 files changed

+19
-0
lines changed

2 files changed

+19
-0
lines changed

iter/drop_test.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,19 @@
11
package iter_test
22

33
import (
4+
"fmt"
45
"testing"
56

67
"github.com/BooleanCat/go-functional/internal/assert"
78
"github.com/BooleanCat/go-functional/iter"
89
)
910

11+
func ExampleDrop() {
12+
counter := iter.Drop[int](iter.Count(), 2)
13+
fmt.Println(counter.Next().Unwrap())
14+
// Output: 2
15+
}
16+
1017
func TestDrop(t *testing.T) {
1118
counter := iter.Drop[int](iter.Count(), 2)
1219
assert.Equal(t, counter.Next().Unwrap(), 2)

iter/take_test.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,24 @@
11
package iter_test
22

33
import (
4+
"fmt"
45
"testing"
56

67
"github.com/BooleanCat/go-functional/internal/assert"
78
"github.com/BooleanCat/go-functional/iter"
89
)
910

11+
func ExampleTake() {
12+
iter := iter.Take[int](iter.Count(), 2)
13+
fmt.Println(iter.Next())
14+
fmt.Println(iter.Next())
15+
fmt.Println(iter.Next())
16+
// Output:
17+
// Some(0)
18+
// Some(1)
19+
// None
20+
}
21+
1022
func TestTakeIter(t *testing.T) {
1123
iter := iter.Take[int](iter.Count(), 2)
1224
assert.Equal(t, iter.Next().Unwrap(), 0)

0 commit comments

Comments
 (0)