def test_generateRandom(self):
"""Testing generation of random deals"""
deal = Deal.fromRandom()
try:
self.validateDeal(deal)
except Exception as e:
self.fail(e, deal)
def test_fromString(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
deal = Deal.fromString(deal_str)
north_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Two', 'Four', 'Eight', 'Nine', 'Ace')) ] + \
[ Card(Rank.Four, Suit.Diamond) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Five', 'Eight', 'Ten')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Three', 'Six', 'Ten', 'Ace')) ]
east_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Three', 'Five', 'Seven', 'Queen')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Three', 'Six', 'Seven', 'Jack', 'King')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Four', 'Nine', 'Queen')) ] + \
[ Card(Rank.Four, Suit.Spade) ]
south_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Ten', 'King')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Five', 'Eight')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Two', 'Six', 'Seven', 'King')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Five', 'Eight', 'Nine', 'Jack', 'King')) ]
west_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Six', 'Jack')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Two', 'Nine', 'Ten', 'Queen', 'Ace')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Three', 'Jack', 'Ace')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Two', 'Seven', 'Queen')) ]
self.assertEqual(
{
Direction.North: north_hand,
Direction.East: east_hand,
Direction.South: south_hand,
Direction.West: west_hand
}, deal)
class TestTrickPlay(unittest.TestCase):
deal = Deal.fromIndex(31415926535897932384626433832) # Pi!
declarer = Direction.North
trumpSuit = Suit.Spade
def setUp(self):
self.trickplay = TrickPlay(self.declarer, self.trumpSuit)
def tearDown(self):
self.trickplay = None
def stepThroughTrickPlay(self):
for i in range(52):
card, position = self.getValidArgsForPlayCard()
yield card, position
self.trickplay.playCard(card, position)
def getValidArgsForPlayCard(self):
# Select the next card to play, deterministically.
assert not self.trickplay.isComplete()
turn = self.trickplay.whoseTurn()
card = self.deal[turn][0]
# Assumes correctness of isValidCardPlay.
while not self.trickplay.isValidCardPlay(card, self.deal):
card = self.deal[turn][self.deal[turn].index(card) + 1]
return card, turn
def testIsComplete(self):
s = self.stepThroughTrickPlay()
try:
while next(s):
self.assertEqual(self.trickplay.isComplete(), False)
except StopIteration:
self.assertEqual(self.trickplay.isComplete(), True)
# for trick in self.trickplay:
# print "leader", trick.leader
# for pos, card in trick.items():
# print str(pos), card
# print "winner", trick.winner, trick.winningCard
def testWhoseTurn(self):
s = self.stepThroughTrickPlay()
self.assertEqual(self.trickplay.whoseTurn(), self.trickplay.lho)
def test_fromIndex(self):
"""Testing Deal.fromIndex over a set of known indexes"""
for index, deal in list(self.samples.items()):
self.assertEqual(Deal.fromIndex(index), deal)
class TestDeck(unittest.TestCase):
cards = sorted(Card(r, s) for r in Rank for s in Suit)
samples = {}
samples[1] = Deal({
Direction.North: [Card(r, Suit.Spade) for r in Rank],
Direction.East: [Card(r, Suit.Heart) for r in Rank],
Direction.South: [Card(r, Suit.Diamond) for r in Rank],
Direction.West: [Card(r, Suit.Club) for r in Rank]
})
# http://bridgehands.com/D/Duke_of_Cumberland_Hand.htm
samples[25216995119420903953708290155] = Deal.fromString(
"N:..Q8765432.AQT84 65432.T9872.JT9. T987.6543..76532 AKQJ.AKQJ.AK.KJ9"
)
# http://bridgehands.com/B/John_Bennett_Murder.htm
samples[49115408832893597588305377049] = Deal.fromString(
"S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
)
# From the PBN v2.0 specification document.
samples[51845212465382378289082480212] = Deal.fromString(
"N:.63.AKQ987.A9732 A8654.KQ5.T.QJT6 J973.J98742.3.K4 KQT2.AT.J6542.85"
)
# http://bridgehands.com/M/Mississippi_Heart_Hand.htm
samples[53520933857671775260919265981] = Deal.fromString(
"S:AKQ.AKQJT9..AKQJ .8765432.AKQJT9. T5432..5432.5432 J9876..876.T9876"
)
def validateDeal(self, deal):
"""Checks that structure of deal conforms to requirements:
- Each position in Direction maps to a hand, represented as a list.
- Hand lists contain exactly 13 Card objects.
- No card may be repeated in the same hand, or between hands.
@param deal: a Deal instance.
"""
assert isinstance(deal, Deal), "deal not a Deal instance"
assert set(deal.keys()) == set(Direction), "invalid set of keys"
extractcards = []
for pos, hand in list(deal.items()):
assert len(hand) == 13, "%s hand does not contain 13 cards" % pos
extractcards.extend(hand)
assert self.cards == sorted(extractcards), "not a pure set of cards"
def test_generateRandom(self):
"""Testing generation of random deals"""
deal = Deal.fromRandom()
try:
self.validateDeal(deal)
except Exception as e:
self.fail(e, deal)
@unittest.skip("not used")
def test_toIndex(self):
"""Testing toIndex method over a set of known deals"""
for index, deal in list(self.samples.items()):
self.assertEqual(deal.toIndex(), index)
@unittest.skip("not used")
def test_fromIndex(self):
"""Testing Deal.fromIndex over a set of known indexes"""
for index, deal in list(self.samples.items()):
self.assertEqual(Deal.fromIndex(index), deal)
def test_fromString(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
deal = Deal.fromString(deal_str)
north_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Two', 'Four', 'Eight', 'Nine', 'Ace')) ] + \
[ Card(Rank.Four, Suit.Diamond) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Five', 'Eight', 'Ten')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Three', 'Six', 'Ten', 'Ace')) ]
east_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Three', 'Five', 'Seven', 'Queen')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Three', 'Six', 'Seven', 'Jack', 'King')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Four', 'Nine', 'Queen')) ] + \
[ Card(Rank.Four, Suit.Spade) ]
south_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Ten', 'King')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Five', 'Eight')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Two', 'Six', 'Seven', 'King')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Five', 'Eight', 'Nine', 'Jack', 'King')) ]
west_hand = [ Card(rnk, Suit.Club) for rnk in (Rank[r] for r in ('Six', 'Jack')) ] + \
[ Card(rnk, Suit.Diamond) for rnk in (Rank[r] for r in ('Two', 'Nine', 'Ten', 'Queen', 'Ace')) ] + \
[ Card(rnk, Suit.Heart) for rnk in (Rank[r] for r in ('Three', 'Jack', 'Ace')) ] + \
[ Card(rnk, Suit.Spade) for rnk in (Rank[r] for r in ('Two', 'Seven', 'Queen')) ]
self.assertEqual(
{
Direction.North: north_hand,
Direction.East: east_hand,
Direction.South: south_hand,
Direction.West: west_hand
}, deal)
def test_multi_space_ok(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
Deal.fromString(deal_str)
def test_fromString_repeated_card(self):
deal_str = "S:KJ985.K762.85.KT KQ72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Card already seen: Card(Rank.King, Suit.Spade)",
ctx.exception.message)
def test_fromString_missing_card(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A942 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Card missing: Card(Rank.Eight, Suit.Club)",
ctx.exception.message)
def test_fromString_unbalanced_hand(self):
deal_str = "S:KQJ985.K762.85.KT 72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Incorrect number of cards (14) in South hand",
ctx.exception.message)
def test_fromString_missing_space(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A98424.Q94.KJ763.Q753"
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_missing_dot(self):
deal_str = "S:KJ985.K762.85KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_extra_hand(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753 AKQJT98765432..."
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_missing_hand(self):
deal_str = "S:KJ985.K762.85.KT AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_invalid_character_cards(self):
deal_str = "S:YJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(KeyError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("KeyError: 'Y'", ctx.exception.message)
def test_invalid_character_direction(self):
deal_str = "F:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(KeyError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("KeyError: 'F'", ctx.exception.message)
def test_invalid_character_direction(self):
deal_str = "F:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(KeyError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("KeyError: 'F'", ctx.exception.message)
def test_fromString_missing_hand(self):
deal_str = "S:KJ985.K762.85.KT AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_extra_hand(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753 AKQJT98765432..."
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_missing_dot(self):
deal_str = "S:KJ985.K762.85KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError):
Deal.fromString(deal_str)
def test_fromString_unbalanced_hand(self):
deal_str = "S:KQJ985.K762.85.KT 72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Incorrect number of cards (14) in South hand",
ctx.exception.message)
def test_fromString_missing_card(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A942 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Card missing: Card(Rank.Eight, Suit.Club)",
ctx.exception.message)
def test_fromString_repeated_card(self):
deal_str = "S:KJ985.K762.85.KT KQ72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
with self.assertRaises(ValueError) as ctx:
Deal.fromString(deal_str)
self.assertEqual("Card already seen: Card(Rank.King, Suit.Spade)",
ctx.exception.message)
def test_multi_space_ok(self):
deal_str = "S:KJ985.K762.85.KT Q72.AJ3.AQT92.J6 AT63.T85.4.A9842 4.Q94.KJ763.Q753"
Deal.fromString(deal_str)