def test_simulate_game_empty_deck(self):
mock_deck = Deck()
mock_deck.deal = MagicMock(return_value=([], []))
game = Egrt(0.5, deck=mock_deck)
winner, turns = game.simulate_game()
self.assertEqual(winner, 'T')
self.assertEqual(turns, 0)
def main():
parser = ArgumentParser(description='play a game of Egyptian Rat Tail')
parser.add_argument('--a_slap_probability', type=float, required=True)
parser.add_argument('--debug_print', action='store_true')
args = parser.parse_args()
game = Egrt(args.a_slap_probability)
result_tuple = game.simulate_game(args.debug_print)
if args.debug_print:
print(result_tuple)
return result_tuple
def test_simulate_queen_wins(self):
mock_deck = Deck()
mock_deck.deal = MagicMock(return_value=([
Card(Rank.SIX, Suit.DIAMOND),
Card(Rank.SEVEN, Suit.CLUB),
Card(Rank.EIGHT, Suit.SPADE)
], [Card(Rank.QUEEN, Suit.HEART),
Card(Rank.TWO, Suit.SPADE)]))
game = Egrt(0.5, deck=mock_deck)
winner, turns = game.simulate_game()
self.assertEqual(winner, 'B')
self.assertEqual(turns, 4)
def test_simulate_game_king_and_jack_win(self):
mock_deck = Deck()
mock_deck.deal = MagicMock(return_value=(
[Card(Rank.KING, Suit.CLUB),
Card(Rank.JACK, Suit.CLUB)], [
Card(Rank.TWO, Suit.CLUB),
Card(Rank.SEVEN, Suit.DIAMOND),
Card(Rank.FOUR, Suit.CLUB),
Card(Rank.NINE, Suit.SPADE)
]))
game = Egrt(0.5, deck=mock_deck)
winner, turns = game.simulate_game()
self.assertEqual(winner, 'A')
self.assertEqual(turns, 6)
def main():
print('output file is {}'.format(FILE_NAME))
filew = open(FILE_NAME, 'w')
filew.write('A_SLAP_PROBABILITY,A_WIN_PERCENTAGE,TURNS_MEAN,TURNS_STDEV\n')
for prob in probabilities:
winners = []
turns = []
for x in range(int(N)):
game = Egrt(prob)
victor, counter = game.simulate_game()
winners.append(victor)
turns.append(counter)
a_win_percentage = winners.count('A') / float(N)
mean_turns = round(mean(turns))
stdev_turns = round(stdev(turns))
print('A slap win probability is {}%'.format(prob))
print('A wins {}%'.format(a_win_percentage))
print(mean_turns, '+/-', stdev_turns)
filew.write('{},{},{},{}\n'.format(prob, a_win_percentage, mean_turns,
stdev_turns))
filew.close()
def test_add_cards_to_hand_no_pile(self):
game = Egrt(0.5)
card_1 = Card(Rank.FOUR, Suit.HEART)
card_2 = Card(Rank.JACK, Suit.CLUB)
game.prev_card = card_1
game.top_card = card_2
actual_hand = deque()
game.add_cards_to_hand(actual_hand)
self.assertEqual(list(actual_hand), [card_1, card_2])
self.assertEqual(len(game.pile), 0)
self.assertIsNone(game.prev_card)
self.assertIsNone(game.top_card)
self.assertIsNone(game.chances_remaining)
def test_add_cards_to_hand_single_card_in_pile(self):
game = Egrt(0.5)
card_1 = Card(Rank.FOUR, Suit.HEART)
card_2 = Card(Rank.SIX, Suit.SPADE)
card_3 = Card(Rank.QUEEN, Suit.CLUB)
game.pile.append(card_1)
game.prev_card = card_2
game.top_card = card_3
actual_hand = deque()
game.add_cards_to_hand(actual_hand)
self.assertEqual(list(actual_hand), [card_1, card_2, card_3])
self.assertEqual(len(game.pile), 0)
self.assertIsNone(game.prev_card)
self.assertIsNone(game.top_card)
self.assertIsNone(game.chances_remaining)
def test_add_cards_to_hand_multiple_cards_in_pile(self):
game = Egrt(0.5)
card_1 = Card(Rank.FOUR, Suit.HEART)
card_2 = Card(Rank.SIX, Suit.SPADE)
card_3 = Card(Rank.QUEEN, Suit.CLUB)
card_4 = Card(Rank.TEN, Suit.DIAMOND)
card_5 = Card(Rank.EIGHT, Suit.CLUB)
card_6 = Card(Rank.TWO, Suit.CLUB)
card_7 = Card(Rank.NINE, Suit.SPADE)
game.pile.extend([card_1, card_2, card_3, card_4, card_5])
game.prev_card = card_6
game.top_card = card_7
actual_hand = deque()
game.add_cards_to_hand(actual_hand)
self.assertEqual(
list(actual_hand),
[card_1, card_2, card_3, card_4, card_5, card_6, card_7])
self.assertEqual(len(game.pile), 0)
self.assertIsNone(game.prev_card)
self.assertIsNone(game.top_card)
self.assertIsNone(game.chances_remaining)
def test_set_chances_remaining_ace(self):
game = Egrt(0.5)
game.top_card = Card(Rank.ACE, Suit.HEART)
game.set_chances_remaining()
self.assertEqual(game.chances_remaining, 4)
def test_set_chances_remaining_non_face_card(self):
game = Egrt(0.5)
game.top_card = Card(Rank.TEN, Suit.HEART)
with self.assertRaises(ValueError):
game.set_chances_remaining()
def test_random_greater_than_a_slap_probability(self, random_mock):
random_mock.return_value = 0.27
game = Egrt(0.26)
self.assertFalse(game.a_won_slap())
def test_is_slap_king_queen_sandwich(self):
game = Egrt(0.5)
game.prev_card = Card(Rank.QUEEN, Suit.DIAMOND)
game.top_card = Card(Rank.SIX, Suit.DIAMOND)
self.assertFalse(game.is_slap(Card(Rank.KING, Suit.SPADE)))
def test_is_slap_single_card_in_pile(self):
game = Egrt(0.5)
game.top_card = Card(Rank.JACK, Suit.DIAMOND)
self.assertTrue(game.is_slap(Card(Rank.JACK, Suit.HEART)))
def test_set_chances_remaining_ace(self):
game = Egrt(0.5)
game.top_card = Card(Rank.JACK, Suit.SPADE)
game.set_chances_remaining()
self.assertEqual(game.chances_remaining, 1)
def test_add_card_to_hand_no_top_or_prev(self):
game = Egrt(0.5)
with self.assertRaises(ValueError):
game.add_cards_to_hand(deque())
def test_a_slap_probability_greater_than_one(self):
with self.assertRaises(ValueError):
_ = Egrt(1.25)
def test_a_slap_probability_is_one(self):
with self.assertRaises(ValueError):
_ = Egrt(1.0)
def test_set_chances_remaining_ace(self):
game = Egrt(0.5)
game.top_card = Card(Rank.KING, Suit.CLUB)
game.set_chances_remaining()
self.assertEqual(game.chances_remaining, 3)
def test_set_chances_remaining_ace(self):
game = Egrt(0.5)
game.top_card = Card(Rank.QUEEN, Suit.DIAMOND)
game.set_chances_remaining()
self.assertEqual(game.chances_remaining, 2)
def test_wrong_deck_class_raises_error(self):
with self.assertRaises(TypeError):
_ = Egrt(0.5, 12)
def test_is_slap_empty_pile(self):
game = Egrt(0.5)
self.assertFalse(game.is_slap(Card(Rank.FIVE, Suit.CLUB)))
def test_is_slap_yes_is_slap(self):
game = Egrt(0.5)
game.prev_card = Card(Rank.ACE, Suit.CLUB)
game.top_card = Card(Rank.QUEEN, Suit.DIAMOND)
self.assertTrue(game.is_slap(Card(Rank.ACE, Suit.HEART)))
def test_is_slap_not_a_slap(self):
game = Egrt(0.5)
game.prev_card = Card(Rank.SEVEN, Suit.CLUB)
game.top_card = Card(Rank.SIX, Suit.DIAMOND)
self.assertFalse(game.is_slap(Card(Rank.FIVE, Suit.HEART)))
def test_add_cards_to_hand_no_prev_card(self):
game = Egrt(0.5)
game.top_card = Card(Rank.THREE, Suit.CLUB)
with self.assertRaises(ValueError):
game.add_cards_to_hand(deque())
def test_is_slap_sandwich_slap(self):
game = Egrt(0.5)
game.prev_card = Card(Rank.SEVEN, Suit.CLUB)
game.top_card = Card(Rank.SIX, Suit.DIAMOND)
self.assertTrue(game.is_slap(Card(Rank.SIX, Suit.HEART)))
def test_random_less_than_a_slap_probability(self, random_mock):
random_mock.return_value = 0.25
game = Egrt(0.26)
self.assertTrue(game.a_won_slap())
def test_random_same_as_a_slap_probability(self, random_mock):
random_mock.return_value = 0.26
game = Egrt(0.26)
self.assertFalse(game.a_won_slap())
def test_a_slap_probability_negative(self):
with self.assertRaises(ValueError):
_ = Egrt(-0.1)
