jecna-api/stringRank_test.go at master · dsorm/jecna-api · 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
package jecnaapi

import (
	"testing"
)

func TestStringRankDo(t *testing.T) {
	rank := stringRank{
		strings: []string{"Lorem ipsum dolor sir amet", "Suplování 2018-19", "Suplování 2017-18"},
		rules: []stringRankRule{
			{stringHas: "Suplování", scoreIfTrue: 5, scoreIfFalse: -5},
			{stringHas: "2018", scoreIfTrue: 20},
			{stringHas: "18", scoreIfTrue: 100},
			{stringHas: "17", scoreIfTrue: -100},
			{stringHas: "2017", scoreIfTrue: -100},
			{stringHas: "2019", scoreIfTrue: 20},
		},
		continuous: true,
	}

	result := rank.do()

	for _, v := range result {
		var expectedScore int64
		switch v.string {
		case "Lorem ipsum dolor sir amet":
			expectedScore = -5
		case "Suplování 2018-19":
			expectedScore = 25
		case "Suplování 2017-18":
			expectedScore = 5
		}
		if v.score != expectedScore {
			t.Errorf("Score u stringu '%v' se neshoduje: ocekavano %+v, ma %+v", v.string, expectedScore, v.score)
		}
	}

	rank.continuous = false
	result = rank.do()

	for _, v := range result {
		var expectedScore int64
		switch v.string {
		case "Lorem ipsum dolor sir amet":
			expectedScore = -5
		case "Suplování 2018-19":
			expectedScore = 125
		case "Suplování 2017-18":
			expectedScore = -95
		}
		if v.score != expectedScore {
			t.Errorf("Score u stringu '%v' se neshoduje: ocekavano %+v, ma %+v", v.string, expectedScore, v.score)
		}
	}
}