def test_unit_suicide(self):
jack = Unit('Jack', 100, 120)
jack.attack(jack)
self.assertEqual(jack.name, 'Jack')
self.assertEqual(jack.hit_points, 100)
self.assertEqual(jack.hit_points_limit, 100)
self.assertEqual(jack.damage, 120)
def test_unit_attack(self):
jack = Unit('Jack', 100, 20)
bob = Unit('Bob', 100, 33)
jack.attack(bob)
self.assertEqual(jack.name, 'Jack')
self.assertEqual(jack.hit_points, 83.5)
self.assertEqual(jack.hit_points_limit, 100)
self.assertEqual(jack.damage, 20)
self.assertEqual(bob.name, 'Bob')
self.assertEqual(bob.hit_points, 80)
self.assertEqual(bob.hit_points_limit, 100)
self.assertEqual(bob.damage, 33)
bob.attack(jack)
self.assertEqual(jack.name, 'Jack')
self.assertEqual(jack.hit_points, 50.5)
self.assertEqual(jack.hit_points_limit, 100)
self.assertEqual(jack.damage, 20)
self.assertEqual(bob.name, 'Bob')
self.assertEqual(bob.hit_points, 70)
self.assertEqual(bob.hit_points_limit, 100)
self.assertEqual(bob.damage, 33)
def test_add_hit_points(self):
jack = Unit('Jack', 100, 20)
jack.take_damage(50)
self.assertEqual(jack.hit_points, 50)
jack.add_hit_points(50)
self.assertEqual(jack.hit_points, 100)
jack.take_damage(100)
with self.assertRaises(UnitIsDead):
jack.add_hit_points(100)
def test_attack_dead_unit(self):
jack = Unit('Jack', 100, 20)
bob = Unit('Bob', 100, 33)
jack.take_damage(100)
with self.assertRaises(UnitIsDead):
bob.attack(jack)
def test_unit_create(self):
jack = Unit('Jack', 100, 20)
self.assertEqual(jack.name, 'Jack')
self.assertEqual(jack.hit_points, 100)
self.assertEqual(jack.hit_points_limit, 100)
self.assertEqual(jack.damage, 20)
def init_unit(m, dims, position):
all_unit = []
for j in range(dims):
for i in range(m):
u_no = i*dims+j
cur_pos = {j: position[u_no]}
unit = Unit(u_no=u_no, cur_pos=cur_pos)
all_unit.append(unit)
return all_unit
def init_unit(self):
all_unit = []
for j in range(self.dims):
for i in range(self.m):
u_no = j * self.m + i # notice: the order of dims an m must as same as the dictionary of pisition
cur_pos = set()
cur_pos.add(u_no)
# cur_pos = {i: position_dict[i]}
unit = Unit(u_no=u_no, cur_pos=cur_pos)
all_unit.append(unit)
return all_unit
def join_units(self, dense_unit_list):
candidates = []
unit_no = 0
for u1 in dense_unit_list:
for u2 in dense_unit_list:
new_pos = u1.add_dim(u2, self.position)
if new_pos and self.filter_pos(new_pos, candidates):
candidates.append(Unit(unit_no, cur_pos=new_pos))
unit_no += 1
return candidates
#Test file for functions from src.unit import Unit cols = 9 rows = 6 grid = [[[0, 0] for j in range(cols)] for i in range(rows)] # col 3 row 4 grid[1][7] = [0, 1] my_unit = Unit(1) my_unit.pos = [7, 1] range_list = my_unit.get_move_range(cols, rows) print(range_list)
def setUp(self):
self.test_soldier = Unit(health=200, mana=100)
self.default_weapon = Weapon()
self.default_spell = Spell()
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)
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_create_instance_negative_mana_should_raise_value_error(self):
with self.assertRaises(ValueError):
Unit(health=20, mana=-20)
def test_create_instance_invalid_input_should_type_error_error(self):
with self.assertRaises(TypeError):
Unit(health='asd', mana=100)
