class TestUnit(unittest.TestCase):
def setUp(self):
self.test_soldier = Unit(health=200, mana=100)
self.default_weapon = Weapon()
self.default_spell = Spell()
def test_create_instance_invalid_input_should_type_error_error(self):
with self.assertRaises(TypeError):
Unit(health='asd', mana=100)
def test_create_instance_negative_health_should_raise_value_error(self):
with self.assertRaises(ValueError):
Unit(health=-20, mana=20)
def test_create_instance_negative_mana_should_raise_value_error(self):
with self.assertRaises(ValueError):
Unit(health=20, mana=-20)
def test_can_cast(self):
self.setUp()
n = self.test_soldier
self.assertFalse(n.can_cast())
def test_can_cast_not_enough_mana(self):
self.setUp()
n = self.test_soldier
n.learn_spell(Spell('Fireball', damage=50, mana_cost=120, cast_range=3))
self.assertFalse(n.can_cast())
def test_can_cast_all_okay(self):
self.setUp()
self.test_soldier.learn_spell(self.default_spell)
self.assertTrue(self.test_soldier.can_cast())
def test_take_damage_int(self):
self.setUp()
npc = self.test_soldier
npc.take_damage(20)
self.assertEqual(npc.health, 180)
def test_take_damage_float(self):
self.setUp()
npc = self.test_soldier
npc.take_damage(10.5)
self.assertEqual(npc.health, 189.5)
def test_take_damage_more_damage_than_hp_should_return_0(self):
npc = Unit(health=20, mana=50)
npc.take_damage(30)
self.assertEqual(npc.get_health(), 0)
def test_take_damage_should_raise_type_error_invalid_input(self):
self.setUp()
with self.assertRaises(TypeError):
npc = self.test_soldier
npc.take_damage('asd')
def test_take_damage_raises_value_error_negative_value(self):
self.setUp()
with self.assertRaises(ValueError):
npc = self.test_soldier
npc.take_damage(-20)
def test_take_healing_should_not_get_over_max_hp(self):
self.setUp()
npc = self.test_soldier
npc.take_damage(20)
npc.take_healing(30)
self.assertEqual(npc.get_health(), 200)
def test_healing_msg_if_instance_is_dead(self):
self.setUp()
npc = self.test_soldier
npc.health = 0
self.assertEqual(npc.take_healing(20), 'Cannot heal a corpse!')
def test_healing_heals(self):
self.setUp()
n = self.test_soldier
n.take_damage(50)
n.take_healing(20)
self.assertEqual(n.health, 170)
def test_equip_weapon(self):
self.setUp()
npc = self.test_soldier
npc.equip_weapon(self.default_weapon)
self.assertEqual(npc.weapon, self.default_weapon)
def test_learn_spell(self):
self.setUp()
npc = self.test_soldier
npc.learn_spell(self.default_spell)
self.assertEqual(npc.spell, self.default_spell)
def test_equip_weapon_invalid_instance_raises_type_error(self):
self.setUp()
with self.assertRaises(TypeError):
self.test_soldier.equip_weapon('asd')
def test_learn_spell_invalid_instance_raises_type_error(self):
self.setUp()
with self.assertRaises(TypeError):
self.test_soldier.learn_spell('asd')
def test_take_mana_should_not_get_over_max_mana(self):
self.setUp()
npc = self.test_soldier
npc.take_mana(20)
npc.take_mana(30)
self.assertEqual(npc.get_mana(), 100)
def test_take_mana_should_raise_type_error_invalid_input(self):
self.setUp()
npc = self.test_soldier
with self.assertRaises(TypeError):
npc.take_mana('as')
def test_take_mana_should_raise_value_error_negative_value(self):
self.setUp()
n = self.test_soldier
with self.assertRaises(ValueError):
n.take_mana(-20)
def test_get_damage_source_weapon_and_spell(self):
self.setUp()
n = self.test_soldier
n.equip_weapon(self.default_weapon)
n.learn_spell(Spell(name='asd', damage=20, mana_cost=30, cast_range=2))
self.assertEqual(n.get_damage_source(), 'spell')
def test_get_damage_source_weapon_and_spell_wepaon_higher_damage(self):
self.setUp()
n = self.test_soldier
n.equip_weapon(Weapon(name='asd', damage=50))
n.learn_spell(self.default_spell)
self.assertEqual(n.get_damage_source(), 'weapon')
def test_get_damage_source_weapon_only(self):
self.setUp()
n = self.test_soldier
n.equip_weapon(self.default_weapon)
self.assertEqual(n.get_damage_source(), 'weapon')
def test_get_damage_source_spell_only(self):
self.setUp()
n = self.test_soldier
n.learn_spell(self.default_spell)
self.assertEqual(n.get_damage_source(), 'spell')
def test_get_damage_source_no_weapon_spell(self):
self.setUp()
n = self.test_soldier
self.assertEqual(n.get_damage_source(), None)
class unit_test(unittest.TestCase):
def setUp(self):
self.unit = Unit(health=100, mana=100, mana_regeneration_rate=5)
def test_init(self):
self.assertEqual(self.unit.health, 100)
self.assertEqual(self.unit.mana, 100)
self.assertEqual(self.unit._max_health, 100)
self.assertEqual(self.unit._max_mana, 100)
self.assertEqual(self.unit.mana_regeneration, 5)
def test_get_health(self):
self.assertEqual(self.unit.get_health(), 100)
def test_get_mana(self):
self.assertEqual(self.unit.get_mana(), 100)
def test_is_alive(self):
self.unit.health = 0
self.assertFalse(self.unit.is_alive())
def test_take_healing(self):
self.unit.health = 0
self.assertFalse(self.unit.take_healing(30))
self.unit.health = 50
self.unit.take_healing(100)
self.assertEqual(self.unit.health, 100)
self.unit.take_healing(-50)
self.assertEqual(self.unit.get_health(), 100)
def test_take_mana(self):
self.unit.mana = 0
self.unit.take_mana(moved=True)
self.assertEqual(self.unit.mana, 5)
self.unit.take_mana(120)
self.assertEqual(self.unit.mana, 100)
def test_equip_weapon(self):
weapon = Weapon("Weapon name", 20)
self.unit.equip(weapon)
self.assertEqual(self.unit.weapon, weapon)
def test_learn_spell(self):
spell = Spell("Spell name", 100, 50, 3)
self.unit.learn(spell)
self.assertEqual(self.unit.spell, spell)
def test_attack_without_weapon_or_spell(self):
self.assertEqual(self.unit.attack()[1], 0)
def test_attack_with_better_weapon(self):
spell = Spell("S", 10, 20, 5)
weapon = Weapon('W', 50)
self.unit.learn(spell)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack()[1], 50)
def test_attack_with_equal_weapon_and_spell(self):
spell = Spell("S", 50, 50, 5)
weapon = Weapon('W', 50)
self.unit.learn(spell)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack()[1], 50)
self.assertEqual(self.unit.mana, 50)
# def test_cant_attack_with_weapon_from_distance(self):
# weapon = Weapon('W', 50)
# self.unit.equip(weapon)
# # self.assertEqual(self.unit.attack(distance=2), 0)
# self.assertEqual(self.unit.attack()[1], 0)
def test_attack_by_weapon(self):
weapon = Weapon("Weapon name", 20)
self.assertEqual(self.unit.attack(by="weapon")[1], 0)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack(by="weapon")[1], 20)
def test_attack_by_spell(self):
spell = Spell("Spell name", 100, 50, 3)
self.assertEqual(self.unit.attack(by="spell")[1], 0)
self.unit.learn(spell)
# self.assertEqual(self.unit.attack(by="spell"), 100)
def test_take_damage(self):
self.unit.take_damage(20)
self.assertEqual(self.unit.health, 80)
self.unit.take_damage(90)
self.assertEqual(self.unit.health, 0)
def test_can_cast(self):
# self.assertFalse(self.unit.can_cast())
self.unit.learn(Spell('a', 10, 10, 3))
# self.assertTrue(self.unit.can_cast(1))
# self.assertTrue(self.unit.can_cast(3))
# self.assertFalse(self.unit.can_cast(4))
self.assertTrue(self.unit.can_cast())
self.unit.mana = 5
# self.assertFalse(self.unit.can_cast(1))
self.assertFalse(self.unit.can_cast())
