def test_deck_correctness(self):
deck = Deck(shuffle=True, is_small=False)
cards = [deck.pull_out() for _ in range(52)]
with self.subTest():
with self.assertRaises(EmptyDeckException):
deck.pull_out()
with self.subTest():
sorted_cards = []
for suit in SUIT:
for value in VALUE:
sorted_cards.append(Card(suit, value))
self.assertNotEqual(cards, sorted_cards)
class Engine:
def __init__(self, deck=None):
self.users = []
self.deck = Deck(shuffle=True,
is_small=False) if deck is None else deck
self.bj_gametable = None
self.is_inited = False
def add_user(self, user: User):
self.users.append(user)
def init_game(self):
self.bj_gametable = BlackJackGameTable(self.users)
for user in self.bj_gametable.get_active_users():
self.bj_gametable.add_card(user, self.deck.pull_out())
self.bj_gametable.add_card(user, self.deck.pull_out())
self.is_inited = True
def get_active_users(self):
return self.bj_gametable.get_active_users()
def get_game_table_info(self):
return deepcopy(self.bj_gametable.users_status), deepcopy(
self.bj_gametable.users_cards)
def one_tick(self):
if not self.is_inited:
raise GameNotInitedException(
"Before .one_tick() you should call .init_game()")
for user in self.bj_gametable.get_active_users():
turn = user.make_turn(*self.get_game_table_info())
if self.process_turn(user, turn):
self.update_users()
else:
break
if self.is_ended():
return
def is_ended(self) -> bool:
alive = self.bj_gametable.get_active_users()
if not alive:
return True
if len(alive) == 1:
scores = [
self.bj_gametable.get_user_score(user) for user in self.users
]
max_score = max(scores, key=lambda x: 0 if x > 21 else x)
if max_score == self.bj_gametable.get_user_score(alive[0]):
return True
one_not_looser = False
for status in self.bj_gametable.users_status.values():
if status != UserStatus.LOSE:
if one_not_looser:
return False
one_not_looser = True
return one_not_looser
def update_users(self):
for user in self.users:
user.update_table(*self.get_game_table_info())
def generate_outcomes(self):
result = dict()
scores = [
self.bj_gametable.get_user_score(user) for user in self.users
]
max_score = max(scores, key=lambda x: 0 if x > 21 else x)
if max_score > 21:
max_score = 0
winners_count = scores.count(max_score)
for score, user in zip(scores, self.users):
result[user.user_id] = (score, GameOutcome.LOSER)
if score == max_score:
outcome = GameOutcome.WINNER if winners_count == 1 else GameOutcome.DRAW
result[user.user_id] = (score, outcome)
return result
def outcomes_notify(self, outcomes: dict):
for user in self.users:
user.outcome_notify(*outcomes[user.user_id])
def process_turn(self, user: User, turn: TURN):
if user in self.bj_gametable.get_active_users():
if turn is TURN.HIT_ME:
new_card = self.deck.pull_out()
self.bj_gametable.add_card(user, new_card)
elif turn is TURN.ENOUGH:
self.bj_gametable.set_status(user, UserStatus.ENOUGH)
else:
return False
elif turn is not None:
raise WrongRightsException(f"User {user.user_id} is not in game")
return True
def test_different_card_orders(self):
first_deck = Deck(shuffle=True, is_small=False)
second_deck = Deck(shuffle=True, is_small=False)
first_deck_cards = [first_deck.pull_out() for _ in range(52)]
second_deck_cards = [second_deck.pull_out() for _ in range(52)]
self.assertNotEqual(first_deck_cards, second_deck_cards)