def do_stuff(net):
def play_game():
new_game = game.copy()
new_game.start()
deck1 = load_deck("example.hsdeck")
deck2 = load_deck("example.hsdeck")
game = Game([deck1, deck2], [CustomAgent(net), RandomAgent()])
game.start()
print(timeit.timeit(play_game, 'gc.enable()', number=1000))
def do_stuff():
def play_game():
new_game = game.copy()
new_game.start()
deck1 = load_deck("example.hsdeck")
deck2 = load_deck("example.hsdeck")
game = Game([deck1, deck2], [RandomAgent(), RandomAgent()])
game.start()
print(timeit.timeit(play_game, 'gc.enable()', number=1000))
def test_deck_shortening(self):
deck1 = Deck([RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(), RagnarosTheFirelord(),
GoldshireFootman(), GoldshireFootman()], CHARACTER_CLASS.DRUID)
deck2 = StackedDeck([StonetuskBoar()], CHARACTER_CLASS.HUNTER)
game = Game([deck1, deck2], [RandomAgent(), RandomAgent()])
replay = record(game)
game.start()
replay.write(StringIO())
def test_first_turn(self):
card_set1 = []
card_set2 = []
for cardIndex in range(0, 30):
card_set1.append(card_lookup("Stonetusk Boar"))
card_set2.append(card_lookup("Novice Engineer"))
deck1 = Deck(card_set1, CHARACTER_CLASS.DRUID)
deck2 = Deck(card_set2, CHARACTER_CLASS.MAGE)
agent1 = mock.Mock(spec=DoNothingBot(), wraps=DoNothingBot())
agent2 = mock.Mock(spec=DoNothingBot(), wraps=DoNothingBot())
game = Game([deck1, deck2], [agent1, agent2])
game.start()
def test_first_turn(self):
card_set1 = []
card_set2 = []
for cardIndex in range(0, 30):
card_set1.append(card_lookup("Stonetusk Boar"))
card_set2.append(card_lookup("Novice Engineer"))
deck1 = Deck(card_set1, CHARACTER_CLASS.DRUID)
deck2 = Deck(card_set2, CHARACTER_CLASS.MAGE)
agent1 = mock.Mock(spec=DoNothingBot(), wraps=DoNothingBot())
agent2 = mock.Mock(spec=DoNothingBot(), wraps=DoNothingBot())
game = Game([deck1, deck2], [agent1, agent2])
game.start()
def test_recording_game(self):
self.maxDiff = None
random.seed(9876)
deck1 = hearthbreaker.game_objects.Deck([StonetuskBoar() for i in range(0, 30)], CHARACTER_CLASS.MAGE)
deck2 = hearthbreaker.game_objects.Deck([Naturalize() for i in range(0, 30)], CHARACTER_CLASS.DRUID)
agent1 = PredictableAgent()
agent2 = PredictableAgent()
game = Game([deck1, deck2], [agent1, agent2])
replay = record(game)
game.start()
output = StringIO()
replay.write_json(output)
f = open("tests/replays/stonetusk_innervate.hsreplay", "r")
dif = self.__compare_json(output.getvalue(), f.read())
self.assertTrue(dif)
f.close()
def test_recording_game(self):
self.maxDiff = None
random.seed(9876)
deck1 = hearthbreaker.game_objects.Deck(
[StonetuskBoar() for i in range(0, 30)], CHARACTER_CLASS.MAGE)
deck2 = hearthbreaker.game_objects.Deck(
[Naturalize() for i in range(0, 30)], CHARACTER_CLASS.DRUID)
agent1 = PredictableAgent()
agent2 = PredictableAgent()
game = Game([deck1, deck2], [agent1, agent2])
replay = record(game)
game.start()
output = StringIO()
replay.write_json(output)
f = open("tests/replays/stonetusk_innervate.hsreplay", 'r')
dif = self.__compare_json(output.getvalue(), f.read())
with open("bler.hsreplay", "w") as debug_file:
replay.write_json(debug_file)
self.assertTrue(dif)
f.close()
def test_deck_shortening(self):
deck1 = Deck([
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
RagnarosTheFirelord(),
GoldshireFootman(),
GoldshireFootman()
], CHARACTER_CLASS.DRUID)
deck2 = StackedDeck([StonetuskBoar()], CHARACTER_CLASS.HUNTER)
game = Game([deck1, deck2], [RandomAgent(), RandomAgent()])
replay = record(game)
game.start()
replay.write(StringIO())
def render_game(stdscr):
class TextAgent:
def __init__(self, game_window, prompt_window, text_window):
self.window = prompt_window
self.game_window = game_window
self.text_window = text_window
curses.init_pair(5, curses.COLOR_WHITE, curses.COLOR_CYAN)
curses.init_pair(6, curses.COLOR_BLACK, curses.COLOR_GREEN)
def do_turn(self, player):
renderer.draw_game()
index = 0
action = self.choose_action()
while not (action == "quit" or action == "end"):
if action == "play":
card = self.choose_card(player)
if card is not None:
player.game.play_card(card)
elif action == "attack":
attacker = self.choose_attacker(player)
if attacker is not None:
attacker.attack()
elif action == "power":
if player.hero.power.can_use():
player.hero.power.use()
index += 1
renderer.draw_game()
action = self.choose_action()
if action == "quit":
sys.exit(0)
def choose_action(self):
self.window.addstr(0, 0, "Choose action")
actions = ["play", "attack", "power", "end", "quit"]
index = 0
selected = 0
for action in actions:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 10, "{0:^9}".format(action), color)
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(actions) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(actions):
selected = 0
index = 0
for action in actions:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 10, "{0:^9}".format(action), color)
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return actions[selected]
def choose_card(self, player):
filtered_cards = [card for card in filter(lambda card: card.can_use(player, player.game), player.hand)]
if len(filtered_cards) is 0:
return None
renderer.targets = filtered_cards
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose Card")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.addstr(0, 0, "Choose Card")
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def choose_attacker(self, player):
filtered_attackers = [minion for minion in filter(lambda minion: minion.can_attack(), player.minions)]
if player.hero.can_attack():
filtered_attackers.append(player.hero)
if len(filtered_attackers) is 0:
return None
renderer.targets = filtered_attackers
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose attacker")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
self.window.addstr(0, 0, "{0}".format(ch))
self.window.refresh()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def do_card_check(self, cards):
self.window.addstr(0, 0, "Select cards to keep (space selects/deselects a card)")
keeping = [True, True, True]
if len(cards) > 3:
keeping.append(True)
index = 0
selected = 0
for card in cards:
if keeping[index]:
if index == selected:
color = curses.color_pair(6)
else:
color = curses.color_pair(5)
else:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(0)
self.text_window.addstr(0, index * 20, "{0:^19}".format(card.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(cards) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(cards):
selected = 0
if ch == 32:
keeping[selected] = not keeping[selected]
index = 0
for card in cards:
if keeping[index]:
if index == selected:
color = curses.color_pair(6)
else:
color = curses.color_pair(5)
else:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(0)
self.text_window.addstr(0, index * 20, "{0:^19}".format(card.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return keeping
def choose_target(self, targets):
if len(targets) is 0:
return None
renderer.targets = targets
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose target")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def choose_index(self, card, player):
renderer.selection_index = 0
renderer.draw_game()
self.window.addstr(0, 0, "Choose placement location")
self.window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
renderer.selection_index -= 1
if renderer.selection_index < 0:
renderer.selection_index = len(player.minions)
if ch == curses.KEY_RIGHT:
renderer.selection_index += 1
if renderer.selection_index > len(player.minions):
renderer.selection_index = 0
renderer.draw_game()
self.window.refresh()
index = renderer.selection_index
renderer.selection_index = -1
if ch == 27:
return -1
return index
def choose_option(self, *options):
self.window.addstr(0, 0, "Choose option")
index = 0
selected = 0
for option in options:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 20, "{0:^19}".format(option.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.game_window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(options) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(options):
selected = 0
index = 0
for option in options:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 20, "{0:^19}".format(option.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return options[selected]
def choose_agent(window):
agents = registry.get_names()
curses.init_pair(3, curses.COLOR_WHITE, curses.COLOR_BLUE)
window.addstr(10, 34, "Choose agent")
ch = 0
selected = 0
while ch != 10 and ch != 27:
index = 0
for agent in agents:
if index == selected:
color = curses.color_pair(3)
else:
color = curses.color_pair(0)
window.addstr(index + 11, 25, "{:^30}".format(agent), color)
index += 1
window.refresh()
ch = window.getch()
if ch == curses.KEY_UP:
selected -= 1
if selected < 0:
selected = len(agents) - 1
if ch == curses.KEY_DOWN:
selected += 1
if selected == len(agents):
selected = 0
window.clear()
if ch == 27:
sys.exit(0)
else:
return registry.create_agent(agents[selected])
stdscr.clear()
prompt_window = stdscr.derwin(1, 80, 23, 0)
text_window = stdscr.derwin(1, 80, 24, 0)
agent = choose_agent(stdscr)
deck1 = load_deck(sys.argv[1])
deck2 = load_deck(sys.argv[2])
game = Game([deck1, deck2], [TextAgent(stdscr, prompt_window, text_window), agent])
if isinstance(game.players[0].agent, TextAgent):
renderer = GameRender(stdscr, game, game.players[0])
else:
renderer = GameRender(stdscr, game, game.players[1])
game.start()
def render_game(stdscr):
class TextAgent:
def __init__(self, game_window, prompt_window, text_window):
self.window = prompt_window
self.game_window = game_window
self.text_window = text_window
curses.init_pair(5, curses.COLOR_WHITE, curses.COLOR_CYAN)
curses.init_pair(6, curses.COLOR_BLACK, curses.COLOR_GREEN)
def do_turn(self, player):
renderer.draw_game()
index = 0
action = self.choose_action()
while not (action == "quit" or action == "end"):
if action == "play":
card = self.choose_card(player)
if card is not None:
player.game.play_card(card)
elif action == "attack":
attacker = self.choose_attacker(player)
if attacker is not None:
attacker.attack()
elif action == "power":
if player.hero.power.can_use():
player.hero.power.use()
index += 1
renderer.draw_game()
action = self.choose_action()
if action == "quit":
sys.exit(0)
def choose_action(self):
self.window.addstr(0, 0, "Choose action")
actions = ["play", "attack", "power", "end", "quit"]
index = 0
selected = 0
for action in actions:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 10, "{0:^9}".format(action), color)
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(actions) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(actions):
selected = 0
index = 0
for action in actions:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 10, "{0:^9}".format(action), color)
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return actions[selected]
def choose_card(self, player):
filtered_cards = [card for card in filter(lambda card: card.can_use(player, player.game), player.hand)]
if len(filtered_cards) is 0:
return None
renderer.targets = filtered_cards
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose Card")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.addstr(0, 0, "Choose Card")
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def choose_attacker(self, player):
filtered_attackers = [minion for minion in filter(lambda minion: minion.can_attack(), player.minions)]
if player.hero.can_attack():
filtered_attackers.append(player.hero)
if len(filtered_attackers) is 0:
return None
renderer.targets = filtered_attackers
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose attacker")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
self.window.addstr(0, 0, "{0}".format(ch))
self.window.refresh()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def do_card_check(self, cards):
self.window.addstr(0, 0, "Select cards to keep (space selects/deselects a card)")
keeping = [True, True, True]
if len(cards) > 3:
keeping.append(True)
index = 0
selected = 0
for card in cards:
if keeping[index]:
if index == selected:
color = curses.color_pair(6)
else:
color = curses.color_pair(5)
else:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(0)
self.text_window.addstr(0, index * 20, "{0:^19}".format(card.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(cards) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(cards):
selected = 0
if ch == 32:
keeping[selected] = not keeping[selected]
index = 0
for card in cards:
if keeping[index]:
if index == selected:
color = curses.color_pair(6)
else:
color = curses.color_pair(5)
else:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(0)
self.text_window.addstr(0, index * 20, "{0:^19}".format(card.name[:19]), color)
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return keeping
def choose_target(self, targets):
if len(targets) is 0:
return None
renderer.targets = targets
renderer.selected_target = renderer.targets[0]
renderer.draw_game()
self.window.addstr(0, 0, "Choose target")
self.window.refresh()
ch = 0
index = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
index -= 1
if index < 0:
index = len(renderer.targets) - 1
if ch == curses.KEY_RIGHT:
index += 1
if index == len(renderer.targets):
index = 0
renderer.selected_target = renderer.targets[index]
renderer.draw_game()
self.window.refresh()
renderer.targets = None
if ch == 27:
return None
return renderer.selected_target
def choose_index(self, card, player):
renderer.selection_index = 0
renderer.draw_game()
self.window.addstr(0, 0, "Choose placement location")
self.window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
renderer.selection_index -= 1
if renderer.selection_index < 0:
renderer.selection_index = len(player.minions)
if ch == curses.KEY_RIGHT:
renderer.selection_index += 1
if renderer.selection_index > len(player.minions):
renderer.selection_index = 0
renderer.draw_game()
self.window.refresh()
index = renderer.selection_index
renderer.selection_index = -1
if ch == 27:
return -1
return index
def choose_option(self, *options):
self.window.addstr(0, 0, "Choose option")
index = 0
selected = 0
for option in options:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 20, "{0:^19}".format(option.name[:19], color))
index += 1
self.window.refresh()
self.text_window.refresh()
ch = 0
while ch != 10 and ch != 27:
ch = self.window.getch()
if ch == curses.KEY_LEFT:
selected -= 1
if selected < 0:
selected = len(options) - 1
if ch == curses.KEY_RIGHT:
selected += 1
if selected == len(options):
selected = 0
index = 0
for option in options:
if index == selected:
color = curses.color_pair(4)
else:
color = curses.color_pair(3)
self.text_window.addstr(0, index * 20, "{0:^19}".format(option.name[:19], color))
index += 1
self.window.refresh()
self.text_window.refresh()
if ch == 27:
return None
return options[selected]
stdscr.clear()
prompt_window = stdscr.derwin(1, 80, 23, 0)
text_window = stdscr.derwin(1, 80, 24, 0)
deck1 = load_deck(sys.argv[1])
deck2 = load_deck(sys.argv[2])
game = Game([deck1, deck2], [TextAgent(stdscr, prompt_window, text_window), RandomAgent()])
if isinstance(game.players[0].agent, TextAgent):
renderer = GameRender(stdscr, game, game.players[0])
else:
renderer = GameRender(stdscr, game, game.players[1])
game.start()
