Merge pull request #149 from chatmartin/main

Test suites for blackjack and main with pytest

Author: NeedToLookAtMyPictures

casino/games/uno/uno.py

@@ -2,7 +2,7 @@
 import time
 import random
 
-from .player import Player
+from .Player import Player
 from casino.types import GameContext
 from casino.utils import clear_screen, cprint, cinput, display_topbar
 from casino.cards import UnoDeck, UnoCard

tests/__init__.py

@@ -0,0 +1 @@
+#TODO: Add test cases for Blackjack EU

tests/test_blackjack_eu.py

@@ -0,0 +1,244 @@
+from casino.games.blackjack.blackjack import *
+from casino.game_types import *
+
+#Add more tests as needed, potentially testing user inputs
+
+def test_aces():
+    """
+    Tests extreme cases for having aces in your hand to ensure calculation is correct
+    """
+    player = Player(account=Account.generate('test', 100))
+    player.hand.append(StandardCard(suit="hearts", rank="A"))
+    player.hand.append(StandardCard(suit="hearts", rank="A"))
+    assert(player.hand_total == 12)
+    player.hand.append(StandardCard(suit="hearts", rank="A"))
+    player.hand[0].rank = 6
+    assert(player.hand_total == 18)
+    player.hand.pop(1)
+    assert(player.hand_total == 17)
+    player.hand[0].rank = 'A'
+    total = 12
+    while total < 21:
+        player.hand.append(StandardCard(suit="hearts", rank="A"))
+        total += 1
+        assert(player.hand_total == total)
+    while len(player.hand) > 3:
+        player.hand.pop()
+    player.hand[0].rank = 10
+    player.hand[1].rank = 10
+    assert(player.hand_total == 21)
+    player.hand.pop(0)
+    assert(player.hand_total == 21)
+
+def test_bust():
+    """
+    tests scenarios in which the dealer or player busts
+    """
+    ctx = GameContext(account=Account.generate('test', 100),config=Config.default())
+    blackjack = StandardBlackjack(ctx)
+    player = blackjack.players[0]
+    player.bet = 10
+    #First test case where the player busts and loses
+    blackjack.deal_cards()
+    #Add cards to the player's hand to ensure that they lose
+    while player.hand_total <= 21:
+        player.hand.append(StandardCard(suit="hearts",rank="K"))
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0]=='lose')
+    assert ('bust' in blackjack.round_results[len(blackjack.round_results) - 1])
+
+    #Next case: Dealer busts and loses
+    blackjack.deal_cards()
+    while blackjack.dealer.hand_total <= 21:
+        blackjack.dealer.hand.append(StandardCard(suit="hearts",rank="K"))
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0]=='win')
+    assert ('bust' in blackjack.round_results[len(blackjack.round_results) - 1])
+
+    #Next case: Player has 10-10-A and Dealer busts
+    blackjack.deal_cards()
+    player.hand[0] = StandardCard(suit="hearts",rank="K")
+    player.hand[1] = StandardCard(suit="hearts", rank="K")
+    player.hand.append(StandardCard(suit="hearts", rank="A"))
+    blackjack.dealer.hand[0] = StandardCard(suit="hearts",rank="K")
+    blackjack.dealer.hand[1] = StandardCard(suit="hearts",rank="K")
+    blackjack.dealer.hand.append(StandardCard(suit="hearts",rank="K"))
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0]=='win')
+    assert('bust' in blackjack.round_results[len(blackjack.round_results)-1])
+    #We will test the payout in this scenario as well
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == (balance+player.bet*2))
+
+def test_payout():
+    """
+    Tests the changes in the user's account for every standard scenario
+    Normal win: profit = amount bet
+    Push: profit = 0
+    Loss: profit = -bet
+    """
+    ctx = GameContext(account=Account.generate('test', 100), config=Config.default())
+    blackjack = StandardBlackjack(ctx)
+    player = blackjack.players[0]
+    dealer = blackjack.dealer
+    player.bet = 10
+    #Case 1: the player beats the dealer normally
+    player.hand.append(StandardCard(suit="hearts",rank="K"))
+    player.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank=9))
+    dealer.hand.append(StandardCard(suit="hearts", rank=3))
+    #Make sure the win result is correct
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'win')
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    #Should be balance before payout + bet*2 because you get back the original money you bet plus the profit
+    assert(player.account.balance == (balance+player.bet*2))
+
+    #Case 2: the dealer beats the player normally
+    player.hand.pop()
+    player.hand.pop()
+    dealer.hand.pop()
+    dealer.hand.pop()
+    dealer.hand.pop()
+    player.hand.append(StandardCard(suit="hearts", rank="10"))
+    player.hand.append(StandardCard(suit="hearts", rank="10"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="10"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="9"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="2"))
+    #Make sure the win result is correct
+    blackjack.check_win()
+    assert (blackjack.player_win_status[0] == 'lose')
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    #Balance should be the same as before because you lost the amount you bet already
+    assert(player.account.balance == balance)
+
+    #Case 3: Ties
+    player.hand.pop()
+    player.hand.pop()
+    dealer.hand.pop()
+    dealer.hand.pop()
+    dealer.hand.pop()
+    player.hand.append(StandardCard(suit="hearts", rank="K"))
+    player.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="K"))
+    #Make sure the win result is correct
+    blackjack.check_win()
+    assert (blackjack.player_win_status[0] == 'tie')
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    #Balance should be higher by the amount bet since it is returned in the case of a tie
+    assert(player.account.balance == (balance+player.bet))
+
+def test_blackjack_payout():
+    """
+    Tests the wins/payouts for every blackjack scenario
+    Both get blackjack round 1: Tie
+    Player gets blackjack and dealer does not: Blackjack win (profit = 1.5*amount bet)
+    Player gets blackjack round 1 and dealer has 21 in round 2: Blackjack win
+    Player gets 21 in round 2 and dealer has blackjack: Loss
+    Both player and dealer have 21 but neither have blackjack: Tie
+    """
+    ctx = GameContext(account=Account.generate('test', 100), config=Config.default())
+    blackjack = StandardBlackjack(ctx)
+    player = blackjack.players[0]
+    dealer = blackjack.dealer
+    player.bet = 10
+    #Case 1: Both the dealer and the player get blackjack
+    player.hand.append(StandardCard(suit="hearts", rank="A"))
+    player.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank="A"))
+    dealer.hand.append(StandardCard(suit="hearts", rank=10))
+    #Make sure the win result is correct
+    blackjack.blackjack_check()
+    assert player.has_blackjack and dealer.has_blackjack
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'tie')
+    assert('blackjack' in blackjack.round_results[len(blackjack.round_results) - 1])
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == (balance+player.bet))
+
+    #Case 2: The player gets blackjack and the dealer does not
+    dealer.hand[0].rank = 10
+    dealer.hand[1].rank = 'A'
+    dealer.hand.append(StandardCard(suit="hearts", rank="A"))
+    #Make sure the results are correct
+    blackjack.blackjack_check()
+    assert(player.has_blackjack and not dealer.has_blackjack)
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'win')
+    assert('blackjack' in blackjack.round_results[len(blackjack.round_results) - 1])
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == (balance+player.bet*(1+BLACKJACK_MULTIPLIER)))
+
+    #Case 3: Player has blackjack, dealer 21 with more than 2 cards
+    dealer.hand[1].rank = 10
+    blackjack.blackjack_check()
+    assert player.has_blackjack and not dealer.has_blackjack
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'win')
+    assert('blackjack' in blackjack.round_results[len(blackjack.round_results) - 1])
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == (balance+player.bet*(1+BLACKJACK_MULTIPLIER)))
+
+    #Case 4: Both dealer and player have 21 but more than 2 cards
+    player.hand[0].rank = 'K'
+    player.hand.append(StandardCard(suit="hearts", rank='A'))
+    blackjack.blackjack_check()
+    assert not(player.has_blackjack or dealer.has_blackjack)
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'tie')
+    assert ('blackjack' not in blackjack.round_results[len(blackjack.round_results) - 1])
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == (balance+player.bet))
+
+    #Case 5: Player has 21 but dealer has blackjack
+    dealer.hand.pop()
+    dealer.hand[0].rank = 'A'
+    dealer.hand[1].rank = 'Q'
+    blackjack.blackjack_check()
+    assert not player.has_blackjack and dealer.has_blackjack
+    blackjack.check_win()
+    assert(blackjack.player_win_status[0] == 'lose')
+    assert 'blackjack' in blackjack.round_results[len(blackjack.round_results) - 1]
+    #Test the payout
+    balance = player.account.balance
+    blackjack.payout()
+    assert(player.account.balance == balance)
+
+def test_dealer_decision():
+    """
+    Tests the dealer's decisions in certain situations
+    Dealer total is less than or equal to 16: Hit
+    Dealer total is greater than or equal to 17: Stand
+    Edge case: Dealer stands on Ace + 6
+    """
+    ctx = GameContext(account=Account.generate('test', 100), config=Config.default())
+    blackjack = StandardBlackjack(ctx)
+    dealer = blackjack.dealer
+    dealer.hand.append(StandardCard(suit="hearts", rank="K"))
+    dealer.hand.append(StandardCard(suit="hearts", rank=8))
+    blackjack.dealer_draw()
+    assert(len(dealer.hand) == 2)
+    dealer.hand[1].rank = 6
+    blackjack.dealer_draw()
+    assert(len(dealer.hand) == 3)
+    dealer.hand.pop()
+    dealer.hand[0].rank = 'A'
+    blackjack.dealer_draw()
+    assert(len(dealer.hand) == 2)

tests/test_blackjack_us.py

@@ -0,0 +1,57 @@
+from unittest.mock import patch,ANY,call,Mock
+from casino.main import *
+from casino.games import *
+import pytest
+
+#Add more tests as needed
+
+
+def test_empty_name_and_quit():
+    inputs = ["","TEST","1","8","q"]
+
+    with patch("casino.main.cinput",side_effect=inputs), \
+        patch("casino.main.get_theme"), \
+        patch("casino.main.Account.generate") as mock_generate, \
+        patch("casino.main.cprint") as mock_print, \
+        patch("casino.main.clear_screen"), \
+        patch("casino.main.display_topbar"):
+
+        main()
+
+    mock_generate.assert_called_with('TEST',ANY)
+    mock_print.assert_called_with("\nGoodbye!\n")
+
+def test_interrupt():
+    with patch("casino.main.cinput", side_effect=KeyboardInterrupt):
+        with pytest.raises(KeyboardInterrupt):
+            main()
+
+def test_invalid_game():
+    ctx = GameContext(account=Account.generate('test', 100), config=Config.default())
+    inputs = ["E","Poker","Blackjack","[1]","20","1.5","Quit","-1",KeyboardInterrupt]
+    with patch("casino.main.cinput",side_effect=inputs), \
+         patch("casino.main.cprint") as mock_print:
+            with pytest.raises(KeyboardInterrupt):
+                main_menu(ctx)
+    mock_print.assert_called_with("\nInvalid input. Please try again.\n")
+    assert mock_print.call_args_list.count(call("\nInvalid input. Please try again.\n"))==len(inputs)-2
+
+@pytest.mark.parametrize("game_index, game_name", [(str(i + 1), name) for i, name in enumerate(ALL_GAMES)])
+def test_game_handler_called(game_index, game_name):
+    ctx = GameContext(account=Account.generate("test", 100), config=Config.default())
+    handlers = {name: Mock() for name in ALL_GAMES}
+    with patch("casino.main.prompt_with_refresh", side_effect=["e", game_index, "q"]), \
+         patch("casino.main.GAME_HANDLERS", handlers), \
+         patch("casino.main.ALL_GAMES", ALL_GAMES), \
+         patch("casino.main.cprint"), \
+         patch("casino.main.clear_screen"), \
+         patch("casino.main.display_topbar"), \
+         patch("casino.main.get_theme"), \
+         patch("casino.main.cinput", return_value="q"):
+          main_menu(ctx)
+
+    handlers[game_name].assert_called_once()
+
+    for other_name, mock_func in handlers.items():
+        if other_name != game_name:
+            mock_func.assert_not_called()

tests/test_inputs.py

@@ -0,0 +1 @@
+#TODO: Add test cases for poker

tests/test_poker.py

@@ -0,0 +1 @@
+#TODO: Add test cases for roulette

tests/test_roulette.py

@@ -0,0 +1 @@
+#TODO: Add test cases for roulette EU

tests/test_roulette_eu.py

@@ -0,0 +1 @@
+#TODO: Add test cases for slots

tests/test_slots.py

@@ -0,0 +1 @@
+#TODO: Add test cases for Uno