Add: test case

Author: openset

problems/accounts-merge/accounts_merge_test.go

@@ -2,6 +2,7 @@ package accounts_merge
 
 import (
 	"reflect"
+	"sort"
 	"testing"
 )
 
@@ -21,12 +22,58 @@ func TestAccountsMerge(t *testing.T) {
 			},
 			expected: [][]string{
 				{"John", "[email protected]", "[email protected]", "[email protected]"},
-				{"John", "[email protected]"}, {"Mary", "[email protected]"},
+				{"John", "[email protected]"},
+				{"Mary", "[email protected]"},
+			},
+		},
+		{
+			input: [][]string{
+				{"Alex", "[email protected]", "[email protected]", "[email protected]"},
+				{"Ethan", "[email protected]", "[email protected]", "[email protected]"},
+				{"Kevin", "[email protected]", "[email protected]", "[email protected]"},
+				{"Gabe", "[email protected]", "[email protected]", "[email protected]"},
+				{"Gabe", "[email protected]", "[email protected]", "[email protected]"},
+			},
+			expected: [][]string{
+				{"Alex", "[email protected]", "[email protected]", "[email protected]"},
+				{"Ethan", "[email protected]", "[email protected]"},
+				{"Gabe", "[email protected]", "[email protected]", "[email protected]", "[email protected]"},
+				{"Kevin", "[email protected]", "[email protected]"},
+			},
+		},
+		{
+			input: [][]string{
+				{"David", "[email protected]", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]", "[email protected]"},
+			},
+			expected: [][]string{
+				{"David", "[email protected]", "[email protected]", "[email protected]", "[email protected]", "[email protected]"},
+			},
+		},
+		{
+			input: [][]string{
+				{"David", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]"},
+				{"David", "[email protected]", "[email protected]"},
+			},
+			expected: [][]string{
+				{"David", "[email protected]", "[email protected]", "[email protected]", "[email protected]", "[email protected]", "[email protected]"},
 			},
 		},
 	}
 	for _, tc := range tests {
 		output := accountsMerge(tc.input)
+		sort.Slice(output, func(i, j int) bool {
+			if output[i][0] == output[i][0] {
+				return output[i][1] < output[j][1]
+			}
+			return output[i][0] < output[i][0]
+		})
 		if !reflect.DeepEqual(output, tc.expected) {
 			t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)
 		}