75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
package pokercore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.eeqj.de/sneak/timingbench"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestShuffleSpeed(t *testing.T) {
|
|
iterations := 1000
|
|
t.Logf("Running %d iterations of shuffle speed test", iterations)
|
|
// Create a context with a timeout for cancellation.
|
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
|
defer cancel()
|
|
|
|
// Measure the execution times of the sample function.
|
|
d := NewDeck()
|
|
result, err := timingbench.TimeFunction(ctx, func() { d.ShuffleRandomly() }, iterations)
|
|
if err != nil {
|
|
t.Fatalf("Error measuring function: %v", err)
|
|
}
|
|
// Print the timing results.
|
|
t.Logf(result.String())
|
|
}
|
|
|
|
func TestHandFindingSpeedFiveCard(t *testing.T) {
|
|
iterations := 1000
|
|
t.Logf("Running %d iterations of hand finding speed test for 5 card hand", iterations)
|
|
measureHandFinding(t, iterations, 5)
|
|
}
|
|
|
|
func TestHandFindingSpeedSevenCard(t *testing.T) {
|
|
iterations := 1000
|
|
t.Logf("Running %d iterations of hand finding speed test for 7 card hand", iterations)
|
|
measureHandFinding(t, iterations, 7)
|
|
}
|
|
|
|
func TestHandFindingSpeedNineCard(t *testing.T) {
|
|
iterations := 100
|
|
t.Logf("Running %d iterations of hand finding speed test for 9 card hand", iterations)
|
|
measureHandFinding(t, iterations, 9)
|
|
}
|
|
|
|
func measureHandFinding(t *testing.T, iterations int, cardCount int) {
|
|
// Create a context with a timeout for cancellation.
|
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
|
defer cancel()
|
|
|
|
// Measure the execution times of the sample function.
|
|
result, err := timingbench.TimeFunction(ctx, func() {
|
|
findHandInRandomCards(t, int(123456789), cardCount) // check for hand in 10 cards
|
|
}, iterations)
|
|
if err != nil {
|
|
t.Fatalf("Error measuring function: %v", err)
|
|
}
|
|
// Print the timing results.
|
|
t.Logf(result.String())
|
|
}
|
|
|
|
func findHandInRandomCards(t *testing.T, shuffleSeed int, cardCount int) {
|
|
d := NewDeck()
|
|
d.ShuffleDeterministically(int64(shuffleSeed))
|
|
cards := d.Deal(cardCount)
|
|
hand, err := cards.IdentifyBestFiveCardPokerHand()
|
|
assert.Nil(t, err, "Expected no error")
|
|
ph, err := hand.PokerHand()
|
|
assert.Nil(t, err, "Expected no error")
|
|
//assert.Greater(t, ph.Score, 0, "Expected score to be nonzero 0")
|
|
desc := ph.Description()
|
|
assert.NotEmpty(t, desc, "Expected description to not be empty")
|
|
}
|