diff --git a/itlib/chunk_test.go b/itlib/chunk_test.go index 602b311..661c09a 100644 --- a/itlib/chunk_test.go +++ b/itlib/chunk_test.go @@ -71,7 +71,7 @@ func TestChunk(t *testing.T) { "A", "B", "C", "D", "E", "F", "G", }, tx: func(el itkit.Iterator[string]) itkit.Iterator[string] { - return itlib.Limit(el, 2) + return itlib.Limit(2, el) }, exp: [][]string{ {"A", "B"}, diff --git a/itlib/limit.go b/itlib/limit.go index 1af1622..51c7acd 100644 --- a/itlib/limit.go +++ b/itlib/limit.go @@ -47,6 +47,6 @@ func newLimitIterator[T any](n uint, src itkit.Iterator[T]) *LimitIterator[T] { } // Limit returns a new [LimitIterator] instance. -func Limit[T any](src itkit.Iterator[T], n uint) itkit.Iterator[T] { +func Limit[T any](n uint, src itkit.Iterator[T]) itkit.Iterator[T] { return newLimitIterator(n, src) } diff --git a/itlib/limit_test.go b/itlib/limit_test.go index 0202860..49a23e3 100644 --- a/itlib/limit_test.go +++ b/itlib/limit_test.go @@ -25,7 +25,7 @@ import ( func TestLimit(t *testing.T) { t.Run("empty", func(t *testing.T) { - it := itlib.Limit(itlib.Empty[any](), 10) + it := itlib.Limit(10, itlib.Empty[any]()) assert.False(t, it.Next()) }) @@ -34,7 +34,7 @@ func TestLimit(t *testing.T) { src := rangeit.Range(10) - it := itlib.Limit(src, 4) + it := itlib.Limit(4, src) asserter.Equal([]int{0, 1, 2, 3}, sliceit.To(it)) @@ -49,7 +49,7 @@ func TestLimit(t *testing.T) { src := rangeit.Range(4) - it := itlib.Limit(src, 10) + it := itlib.Limit(10, src) asserter.Equal([]int{0, 1, 2, 3}, sliceit.To(it)) diff --git a/itlib/tee_test.go b/itlib/tee_test.go index c7b782d..f49abf5 100644 --- a/itlib/tee_test.go +++ b/itlib/tee_test.go @@ -150,7 +150,7 @@ func TestTee(t *testing.T) { it := funcit.PullFn(func() (*int, bool) { return c.new(), true }) - it = itlib.Limit(it, 10) + it = itlib.Limit(10, it) l, r := itlib.Tee(it) @@ -161,7 +161,7 @@ func TestTee(t *testing.T) { asserter.Equal(c.n, uint32(10)) // Consume half on the right side. - s := sliceit.To(itlib.Limit(r.Iter(), 5)) + s := sliceit.To(itlib.Limit(5, r.Iter())) asserter.Len(s, 5) // Force garbage collection to clean up references.