Hanabi/Py-Hanabi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

hanabi.py: raise error if to-be-replaced card is not in hand

Browse source
This commit is contained in:
Maximilian Keßler 2023-06-24 17:21:55 +02:00
parent e43b062dda
commit 19501aa4da
Signed by: max
GPG key ID: BCC5A619923C0BA5
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
hanabi.py
View file

@ -103,6 +103,7 @@ class HanabiInstance:
self.fives_give_clue = fives_give_clue self.fives_give_clue = fives_give_clue
self.deck_plays = deck_plays, self.deck_plays = deck_plays,
self.all_or_nothing = all_or_nothing self.all_or_nothing = all_or_nothing
assert not self.all_or_nothing, "All or nothing not implemented"
# normalize deck indices # normalize deck indices
for (idx, card) in enumerate(self.deck): for (idx, card) in enumerate(self.deck):
@ -145,7 +146,7 @@ class HanabiInstance:
class GameState: class GameState:
def __init__(self, instance: HanabiInstance): def __init__(self, instance: HanabiInstance, starting_player: int = 0):
# will not be modified # will not be modified
self.instance = instance self.instance = instance
@ -156,7 +157,7 @@ class GameState:
self.stacks = [0 for i in range(0, self.instance.num_suits)] self.stacks = [0 for i in range(0, self.instance.num_suits)]
self.strikes = 0 self.strikes = 0
self.clues = 8 self.clues = 8
self.turn = 0 self.turn = starting_player
self.pace = self.instance.initial_pace self.pace = self.instance.initial_pace
self.remaining_extra_turns = self.instance.num_players + 1 self.remaining_extra_turns = self.instance.num_players + 1
self.trash = [] self.trash = []
@ -310,7 +311,7 @@ class GameState:
# replaces the specified card (has to be in current player's hand) with the next card of the deck (if nonempty) # replaces the specified card (has to be in current player's hand) with the next card of the deck (if nonempty)
def _replace(self, card_idx, allow_not_present: bool = False): def _replace(self, card_idx, allow_not_present: bool = False):
try: try:
idx_in_hand = next((i for (i, card) in enumerate(self.cur_hand) if card.deck_index == card_idx), None) idx_in_hand = next((i for (i, card) in enumerate(self.cur_hand) if card.deck_index == card_idx))
except StopIteration: except StopIteration:
if not allow_not_present: if not allow_not_present:
raise raise