LeetCode-Go/leetcode/0001.Two-Sum/1. Two Sum_test.go

package leetcode

import (
	"fmt"
	"testing"
)

func TestTwoSum(t *testing.T) {
	tests := [][]int{
		[]int{3, 2, 4},
		[]int{3, 2, 4},
		[]int{0, 8, 7, 3, 3, 4, 2},
		[]int{0, 1},
	}
	targets := []int{
		6,
		5,
		11,
		1,
	}
	results := [][]int{
		[]int{1, 2},
		[]int{0, 1},
		[]int{1, 3},
		[]int{0, 1},
	}
	fmt.Printf("------------------------Leetcode Problem 1------------------------\n")
	for i := 0; i < len(targets); i++ {
		fmt.Printf("nums = %v target = %v result = %v\n", tests[i], targets[i], twoSum(tests[i], targets[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)
		}
	}
}