base-golang/tool/help/zip_test.go

package help

import "testing"

func TestZip(t *testing.T) {
	input1 := []int{1, 2, 3}
	input2 := []int{2, 4, 5, 1}
	want := []interface{}{3, 6, 8}

	if q := From(input1).Zip(From(input2), func(i, j interface{}) interface{} {
		return i.(int) + j.(int)
	}); !validateQuery(q, want) {
		t.Errorf("From(%v).Zip(%v)=%v expected %v", input1, input2, toSlice(q), want)
	}
}

func TestZipT_PanicWhenResultSelectorFnIsInvalid(t *testing.T) {
	mustPanicWithError(t, "ZipT: parameter [resultSelectorFn] has a invalid function signature. Expected: 'func(T,T)T', actual: 'func(int,int,int)int'", func() {
		input1 := []int{1, 2, 3}
		input2 := []int{2, 4, 5, 1}

		From(input1).ZipT(From(input2), func(i, j, k int) int {
			return i + j
		})
	})
}