counting-sort_test.go

package algorithms

import (
	"reflect"
	"testing"
)

func TestCountingSort(t *testing.T) {
	type args struct {
		arr []int
		max int
	}
	tests := []struct {
		name string
		args args
		want []int
	}{
		{
			name: "10 unordered numbers",
			args: args{max: 993, arr: []int{93, 992, 34, 24, 92, 5, 34, 11, 20, 0}},
			want: []int{0, 5, 11, 20, 24, 34, 34, 92, 93, 992},
		},
		{
			name: "15 unordered numbers",
			args: args{
				max: 9523,
				arr: []int{
					3024, 6446, 2647, 5502, 9523,
					4432, 4140, 6344, 1766, 3384,
					3827, 1535, 8130, 2287, 5433,
				},
			},
			want: []int{
				1535, 1766, 2287, 2647, 3024,
				3384, 3827, 4140, 4432, 5433,
				5502, 6344, 6446, 8130, 9523,
			},
		},
		{
			name: "20 unordered numbers",
			args: args{
				max: 9994,
				arr: []int{
					1831, 2776, 9994, 9924, 7054,
					3804, 6032, 7104, 7734, 9671,
					3942, 9842, 3757, 2360, 8972,
					1979, 6145, 2799, 2164, 4108,
				},
			},
			want: []int{
				1831, 1979, 2164, 2360, 2776,
				2799, 3757, 3804, 3942, 4108,
				6032, 6145, 7054, 7104, 7734,
				8972, 9671, 9842, 9924, 9994,
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := CountingSort(tt.args.arr, tt.args.max); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("CountingSort() = %v, want %v", got, tt.want)
			}
		})
	}
}