Add arrays package

This commit is contained in:
Fabian Becker 2022-01-05 17:56:38 +01:00
parent 3569bcd485
commit 01e7a6fa26
2 changed files with 48 additions and 1 deletions

View File

@ -7,3 +7,17 @@ func Sum(numbers []int) int {
}
return sum
}
func SumAllTails(numbersToSum ...[]int) []int {
sums := make([]int, 0, len(numbersToSum))
for _, numbers := range numbersToSum {
if len(numbers) == 0 {
sums = append(sums, 0)
} else {
tail := numbers[1:]
sums = append(sums, Sum(tail))
}
}
return sums
}

View File

@ -1,6 +1,9 @@
package main
import "testing"
import (
"reflect"
"testing"
)
func TestSum(t *testing.T) {
@ -27,3 +30,33 @@ func TestSum(t *testing.T) {
})
}
//func TestSumAll(t *testing.T) {
//got := SumAll([]int{1, 2}, []int{0, 9})
//want := []int{3, 9}
//if !reflect.DeepEqual(got, want) {
//t.Errorf("got %v want %v", got, want)
//}
//}
func TestSumAllTails(t *testing.T) {
checkSums := func(t testing.TB, got, want []int) {
t.Helper()
if !reflect.DeepEqual(got, want) {
t.Errorf("got %v want %v", got, want)
}
}
t.Run("make the sums of tails of", func(t *testing.T) {
got := SumAllTails([]int{1, 2}, []int{0, 9})
want := []int{2, 9}
checkSums(t, got, want)
})
t.Run("safely sum empty slices", func(t *testing.T) {
got := SumAllTails([]int{}, []int{3, 4, 5})
want := []int{0, 9}
checkSums(t, got, want)
})
}