LeetCode-Go/two_sum/two_sum_test.go

package two_sum

import (
	"testing"
)

func TestTwoSum(t *testing.T) {
	tests := [][]int{
		[]int{3, 2, 4},
		[]int{0, 8, 7, 3, 3, 4, 2},
		[]int{0, 1},
	}
	targets := []int{
		6,
		11,
		1,
	}
	results := [][]int{
		[]int{1, 2},
		[]int{1, 3},
		[]int{0, 1},
	}
	caseNum := 3
	for i := 0; i < caseNum; i++ {
		if ret := twoSum(tests[i], targets[i]); ret[0] != results[i][0] && ret[1] != results[i][1] {
			t.Fatalf("case %d fails: %v\n", i, ret)
		}
	}
}