mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-25 20:21:47 +08:00
53 lines
834 B
Go
53 lines
834 B
Go
package leetcode
|
|
|
|
import "testing"
|
|
|
|
func Test_pivotIndex(t *testing.T) {
|
|
type args struct {
|
|
nums []int
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want int
|
|
}{
|
|
{
|
|
name: "test_case1",
|
|
args: args{[]int{1, 7, 3, 6, 5, 6}},
|
|
want: 3,
|
|
},
|
|
{
|
|
name: "test_case2",
|
|
args: args{[]int{1,2,3}},
|
|
want: -1,
|
|
},
|
|
{
|
|
name: "test_case3",
|
|
args: args{[]int{-1,-1,-1,-1,-1,-1}},
|
|
want: -1,
|
|
},
|
|
{
|
|
name: "test_case4",
|
|
args: args{[]int{-1,-1,-1,-1,-1,0}},
|
|
want: 2,
|
|
},
|
|
{
|
|
name: "test_case5",
|
|
args: args{[]int{1}},
|
|
want: 0,
|
|
},
|
|
{
|
|
name: "test_case5",
|
|
args: args{[]int{}},
|
|
want: -1,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := pivotIndex(tt.args.nums); got != tt.want {
|
|
t.Errorf("pivotIndex() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|