New similarity.CompareTracks function

internal/util/util.go

@@ -32,3 +32,19 @@ func Min[T constraints.Ordered](m, n T) T {
 		return m
 	}
 }
+
+func Sum[T constraints.Integer | constraints.Float](v ...T) T {
+	var sum T
+	for _, i := range v {
+		sum += i
+	}
+	return sum
+}
+
+func Average[T constraints.Integer | constraints.Float](v ...T) float64 {
+	length := len(v)
+	if length == 0 {
+		return 0.0
+	}
+	return float64(Sum(v...)) / float64(length)
+}

internal/util/util_test.go

@@ -17,7 +17,9 @@ package util_test
 
 import (
 	"fmt"
+	"testing"
 
+	"github.com/stretchr/testify/assert"
 	"go.uploadedlobster.com/scotty/internal/util"
 )
 
@@ -32,3 +34,31 @@ func ExampleMin() {
 	fmt.Print(v)
 	// Output: 2
 }
+
+func ExampleSum() {
+	values := []float64{1.4, 2.2}
+	sum := util.Sum(values...)
+	fmt.Print(sum)
+	// Output: 3.6
+}
+
+func TestSumEmpty(t *testing.T) {
+	assert.Equal(t, 0, util.Sum([]int{}...))
+}
+
+func ExampleAverage() {
+	values := []float64{1.4, 2.2, 0.9}
+	sum := util.Average(values...)
+	fmt.Print(sum)
+	// Output: 1.5
+}
+
+func TestAverageEmpty(t *testing.T) {
+	assert.Equal(t, 0.0, util.Average([]int{}...))
+}
+
+func TestAverageInt(t *testing.T) {
+	assert := assert.New(t)
+	assert.Equal(3.0, util.Average([]int{2, 4, 3}...))
+	assert.Equal(1.5, util.Average([]int{2, 1, 1, 2}...))
+}