initial commit

pkg/array/aggregate_test.go

@@ -0,0 +1,30 @@
+package array
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestSum_WithNumberArray_ShouldBeAsExpected(t *testing.T) {
+	// Arrange
+	arr := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
+	// Act
+	r := Sum(arr, func(val int) int {
+		return val
+	})
+	// Assert
+	const expected = 55
+	assert.True(t, r == expected)
+}
+
+func TestSum_WithStructArray_ShouldBeAsExpected(t *testing.T) {
+	// Arrange
+	arr := []struct{ d float64 }{{d: 0.1}, {d: 1.5}, {d: 0.4}, {d: 2.5}, {d: 5.521}}
+	// Act
+	r := Sum(arr, func(val struct{ d float64 }) float64 {
+		return val.d
+	})
+	// Assert
+	const expected = 10.021
+	assert.True(t, r == expected)
+}