71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
|
package lambda_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"git.tuxpa.in/a/lambda"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFlattenIntSix(t *testing.T) {
|
||
|
|
var arr [][]int
|
||
|
|
arr = [][]int{{1}, {2}, {3}, {4}, {6}, {5}}
|
||
|
|
ans := lambda.Flatten(arr)
|
||
|
|
exp := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSplitIntSix0(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 0)
|
||
|
|
if ans != nil {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, nil)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSplitIntSix1(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 1)
|
||
|
|
exp := [][]int{{1, 2, 3, 4, 6, 5}}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSplitIntSix2(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 2)
|
||
|
|
exp := [][]int{{1, 2, 3}, {4, 6, 5}}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func TestSplitIntSix6(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 6)
|
||
|
|
exp := [][]int{{1}, {2}, {3}, {4}, {6}, {5}}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSplitIntSix5(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 5)
|
||
|
|
exp := [][]int{{1}, {2}, {3}, {4}, {6, 5}}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSplitIntSix10(t *testing.T) {
|
||
|
|
arr := []int{1, 2, 3, 4, 6, 5}
|
||
|
|
ans := lambda.Split(arr, 10)
|
||
|
|
exp := [][]int{{1, 2, 3, 4, 6, 5}}
|
||
|
|
if !reflect.DeepEqual(ans, exp) {
|
||
|
|
t.Fatalf("expect %v to be %v", ans, exp)
|
||
|
|
}
|
||
|
|
}
|