underscore/parallel_map_test.go
Ruidy 1031038d42
feat: add Chunk, ContainsBy, UniqueBy, ParallelMap, map helpers
- Add `Chunk` to split slices into groups of size n.
- Add `ContainsBy` for predicate-based containment checks.
- Add `UniqueBy` to deduplicate slices by key selector.
- Add `ParallelMap` for concurrent mapping with context and error
handling.
- Add `maps.Keys` and `maps.Values` helpers for extracting map
keys/values.
- Update README and docs for new features.
- Refactor `Contains` to use `slices.Contains`.
2025-09-01 18:03:38 -04:00

33 lines
745 B
Go

package underscore_test
import (
"context"
"errors"
"testing"
"github.com/stretchr/testify/assert"
u "github.com/rjNemo/underscore"
)
func TestParallelMap_OrderAndResult(t *testing.T) {
values := []int{1, 2, 3, 4, 5}
out, err := u.ParallelMap(context.Background(), values, 2, func(_ context.Context, n int) (int, error) {
return n * n, nil
})
assert.NoError(t, err)
assert.Equal(t, []int{1, 4, 9, 16, 25}, out)
}
func TestParallelMap_Error(t *testing.T) {
values := []int{1, 2, 3, 4, 5}
wantErr := errors.New("boom")
out, err := u.ParallelMap(context.Background(), values, 4, func(_ context.Context, n int) (int, error) {
if n == 3 {
return 0, wantErr
}
return n, nil
})
assert.Error(t, err)
assert.Nil(t, out)
}