partition_test.go

package underscore_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	u "github.com/rjNemo/underscore"
)

func TestPartition(t *testing.T) {
	nums := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
	isEven := func(n int) bool { return n%2 == 0 }

	wantEvens := []int{0, 2, 4, 6, 8}
	wantOdds := []int{1, 3, 5, 7, 9}

	evens, odds := u.Partition(nums, isEven)

	assert.Equal(t, wantEvens, evens)
	assert.Equal(t, wantOdds, odds)
}

func TestPartitionEmpty(t *testing.T) {
	keep, reject := u.Partition([]int{}, func(n int) bool { return n > 0 })
	assert.Empty(t, keep)
	assert.Empty(t, reject)
}

func TestPartitionSingleElement(t *testing.T) {
	keep, reject := u.Partition([]int{5}, func(n int) bool { return n > 3 })
	assert.Equal(t, []int{5}, keep)
	assert.Empty(t, reject)
}

func TestPartitionAllPass(t *testing.T) {
	nums := []int{2, 4, 6, 8}
	keep, reject := u.Partition(nums, func(n int) bool { return n%2 == 0 })
	assert.Equal(t, nums, keep)
	assert.Empty(t, reject)
}

func TestPartitionAllReject(t *testing.T) {
	nums := []int{1, 3, 5, 7}
	keep, reject := u.Partition(nums, func(n int) bool { return n%2 == 0 })
	assert.Empty(t, keep)
	assert.Equal(t, nums, reject)
}

func BenchmarkPartition(b *testing.B) {
	data := make([]int, 1000)
	for i := range data {
		data[i] = i
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		u.Partition(data, func(n int) bool { return n%2 == 0 })
	}
}