def test_hero_move_with_as_and_not_learned_spell_should_return_false_and_as(
self):
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
enemy = Enemy(health=100, mana=40, damage=60)
enemy.x = 3
enemy.y = 3
fight = Fight(enemy)
status, error = fight.hero_move(h, 'as')
self.assertFalse(status)
self.assertEqual(error, 'as')
def test_if_hero_move_input_is_not_from_our_options_should_return_false_and_other(
self):
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
enemy = Enemy(health=100, mana=40, damage=60)
enemy.x = 3
enemy.y = 3
fight = Fight(enemy)
status, error = fight.hero_move(h, 'test')
self.assertFalse(status)
self.assertEqual(error, 'other')
def test_hero_move_with_as_and_learned_spell_should_return_true_and_as(
self):
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
s = Spell('fireball', 25, 15, 3)
h.learn(s)
enemy = Enemy(health=100, mana=40, damage=60)
enemy.x = 3
enemy.y = 3
fight = Fight(enemy)
status = fight.hero_move(h, 'as')
self.assertTrue(status)
def test_hero_move_with_aw_and_equiped_weapon_should_return_true(self):
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
w = Weapon('nojka', 25)
h.equip(w)
h.x = 2
h.y = 3
enemy = Enemy(health=100, mana=40, damage=60)
enemy.x = 3
enemy.y = 3
fight = Fight(enemy)
status, error = fight.hero_move(h, 'aw')
self.assertTrue(status)
def test_hero_move_with_as_and_learned_spell_but_not_enough_mana_should_return_false_and_not_enough_mana(
self):
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=50,
mana_regeneration_rate=2)
enemy = Enemy(health=100, mana=40, damage=60)
spell = Spell(name='Test Magiq', damage=40, mana_cost=60, cast_range=1)
enemy.x = 3
enemy.y = 3
h.learn(spell)
fight = Fight(enemy)
status, error = fight.hero_move(h, 'as')
self.assertFalse(status)
self.assertEqual(error, 'not enough mana')
def test_hero_move_with_aw_and_equiped_weapon_and_not_in_range_should_return_false_and_not_in_range_2(
self):
# x column is the same, but abs(enemy.y - hero.y) is not equal to 1
h = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
w = Weapon('nojka', 25)
h.equip(w)
h.x = 3
h.y = 1
enemy = Enemy(health=100, mana=40, damage=60)
enemy.x = 3
enemy.y = 3
fight = Fight(enemy)
status, error = fight.hero_move(h, 'aw')
self.assertFalse(status)
self.assertEqual(error, 'not in range')
class FightTests(unittest.TestCase):
def setUp(self):
self.spell = Spell(name="Fireball",
damage=5,
mana_cost=20,
cast_range=1)
self.weapon = Weapon(name="The Axe of Destiny", damage=10)
self.hero = Hero(name="Bron",
title="Dragonslayer",
health=100,
mana=100,
mana_regeneration_rate=2)
self.map = [['.', '.', '#', '#', '.', 'S', '.', '.', '.', 'T'],
['#', 'T', '#', '#', '.', '.', '#', '#', '#', '.'],
['#', '.', '#', '#', '#', 'E', '#', '#', '#', 'E'],
['#', '.', 'E', '.', '.', '.', '#', '#', '#', '.'],
['#', '#', '#', 'T', '#', 'T', '#', '#', '#', 'G']]
self.enemies = {
"2,5": {
"damage": 10,
"health": 40,
"mana": 20,
"weapon": None,
"spell": self.spell.to_json()
},
"2,9": {
"damage": 10,
"health": 40,
"mana": 20,
"weapon": self.weapon.to_json(),
"spell": None
},
"3,2": {
"damage": 10,
"health": 40,
"mana": 20,
"weapon": None,
"spell": None
}
}
self.treasures = {
"1,1": {
"class": "Potion",
"type": "mana",
"amount": 50
},
"0,9": {
"class": "Potion",
"type": "health",
"amount": 50
},
"4,3": {
"class": "Weapon",
"name": "The Axe of Destiny",
"damage": 10
},
"4,5": {
'class': "Spell",
'name': 'Fireball',
'damage': 5,
'mana_cost': 20,
'cast_range': 1
}
}
self.json_data = {
"map": self.map,
"enemies": self.enemies,
"treasures": self.treasures
}
self.dungeon = Dungeon(self.json_data)
self.dungeon.spawn(self.hero)
self.fight = Fight(dungeon=self.dungeon, enemy_pos=(2, 5))
def test_init_initializing_correctly(self):
self.assertEqual(self.fight.dungeon, self.dungeon)
self.assertEqual(self.fight.enemy,
Enemy.from_json(self.enemies['2,5']))
self.assertEqual(self.fight.enemy_pos, (2, 5))
def test_init_raises_Assertion_error_when_there_is_no_enemy_on_the_given_pos(
self):
with self.assertRaises(AssertionError):
Fight(dungeon=self.dungeon, enemy_pos=(0, 1))
def test__range_between(self):
self.assertEqual(self.fight._range_between(), 2)
def test__direction_to_enemy_without_arguments(self):
with self.subTest('Hero is up from enemy'):
self.assertEqual(self.fight._direction_to_enemy(), (1, 0))
with self.subTest('Hero is down from enemy'):
self.fight.dungeon._hero_pos = (3, 5)
self.assertEqual(self.fight._direction_to_enemy(), (-1, 0))
def test__direction_to_enemy_with_argument(self):
self.fight.enemy_pos = (3, 2)
with self.subTest('Hero is left from enemy'):
self.fight.dungeon._hero_pos = (3, 1)
self.assertEqual(self.fight._direction_to_enemy(False), (0, 1))
with self.subTest('Hero is right from enemy'):
self.fight.dungeon._hero_pos = (3, 5)
self.assertEqual(self.fight._direction_to_enemy(False), (0, -3))
def test_hero_move_moves_hero_on_the_map_towards_enemy(self):
self.fight.hero_move()
self.assertEqual(self.fight.dungeon._map[1][5], self.hero)
def test_enemy_move_moves_enemy_on_the_map_towards_hero(self):
self.fight.enemy_move()
self.assertEqual(self.fight.dungeon._map[1][5], self.fight.enemy)
self.assertEqual(self.fight.enemy_pos, (1, 5))
def test_player_makes_move_one_of_the_players_move_when_hero_has_to_move(
self):
self.fight.player_makes_move(self.dungeon._hero)
self.assertEqual(self.fight.dungeon._map[1][5], self.hero)
def test_player_makes_move_one_of_the_players_move_when_enemy_has_to_move(
self):
self.fight.player_makes_move(self.fight.enemy)
self.assertEqual(self.fight.dungeon._map[1][5], self.fight.enemy)
def test_player_makes_move_hero_attacks(self):
self.fight.dungeon._hero.learn(self.spell)
self.fight.enemy_move()
self.fight.player_makes_move(self.fight.dungeon._hero)
self.assertEqual(self.fight.enemy.get_health(), 35)
def test_player_makes_move_enemy_attacks(self):
self.fight.hero_move()
self.fight.player_makes_move(self.fight.enemy)
self.assertEqual(self.fight.dungeon._hero.get_health(), 85)
def test_fight(self):
spell = Spell(name="Fireball", damage=20, mana_cost=4, cast_range=1)
self.fight.dungeon._hero.learn(spell)
print()
self.fight.dungeon.print_map()
self.fight.fight()
self.assertEqual(self.fight.dungeon._hero.get_health(), 85)
self.assertEqual(self.fight.dungeon._hero_pos, (1, 5))
self.assertFalse(self.fight.enemy.is_alive())
def test_fight_hero_is_dead(self):
self.hero._health = 10
with self.assertRaises(HeroIsDeadError):
self.fight.fight()
