1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
package slice
import "testing"
func Test_Take(t *testing.T) {
for i, tc := range []struct {
input []int
take int
expected []int
}{
{
input: []int{1, 2, 3, 4, 5},
take: 3,
expected: []int{1, 2, 3},
},
{
input: []int{1, 2, 3},
take: 5,
expected: []int{1, 2, 3},
},
{
input: []int{},
take: 2,
expected: []int{},
},
{
input: []int{1, 2, 3},
take: 0,
expected: []int{},
},
{
input: nil,
take: 2,
expected: []int{},
},
} {
actual := Take(tc.input, tc.take)
if len(actual) != len(tc.expected) {
t.Errorf("Test %d: Expected %v, got %v", i, tc.expected, actual)
}
}
}
|
