Function TestFilter

transform_test.go:93–162

(t *testing.T)

}

func TestFilter(t *testing.T) {
	th.TestBothOrderings(t, func(t *testing.T, ord bool) {
		for _, n := range []int{1, 5} {

			t.Run(th.Name("nil", n), func(t *testing.T) {
				out := universalFilter(ord, nil, n, func(x int) (bool, error) { return true, nil })
				th.ExpectValue(t, out, nil)
			})

			t.Run(th.Name("correctness", n), func(t *testing.T) {
				in := FromChan(th.FromRange(0, 20), nil)
				in = replaceWithError(in, 15, fmt.Errorf("err15"))

				out := universalFilter(ord, in, n, func(x int) (bool, error) {
					if x == 5 {
						return false, fmt.Errorf("err05")
					}
					if x == 6 {
						return true, fmt.Errorf("err06")
					}

					return x%2 == 0, nil
				})

				outSlice, errSlice := toSliceAndErrors(out)

				expectedSlice := make([]int, 0, 20)
				for i := 0; i < 20; i++ {
					if i%2 == 1 || i == 5 || i == 6 || i == 15 {
						continue
					}
					expectedSlice = append(expectedSlice, i)
				}

				th.Sort(outSlice)
				th.Sort(errSlice)

				th.ExpectSlice(t, outSlice, expectedSlice)
				th.ExpectSlice(t, errSlice, []string{"err05", "err06", "err15"})
			})

			t.Run(th.Name("ordering", n), func(t *testing.T) {
				in := FromChan(th.FromRange(0, 20000), nil)

				out := universalFilter(ord, in, n, func(x int) (bool, error) {
					switch x % 3 {
					case 2:
						return false, fmt.Errorf("err%06d", x)
					case 1:
						return false, nil
					default:
						return true, nil

					}
				})

				outSlice, errSlice := toSliceAndErrors(out)