def test_attacks_vulnerable_hero(self):
"""Test that the bot attacks a vulnerable hero"""
# Build models
status_dict = get_status_samples_dict()["attack0"][500]
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
hero_id = status_dict["hero"]["id"] - 1
bot = MaxnBot(hero_id)
action = bot.think(status)
self.assertEqual(action, Action.east)
def test_attacks_mine(self):
"""Test that the bot attacks a mine"""
# Build models
status_dict = get_status_samples_dict()["mine0"][0]
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
hero_id = status_dict["hero"]["id"] - 1
bot = MaxnBot(hero_id)
action = bot.think(status)
self.assertEqual(action, Action.north)
