def test_all_generate_actions_are_applicable(game):
g = game()
gen = ActionGenerator(g)
assert list(gen.random_actions()) == []
g.start()
available_actions = list(gen.random_actions())
assert len(available_actions) > 0
for action in available_actions:
action.apply(g.copy())
def test_playout(game):
g = game()
g.start()
action = None
for action in ActionGenerator(g, apply=True):
print(g.turn, action)
assert action.is_terminal
def get_move(self, game: Game):
actions = set(ActionGenerator(game))
attack_actions = set(filter(lambda a: isinstance(a, AttackNode), actions))
attack_hero = set(filter(lambda a: isinstance(a.victim, GamePlayer), attack_actions))
attack_minions = attack_actions - attack_hero
for action_type in [attack_hero, attack_minions, actions]:
actions = list(action_type)
random.shuffle(actions)
for action in actions:
return action
def test_play_action_with_ability_cards(game):
cards = [
MinionCard(name='Animated Statue', cost=0, attack=10, health=10),
MinionCard(name='Aberration',
cost=0,
attack=1,
health=1,
ability=ChargeAbility()),
AbilityCard(name='Flamestrike', cost=0, ability=DealDamage(4)),
]
deck = Deck(cards)
game.NUMBERS_OF_START_CARDS = (2, 0)
g = game(player_decks=[deck, deck])
player, _ = g.players
g.start()
g.play(player.id, player.hand[0].id)
gen = ActionGenerator(g)
assert len(list(gen.play_actions())) == 2
def test_random_action_generator(game):
cards = [
MinionCard('test minion', health=1, attack=1, cost=0),
MinionCard('test minion 2', health=1, attack=1, cost=0),
MinionCard('test charge minion',
health=1,
attack=1,
cost=0,
ability=ChargeAbility()),
]
deck = Deck(cards)
game.NUMBERS_OF_START_CARDS = (0, 0)
g = game(player_decks=[deck, deck])
first_player, second_player = g.players
g.start()
g.play(first_player.id, first_player.hand[0].id)
g.endturn(g.current_player.id)
g.play(second_player.id, second_player.hand[0].id)
g.endturn(g.current_player.id)
gen = ActionGenerator(g)
assert len(list(gen.attack_actions())) == 2
assert len(list(gen.play_actions())) == 1
assert len(list(gen.endturn_action())) == 1
assert len(list(gen.random_actions())) == 4
def test_attack_action_generator_with_enemies(game):
cards = [
MinionCard('test minion', health=1, attack=1, cost=0),
MinionCard('test minion 2', health=1, attack=1, cost=0),
MinionCard('test charge minion',
health=1,
attack=1,
cost=0,
ability=ChargeAbility()),
]
deck = Deck(cards)
game.NUMBERS_OF_START_CARDS = (0, 0)
g = game(player_decks=[deck, deck])
first_player, second_player = g.players
g.start()
g.play(first_player.id, first_player.hand[0].id)
g.endturn(g.current_player.id)
g.play(second_player.id, second_player.hand[0].id)
g.endturn(g.current_player.id)
g.play(first_player.id, first_player.hand[0].id)
gen = ActionGenerator(g)
assert len(g.board) == 3
assert len(g.board.played_cards(first_player)) == 2
assert len(g.board.played_cards(second_player)) == 1
# two minions (played in previous turn and having charge ability) can attack minion and second player
assert len(list(gen.attack_actions())) == 4
assert len(list(gen.play_actions())) == 0
assert len(list(gen.endturn_action())) == 1
assert len(list(gen.random_actions())) == 5
assert first_player.health == second_player.health == 20
assert len(g.board) == 3
def test_attack_action_generator_no_board_enemies(game):
cards = [
MinionCard('test minion', health=1, attack=1, cost=0),
MinionCard('test minion 2', health=1, attack=1, cost=0),
MinionCard('test charge minion',
health=1,
attack=1,
cost=0,
ability=ChargeAbility()),
]
deck = Deck(cards)
game.NUMBERS_OF_START_CARDS = (2, 0)
g = game(player_decks=[deck, deck])
player, _ = g.players
g.start()
for card in cards:
g.play(player.id, card.id)
gen = ActionGenerator(g)
assert len(player.hand) == 0
assert len(list(
gen.attack_actions())) == 1 # only charge minion can attack
assert len(list(gen.play_actions())) == 0
assert len(list(gen.endturn_action())) == 1
assert len(list(gen.random_actions())) == 2
assert player.health == 20
def test_node_expand_abilities():
tree = GameTree()
root = tree.root
tree.game.start()
actions = list(ActionGenerator(tree.game))
assert root.is_expandable
for _ in actions:
assert root.is_expandable
tree.expand(root)
assert len(root.children) == len(actions)
tree.expand(root)
assert len(root.children) == len(actions)
assert not root.is_expandable
def test_play_action_generator(game):
deck = Deck([
MinionCard('test minion', health=1, attack=1, cost=1),
MinionCard('test minion 2', health=1, attack=1, cost=1),
MinionCard('test costly minion', health=1, attack=1, cost=100),
])
game.NUMBERS_OF_START_CARDS = (2, )
g = game(player_decks=[deck])
player, = g.players
g.start()
gen = ActionGenerator(g)
assert len(player.hand) == 3
assert len(list(gen.attack_actions())) == 0
assert len(list(gen.play_actions())) == 2
assert len(list(gen.endturn_action())) == 1
assert len(list(gen.random_actions())) == 3
def get_move(self, game: Game):
generator = iter(ActionGenerator(game))
return next(generator)