def test_all_combinations(self):
pairs = []
for card1 in Card.cards_52():
for card2 in Card.cards_52():
if card1 != card2:
pairs += [CardsPair(card1, card2)]
pair = CardsPair(card1)
pair.set(card2)
pairs += [pair]
pair = CardsPair(card2)
pair.set(card1)
pairs += [pair]
pair = CardsPair()
pair.set(card1)
pair.set(card2)
pairs += [pair]
pair = CardsPair()
pair.set(card2)
pair.set(card1)
pairs += [pair]
pairs2 = CardsPair.get_all_pairs()
self.assertEqual(len(pairs2), len(set(pairs)))
def test_can_not_add_same_card(self):
for card1 in Card.cards_52():
with self.assertRaises(InitializeWithSameCard):
CardsPair(card1, card1)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair(card1)
pair.set(card1)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair()
pair.set(card1)
pair.set(card1)
for card2 in Card.cards_52():
if card1 != card2:
CardsPair(card1, card2)
pair = CardsPair(card1)
pair.set(card2)
pair = CardsPair(card2)
pair.set(card1)
pair = CardsPair()
pair.set(card1)
pair.set(card2)
pair = CardsPair()
pair.set(card2)
pair.set(card1)
else:
with self.assertRaises(InitializeWithSameCard):
CardsPair(card1, card2)
with self.assertRaises(InitializeWithSameCard):
CardsPair(card2, card1)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair(card1)
pair.set(card2)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair(card2)
pair.set(card1)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair()
pair.set(card1)
pair.set(card2)
with self.assertRaises(InitializeWithSameCard):
pair = CardsPair()
pair.set(card2)
pair.set(card1)
def test_first_always_stronger(self):
weak = Card('2C')
strong = Card('AD')
pair = CardsPair(strong, weak)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
pair = CardsPair(weak, strong)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
pair = CardsPair(weak)
self.assertEqual(pair.first, weak)
self.assertEqual(pair.second, None)
pair.set(strong)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
pair = CardsPair(strong)
self.assertEqual(pair.first, strong)
pair.set(weak)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
pair = CardsPair()
self.assertEqual(pair.first, None)
self.assertEqual(pair.second, None)
pair.set(weak)
self.assertEqual(pair.first, weak)
self.assertEqual(pair.second, None)
pair.set(strong)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
pair = CardsPair()
pair.set(strong)
pair.set(weak)
self.assertEqual(pair.first, strong)
self.assertEqual(pair.second, weak)
def test_raise_third_card_add(self):
with self.assertRaises(CanNotAddAnotherCard):
self.full.set(Card('AS'))
self.half.set(Card('TC'))
with self.assertRaises(CanNotAddAnotherCard):
self.half.set(Card('AS'))
self.none.set(Card('2D'))
self.none.set(Card('3D'))
with self.assertRaises(CanNotAddAnotherCard):
self.none.set(Card('AS'))
def setUp(self):
self.none = CardsPair()
self.half = CardsPair(Card('KS'))
self.full = CardsPair(Card('QS'), Card('TH'))
def test_suitability(self):
suited = CardsPair(Card('AS'), Card('QS'))
self.assertEqual(suited.suitability, Suitability.Suited)
offsuited = CardsPair(Card('AS'), Card('QC'))
self.assertEqual(offsuited.suitability, Suitability.Offsuited)
def test_hash(self):
pair1 = CardsPair(Card('AS'), Card('5D'))
pair2 = CardsPair(Card('AS'), Card('5C'))
pair3 = CardsPair(Card('AS'), Card('5D'))
self.assertEqual(len({pair1, pair2, pair3}), 2)
def test_equal(self):
pair1 = CardsPair(Card('AS'), Card('5D'))
pair2 = CardsPair(Card('AS'), Card('5C'))
pair3 = CardsPair(Card('AS'), Card('5D'))
self.assertNotEqual(pair1, pair2)
self.assertEqual(pair1, pair3)