trueskill/sort_test.go at master · fasmat/trueskill · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package trueskill_test

import (
	"testing"

	"github.com/fasmat/trueskill"
)

func TestSortPlayersByRank(t *testing.T) {
	t.Parallel()

	players := generatePlayers(5)

	dataIn := []trueskill.Player{
		players[0],
		players[4],
		players[3],
		players[1],
		players[2],
	}

	rankIn := []int{0, 4, 3, 1, 2}

	data, rank := trueskill.SortPlayersByRank(dataIn, rankIn)

	if len(data) != len(players) {
		t.Error("Lost data values during sorting. Expected",
			len(players), "got", len(data))
	}

	if len(rank) != len(rankIn) {
		t.Error("Lost rank values during sorting. Expected",
			len(rankIn), "got", len(rank))
	}

	for i, v := range data {
		if v.GetID() != i {
			t.Error("Element", i, "of players is not the same as expected")
		}

		if rank[i] != i {
			t.Error("Element", i, "of rank is not the same as expected")
		}
	}
}

func TestSortTeamsByRank(t *testing.T) {
	t.Parallel()

	players := generatePlayers(5)
	dataIn := []trueskill.Team{
		trueskill.NewTeam([]trueskill.Player{players[4]}),
		trueskill.NewTeam([]trueskill.Player{players[3]}),
		trueskill.NewTeam([]trueskill.Player{players[2]}),
		trueskill.NewTeam([]trueskill.Player{players[1]}),
		trueskill.NewTeam([]trueskill.Player{players[0]}),
	}
	rankIn := []int{5, 4, 3, 1, 1}

	dataOut := []trueskill.Team{
		trueskill.NewTeam([]trueskill.Player{players[1]}),
		trueskill.NewTeam([]trueskill.Player{players[0]}),
		trueskill.NewTeam([]trueskill.Player{players[2]}),
		trueskill.NewTeam([]trueskill.Player{players[3]}),
		trueskill.NewTeam([]trueskill.Player{players[4]}),
	}
	rankOut := []int{1, 1, 3, 4, 5}

	data, rank := trueskill.SortTeamsByRank(dataIn, rankIn)

	if len(data) != len(dataOut) {
		t.Error("Lost data values during sorting. Expected",
			len(dataOut), "got", len(data))
	}

	if len(rank) != len(rankOut) {
		t.Error("Lost rank values during sorting. Expected",
			len(rankOut), "got", len(rank))
	}

	for i, v := range data {
		if v.String() != dataOut[i].String() {
			t.Error("Element", i, "of teams is not the same as expected")
		}
	}

	for i, v := range rank {
		if v != rankOut[i] {
			t.Error("Element", i, "of rank is not the same as expected")
		}
	}
}

func generatePlayers(n int) []trueskill.Player {
	p := make([]trueskill.Player, n)

	for i := range p {
		p[i] = trueskill.NewDefaultPlayer(i)
	}
	return p
}