From bfac048fb459cbcd67ee64764015149bf317f971 Mon Sep 17 00:00:00 2001 From: Andy Long Date: Tue, 13 Jun 2023 13:21:56 +0100 Subject: [PATCH] Updating Count function (#36) * Adding some new funky functions which I find useful Created a Tuple struct as some of the new functions require you to return a new slice with two fields which is the result of the new functions Created the Join, JoinProjection, Range, SumMap, Zip functions, ecah fuction is documented with how it works and had a unit test or maybe more * Added in an OrderBy function * Documentation comment for OrderBy which I missed out * Adding a Unit test for JoinProject function Updated the comments on the Join & OrderBy functions so they make a little more sense. Covered an extra test case with the Join test, where the left set has more data than the right and so the Right handside array of the join is empty * Adding a count method to the package, so you can find out how many items in a slice satisfy and given condition * Updating count to work with any so you can count structs as well as basic types * Removing extra underscores --- count.go | 2 +- count_test.go | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/count.go b/count.go index 581c053..9a3d287 100644 --- a/count.go +++ b/count.go @@ -2,7 +2,7 @@ package underscore // Count returns the number of elements in the slice that satisfy the predicate. // example: Count([]int{1,2,3,4,5}, func(n int) bool { return n%2 == 0 }) // 2 -func Count[T comparable](slice []T, predicate func(T) bool) int { +func Count[T any](slice []T, predicate func(T) bool) int { count := 0 for _, item := range slice { if predicate(item) { diff --git a/count_test.go b/count_test.go index 9a876ef..cef8e32 100644 --- a/count_test.go +++ b/count_test.go @@ -1,9 +1,10 @@ package underscore import ( - "github.com/stretchr/testify/assert" "strings" "testing" + + "github.com/stretchr/testify/assert" ) func Test_Count_Can_Count_Numbers(t *testing.T) { @@ -21,7 +22,7 @@ type People struct { Gender string } -func Test_Count_Can_Count__People(t *testing.T) { +func Test_Count_Can_Count_People(t *testing.T) { people := []People{ {Name: "Andy", Age: 43, Gender: "M"}, {Name: "Fred", Age: 33, Gender: "M"},