blob: 8ab173c86a7009d51f22eef5bf2b6b45c5eefae6 [file] [log] [blame]
// Copyright 2023 The Dawn & Tint Authors
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this
// list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
package transform_test
import (
"fmt"
"reflect"
"testing"
"dawn.googlesource.com/dawn/tools/src/transform"
)
func check(got, expect any) error {
if !reflect.DeepEqual(got, expect) {
return fmt.Errorf(`
got: %v
expect: %v
`, got, expect)
}
return nil
}
func TestFilter(t *testing.T) {
in := []int{5, 8, 2, 4}
out := transform.Filter(in, func(i int) bool {
return i != 8
})
if e := check(out, []int{5, 2, 4}); e != nil {
t.Error(e)
}
}
func TestFlatten(t *testing.T) {
in := [][]int{{5, 8}, {2}, {4, 6, 5}}
out := transform.Flatten(in)
if e := check(out, []int{5, 8, 2, 4, 6, 5}); e != nil {
t.Error(e)
}
}
func TestSlice(t *testing.T) {
in := []int{5, 8, 2, 4}
out, err := transform.Slice(in, func(i int) (int, error) {
return i + 1, nil
})
if e := check(out, []int{6, 9, 3, 5}); e != nil {
t.Error(e)
}
if e := check(err, nil); e != nil {
t.Error(e)
}
}
func TestSliceErr(t *testing.T) {
in := []int{5, 8, 2, 4}
out, err := transform.Slice(in, func(i int) (int, error) {
return 0, fmt.Errorf("E(%v)", i)
})
if e := check(out, ([]int)(nil)); e != nil {
t.Error(e)
}
if e := check(err.Error(), "E(5)"); e != nil {
t.Error(e)
}
}
func TestGoSlice(t *testing.T) {
in := []int{5, 8, 2, 4}
out, err := transform.GoSlice(in, func(i int) (int, error) {
return i + 1, nil
})
if e := check(out, []int{6, 9, 3, 5}); e != nil {
t.Error(e)
}
if e := check(err, nil); e != nil {
t.Error(e)
}
}
func TestGoSliceErr(t *testing.T) {
in := []int{5, 8, 2, 4}
out, err := transform.GoSlice(in, func(i int) (int, error) {
return 0, fmt.Errorf("E(%v)", i)
})
if e := check(out, ([]int)(nil)); e != nil {
t.Error(e)
}
if e := check(err.Error(), `E(2)
E(4)
E(5)
E(8)`); e != nil {
t.Error(e)
}
}
func TestSliceToChan(t *testing.T) {
in := []int{5, 8, 2, 4}
c := transform.SliceToChan(in)
out := []int{}
for i := range c {
out = append(out, i)
}
if e := check(in, out); e != nil {
t.Error(e)
}
}