46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFindBooks(t *testing.T) {
|
|
|
|
hand := []Card{15, 16, 17} // [ 3♠ 3♥ 3♦]
|
|
|
|
t.Logf("Finding books in hand: %v", hand)
|
|
|
|
p := MakeBooks(hand, 10)
|
|
assert.EqualValues(t, [][]int{[]int{0, 1, 2}}, p)
|
|
}
|
|
|
|
func TestFindRuns(t *testing.T) {
|
|
hand := []Card{15, 20, 25, 25, 30} // [ 3♠ 4♠ 5♠ 5♠ 6♠]
|
|
|
|
t.Logf("Finding runs in hand: %v", hand)
|
|
|
|
p := MakeRuns(hand, 10)
|
|
assert.EqualValues(t, [][]int{[]int{0, 1, 2}, []int{0, 1, 2, 4}, []int{0, 1, 3}, []int{0, 1, 3, 4}, []int{1, 2, 4}, []int{1, 3, 4}}, p)
|
|
}
|
|
|
|
func TestFindMultiGroups(t *testing.T) {
|
|
hand := []Card{15, 20, 25, 30, 16, 17}
|
|
|
|
t.Logf("Finding all groupups in hand: %v", hand)
|
|
|
|
p := MakeMultiGroups(hand, 10)
|
|
assert.EqualValues(t, [][][]int{[][]int{[]int{0, 4, 5}}, [][]int{[]int{0, 4, 5}, []int{1, 2, 3}}, [][]int{[]int{0, 1, 2}}, [][]int{[]int{0, 1, 2, 3}}, [][]int{[]int{1, 2, 3}}, [][]int{[]int{1, 2, 3}, []int{0, 4, 5}}}, p)
|
|
}
|
|
|
|
func TestSolveHand(t *testing.T) {
|
|
hand := []Card{15, 20, 25, 30, 16, 17}
|
|
|
|
t.Logf("Finding best groupups in hand: %v", hand)
|
|
|
|
grouping, score := FindBestGrouping(hand, 10)
|
|
assert.EqualValues(t, [][]int{[]int{0, 4, 5}, []int{1, 2, 3}}, grouping)
|
|
assert.EqualValues(t, 0, score)
|
|
}
|