forked from chehsunliu/poker
-
Notifications
You must be signed in to change notification settings - Fork 0
/
card_test.go
93 lines (77 loc) · 2.53 KB
/
card_test.go
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
package cards
import (
"encoding"
"encoding/json"
"testing"
"github.com/stretchr/testify/assert"
)
func TestRankIntegers(t *testing.T) {
for i := 0; i < 13; i++ {
assert.Equal(t, int32(i), intRanks[i])
}
}
func TestNewCard(t *testing.T) {
assert.Equal(t, Card(268446761), NewCard("Ah"))
assert.Equal(t, Card(134224677), NewCard("Ks"))
}
func TestNewCards(t *testing.T) {
assert.Equal(t, []Card{Card(268446761), Card(134224677)}, NewCards("AhKs"))
}
func TestMarshalText(t *testing.T) {
var c encoding.TextMarshaler
c = NewCard("Ah")
b, err := c.MarshalText()
assert.NoError(t, err)
assert.Equal(t, "Ah", string(b))
}
func TestUnmarshalText(t *testing.T) {
var c Card
assert.Nil(t, c.UnmarshalText([]byte("Ah")))
assert.Equal(t, NewCard("Ah"), c)
assert.Error(t, c.UnmarshalText([]byte("Ahh")), "must have len 2")
}
func TestMarshalJSON(t *testing.T) {
cards := []Card{
NewCard("Ah"),
NewCard("Kh"),
NewCard("Qh"),
NewCard("Jh"),
NewCard("Th"),
}
b, err := json.Marshal(cards)
assert.NoError(t, err)
assert.Equal(t, `["Ah","Kh","Qh","Jh","Th"]`, string(b))
}
func TestUnmarshalJSON(t *testing.T) {
var cards []Card
data := `["Ah","Kh","Qh","Jh","Th"]`
err := json.Unmarshal([]byte(data), &cards)
assert.NoError(t, err)
assert.Contains(t, cards, NewCard("Ah"))
assert.Contains(t, cards, NewCard("Kh"))
assert.Contains(t, cards, NewCard("Qh"))
assert.Contains(t, cards, NewCard("Jh"))
assert.Contains(t, cards, NewCard("Th"))
}
func TestString(t *testing.T) {
assert.Equal(t, "3s", NewCard("3s").String())
}
func TestBitRank(t *testing.T) {
assert.Equal(t, int32(2048), NewCard("Ks").BitRank())
}
func TestPrimeProductFromHand(t *testing.T) {
assert.Equal(t, int32(20387), primeProductFromHand([]Card{8398611, 134236965, 33564957}))
assert.Equal(t, int32(61161), primeProductFromHand([]Card{8398611, 134236965, 33564957, 135427}))
assert.Equal(t, int32(183483), primeProductFromHand([]Card{8398611, 134236965, 33564957, 135427, 139523}))
}
func TestPrimeProductFromRankBits(t *testing.T) {
assert.Equal(t, int32(2), primeProductFromRankBits(1))
assert.Equal(t, int32(42), primeProductFromRankBits(11))
assert.Equal(t, int32(110), primeProductFromRankBits(21))
assert.Equal(t, int32(2310), primeProductFromRankBits(31))
assert.Equal(t, int32(4290), primeProductFromRankBits(55))
assert.Equal(t, int32(1785), primeProductFromRankBits(78))
assert.Equal(t, int32(1326), primeProductFromRankBits(99))
assert.Equal(t, int32(34034), primeProductFromRankBits(121))
assert.Equal(t, int32(30107), primeProductFromRankBits(344))
}