def test_fireball(self): battle_abilities.FIREBALL().use(self.messenger, self.attacker, self.defender) self.assertTrue(self.defender.health < self.defender.max_health) self.assertFalse(self.attacker.context.damage_queue_fire) self.assertTrue(self.defender.context.damage_queue_fire) self.assertEqual(self.messenger.messages, ['hero_ability_fireball'])
class BattleAbilityFireball(BaseBattleAbility): TYPE = relations.EFFECT.BATTLE_ABILITY_FIREBALL ABILITY = ability = battle_abilities.FIREBALL(5)