36 lines
742 B
Go
36 lines
742 B
Go
package arrays
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestTwoSum(t *testing.T) {
|
|
|
|
arr := []int{2, 7, 11, 15, 1, 13, 18}
|
|
elem := 26
|
|
|
|
want := []int{2, 3}
|
|
|
|
if got := twoSum(arr, elem); got == nil || got[0] != want[0] && got[1] != want[1] {
|
|
if got == nil {
|
|
t.Errorf("got no indexes")
|
|
}
|
|
t.Errorf("want indexes [%d,%d] and got indexes [%d,%d]", want[0], want[1], got[0], got[1])
|
|
}
|
|
}
|
|
|
|
func TestTwoSumEfficient(t *testing.T) {
|
|
|
|
arr := []int{2, 7, 11, 15, 1, 13, 18}
|
|
elem := 26
|
|
|
|
want := []int{2, 3}
|
|
|
|
if got := twoSumEfficient(arr, elem); got == nil || got[0] != want[0] && got[1] != want[1] {
|
|
if got == nil {
|
|
t.Errorf("got no indexes")
|
|
}
|
|
t.Errorf("want indexes [%d,%d] and got indexes [%d,%d]", want[0], want[1], got[0], got[1])
|
|
}
|
|
}
|