232 lines
9.2 KiB
Go
232 lines
9.2 KiB
Go
package pokercore
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAceLowStraight(t *testing.T) {
|
|
hand := Cards{
|
|
AceOfSpades(),
|
|
DeuceOfHearts(),
|
|
ThreeOfDiamonds(),
|
|
FourOfClubs(),
|
|
FiveOfSpades(),
|
|
}
|
|
assert.True(t, hand.containsStraight(), "Expected hand to be a straight")
|
|
ph, err := hand.PokerHand()
|
|
assert.Nil(t, err, "Expected no error")
|
|
assert.Greater(t, ph.Score, 0, "Expected score to be nonzero 0")
|
|
assert.Less(t, ph.Score, 100000000000000000, "Expected score to be less than 100000000000000000")
|
|
assert.Equal(t, ph.Score, ScoreStraight+100*FIVE.Score())
|
|
assert.Equal(t, ph.Description(), "a five high straight")
|
|
assert.True(t, hand.HighestRank() == ACE, "Expected highest rank to be an ace")
|
|
assert.True(t, hand.SortByRankAscending().First().Rank == DEUCE, "Expected first card to be a deuce")
|
|
assert.True(t, hand.SortByRankAscending().Last().Rank == ACE, "Expected last card in sorted to be a ace")
|
|
assert.True(t, hand.SortByRankAscending().Second().Rank == THREE, "Expected second card to be a three")
|
|
assert.True(t, hand.SortByRankAscending().Third().Rank == FOUR, "Expected third card to be a four")
|
|
assert.True(t, hand.SortByRankAscending().Fourth().Rank == FIVE, "Expected fourth card to be a five")
|
|
assert.True(t, hand.SortByRankAscending().Fifth().Rank == ACE, "Expected fifth card to be an ace")
|
|
}
|
|
|
|
func TestAceHighStraight(t *testing.T) {
|
|
hand := Cards{
|
|
TenOfSpades(),
|
|
JackOfHearts(),
|
|
KingOfClubs(),
|
|
AceOfSpades(),
|
|
QueenOfDiamonds(),
|
|
}
|
|
|
|
assert.True(t, hand.containsStraight(), "Expected hand to be a straight")
|
|
newDeck := NewDeckFromCards(hand)
|
|
newDeck.ShuffleDeterministically(123456789)
|
|
shuffledHand := newDeck.Deal(5)
|
|
assert.True(t, shuffledHand.containsStraight(), "Expected hand to still be a straight after shuffle")
|
|
assert.True(t, shuffledHand.HighestRank() == ACE, "Expected highest rank to be an ace")
|
|
sortedHand := shuffledHand.SortByRankAscending()
|
|
assert.True(t, sortedHand[0].Rank == TEN, "Expected lowest rank to be a ten")
|
|
assert.True(t, sortedHand[1].Rank == JACK, "Expected second lowest rank to be a jack")
|
|
assert.True(t, sortedHand[2].Rank == QUEEN, "Expected third lowest rank to be a queen")
|
|
assert.True(t, sortedHand[3].Rank == KING, "Expected fourth lowest rank to be a king")
|
|
assert.True(t, sortedHand[4].Rank == ACE, "Expected highest rank to be an ace")
|
|
}
|
|
|
|
func TestOtherStraight(t *testing.T) {
|
|
hand := Cards{
|
|
DeuceOfSpades(),
|
|
ThreeOfHearts(),
|
|
FourOfDiamonds(),
|
|
FiveOfClubs(),
|
|
SixOfSpades(),
|
|
}
|
|
assert.True(t, hand.containsStraight(), "Expected hand to be a straight")
|
|
|
|
newDeck := NewDeckFromCards(hand)
|
|
newDeck.ShuffleDeterministically(123456789)
|
|
fmt.Printf("Shuffled deck: %s\n", newDeck.String())
|
|
fmt.Printf("new deck has %d cards\n", newDeck.Count())
|
|
shuffledHand := newDeck.Deal(5)
|
|
assert.True(t, shuffledHand.containsStraight(), "Expected hand to still be a straight after shuffle")
|
|
assert.False(t, shuffledHand.containsTwoPair(), "Expected hand to not be two pair")
|
|
|
|
}
|
|
|
|
func TestFlush(t *testing.T) {
|
|
hand := Cards{
|
|
AceOfSpades(),
|
|
DeuceOfSpades(),
|
|
ThreeOfSpades(),
|
|
FourOfSpades(),
|
|
SixOfSpades(),
|
|
}
|
|
assert.True(t, hand.containsFlush(), "Expected hand to be a flush")
|
|
newDeck := NewDeckFromCards(hand)
|
|
newDeck.ShuffleDeterministically(123456789)
|
|
fmt.Printf("Shuffled deck: %s\n", newDeck.String())
|
|
fmt.Printf("new deck has %d cards\n", newDeck.Count())
|
|
shuffledHand := newDeck.Deal(5)
|
|
assert.True(t, shuffledHand.containsFlush(), "Expected hand to still be a flush after shuffle")
|
|
|
|
x := ScoreFlush
|
|
var multiplier HandScore = 100
|
|
x += multiplier * DEUCE.Score()
|
|
multiplier *= 100
|
|
x += multiplier * THREE.Score()
|
|
multiplier *= 100
|
|
x += multiplier * FOUR.Score()
|
|
multiplier *= 100
|
|
x += multiplier * SIX.Score()
|
|
fmt.Printf("a-2-3-4-6 flush score should be: %d\n", x)
|
|
}
|
|
|
|
func TestStraightFlush(t *testing.T) {
|
|
hand := Cards{
|
|
SixOfSpades(),
|
|
DeuceOfSpades(),
|
|
ThreeOfSpades(),
|
|
FourOfSpades(),
|
|
FiveOfSpades(),
|
|
}
|
|
assert.True(t, hand.containsStraight(), "Expected hand to be a straight")
|
|
assert.True(t, hand.containsFlush(), "Expected hand to be a flush")
|
|
assert.True(t, hand.containsStraightFlush(), "Expected hand to be a straight flush")
|
|
assert.False(t, hand.containsRoyalFlush(), "Expected hand to not be a royal flush")
|
|
assert.False(t, hand.containsFourOfAKind(), "Expected hand to not be four of a kind")
|
|
assert.False(t, hand.containsFullHouse(), "Expected hand to not be a full house")
|
|
assert.False(t, hand.containsThreeOfAKind(), "Expected hand to not be three of a kind")
|
|
assert.False(t, hand.containsTwoPair(), "Expected hand to not be two pair")
|
|
assert.False(t, hand.containsPair(), "Expected hand to not be a pair")
|
|
|
|
assert.True(t, hand.HighestRank() == SIX, "Expected highest rank to be a six")
|
|
|
|
nd := NewDeckFromCards(hand)
|
|
nd.ShuffleDeterministically(123456789)
|
|
fmt.Printf("Shuffled deck: %s\n", nd.String())
|
|
fmt.Printf("new deck has %d cards\n", nd.Count())
|
|
shuffledHand := nd.Deal(5)
|
|
assert.True(t, shuffledHand.containsStraightFlush(), "Expected hand to still be a straight flush after shuffle")
|
|
assert.True(t, shuffledHand.HighestRank() == SIX, "Expected highest rank to still be a six after shuffle")
|
|
assert.True(t, shuffledHand.HighestRank() == SIX, "Expected highest rank to be a six after shuffle even with aces low")
|
|
}
|
|
|
|
func TestRoyalFlush(t *testing.T) {
|
|
hand := Cards{
|
|
TenOfSpades(),
|
|
JackOfSpades(),
|
|
QueenOfSpades(),
|
|
KingOfSpades(),
|
|
AceOfSpades(),
|
|
}
|
|
assert.True(t, hand.containsStraight(), "Expected hand to be a straight")
|
|
assert.True(t, hand.containsFlush(), "Expected hand to be a flush")
|
|
assert.True(t, hand.containsStraightFlush(), "Expected hand to be a straight flush")
|
|
assert.True(t, hand.containsRoyalFlush(), "Expected hand to be a royal flush")
|
|
|
|
assert.False(t, hand.containsFourOfAKind(), "Expected hand to not be four of a kind")
|
|
assert.False(t, hand.containsFullHouse(), "Expected hand to not be a full house")
|
|
assert.False(t, hand.containsThreeOfAKind(), "Expected hand to not be three of a kind")
|
|
assert.False(t, hand.containsTwoPair(), "Expected hand to not be two pair")
|
|
assert.False(t, hand.containsPair(), "Expected hand to not be a pair")
|
|
|
|
assert.True(t, hand.HighestRank() == ACE, "Expected highest rank to be an ace")
|
|
assert.False(t, hand.HighestRank() == TEN, "Expected highest rank to not be an ace")
|
|
}
|
|
|
|
func TestUnmadeHand(t *testing.T) {
|
|
hand := Cards{
|
|
TenOfSpades(),
|
|
JackOfDiamonds(),
|
|
QueenOfSpades(),
|
|
KingOfSpades(),
|
|
DeuceOfSpades(),
|
|
}
|
|
assert.False(t, hand.containsStraight(), "Expected hand to not be a straight")
|
|
assert.False(t, hand.containsFlush(), "Expected hand to not be a flush")
|
|
assert.False(t, hand.containsStraightFlush(), "Expected hand to not be a straight flush")
|
|
assert.False(t, hand.containsRoyalFlush(), "Expected hand to not be a royal flush")
|
|
assert.False(t, hand.containsFourOfAKind(), "Expected hand to not be four of a kind")
|
|
assert.False(t, hand.containsFullHouse(), "Expected hand to not be a full house")
|
|
assert.False(t, hand.containsThreeOfAKind(), "Expected hand to not be three of a kind")
|
|
assert.False(t, hand.containsTwoPair(), "Expected hand to not be two pair")
|
|
assert.False(t, hand.containsPair(), "Expected hand to not be a pair")
|
|
assert.True(t, hand.HighestRank() == KING, "Expected highest rank to be a king")
|
|
assert.True(t, hand.isUnmadeHand(), "Expected hand to be unmade")
|
|
}
|
|
|
|
func TestTwoPair(t *testing.T) {
|
|
hand := Cards{
|
|
KingOfSpades(),
|
|
JackOfDiamonds(),
|
|
JackOfSpades(),
|
|
KingOfDiamonds(),
|
|
TenOfSpades(),
|
|
}
|
|
assert.False(t, hand.containsStraight(), "Expected hand to not be a straight")
|
|
assert.False(t, hand.containsFlush(), "Expected hand to not be a flush")
|
|
assert.False(t, hand.containsStraightFlush(), "Expected hand to not be a straight flush")
|
|
assert.False(t, hand.containsRoyalFlush(), "Expected hand to not be a royal flush")
|
|
assert.False(t, hand.containsFourOfAKind(), "Expected hand to not be four of a kind")
|
|
assert.False(t, hand.containsFullHouse(), "Expected hand to not be a full house")
|
|
assert.False(t, hand.containsThreeOfAKind(), "Expected hand to not be three of a kind")
|
|
assert.True(t, hand.containsTwoPair(), "Expected hand to be two pair")
|
|
assert.True(t, hand.containsPair(), "Expected hand to also be a pair")
|
|
assert.True(t, hand.HighestRank() == KING, "Expected highest rank to be a king")
|
|
assert.False(t, hand.isUnmadeHand(), "Expected hand to not be unmade")
|
|
}
|
|
|
|
func TestHandScore(t *testing.T) {
|
|
hand := Cards{
|
|
KingOfSpades(),
|
|
JackOfDiamonds(),
|
|
JackOfSpades(),
|
|
KingOfDiamonds(),
|
|
TenOfSpades(),
|
|
}
|
|
|
|
ph, error := hand.PokerHand()
|
|
assert.Nil(t, error, "Expected no error")
|
|
assert.True(t, ph.Score > 0, "Expected score to be nonzero 0")
|
|
assert.True(t, ph.Score < 100000000000000000, "Expected score to be less than 100000000000000000")
|
|
|
|
fmt.Printf("PokerHand: %v+\n", ph)
|
|
fmt.Printf("PH score: %d\n", ph.Score)
|
|
}
|
|
|
|
func TestTwoPairBug(t *testing.T) {
|
|
// this is an actual bug in the first implementation
|
|
c, err := NewCardsFromString("9♠,9♣,Q♥,Q♦,K♣")
|
|
assert.Nil(t, err, "Expected no error")
|
|
|
|
ph, err := c.PokerHand()
|
|
assert.Nil(t, err, "Expected no error")
|
|
assert.Greater(t, ph.Score, 0, "Expected score to be nonzero 0")
|
|
desc := ph.Description()
|
|
assert.Equal(t, desc, "two pair, queens and nines with a king")
|
|
|
|
fmt.Printf("PokerHand: %v+\n", ph)
|
|
fmt.Printf("PH score: %d\n", ph.Score)
|
|
}
|