def test_allies_in_range_melee_weapon(self): points_to_elevation = { Point(1, 0): -3, Point(0, 1): 4, Point(1, 1): 0, Point(2, 1): -4, Point(1, 2): 3 } tiles = [ Tile(elevation=elevation, point=point) for point, elevation in points_to_elevation.items() ] the_map = Map(3, 3, tiles) origin, below, above = self.a_units far_below = Soldier() far_above = Soldier() self.team_a.add_player(far_above) self.team_a.add_player(far_below) origin.equip_weapon(MeleeWeapon(10, 10, 10, 10)) the_map.place_unit(origin, Point(1, 1)) the_map.place_unit(below, Point(1, 0)) the_map.place_unit(above, Point(1, 2)) the_map.place_unit(far_above, Point(0, 1)) the_map.place_unit(far_below, Point(2, 1)) tf = TargetFinder(the_map, [self.team_a, self.team_b]) answer = tf.allies_in_range(origin) expected = {below: (1, 1), above: (1, -1)} self.assertEqual(answer, expected)
def test_attack_with_different_weapon(self): opponent = Soldier() new_weapon = MeleeWeapon(10, 1) self.soldier.equip_weapon(new_weapon) self.assertEqual(self.soldier.get_weapon().dmg, 10) self.soldier.attack(opponent) self.assertEqual(opponent.get_health(), 90) self.soldier.attack(opponent) self.assertEqual(opponent.get_health(), 80)
def test_attack_out_of_ammo_reloads_and_uses_action_points(self): soldier = Soldier(action_pts=10) weapon = MeleeWeapon(5, 1, 1, 2) soldier.equip_weapon(weapon) for _ in range(2): soldier.attack(self.soldier) self.assertEqual(weapon.ammo, 0) self.assertEqual(self.soldier.get_health(), 90) self.assertEqual(soldier.get_action_points(), 8) soldier.attack(self.soldier) self.assertEqual(weapon.ammo, 2) self.assertEqual(self.soldier.get_health(), 90) self.assertEqual(soldier.get_action_points(), 7)
def test_is_melee_weapon(self): melee = MeleeWeapon(1, 2, 3) ranged = RangedWeapon(2, 3, 4, 5) self.assertTrue(melee.is_melee_weapon()) self.assertFalse(ranged.is_melee_weapon())
def test_equip_weapon(self): stick = MeleeWeapon(3, 2) self.soldier.equip_weapon(stick) self.assertEqual(self.soldier.get_weapon(), stick)
def setUp(self): self.ranged = RangedWeapon(dmg=3, action_pts=4, range_=5, ammo=5) self.melee = MeleeWeapon(dmg=4, action_pts=2)
def test_melee_init_no_defaults(self): new = MeleeWeapon(2, 3, 4, 5) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 4) self.assertEqual(new.ammo, 5)
def test_melee_init_defaults(self): new = MeleeWeapon(2, 3) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 1) self.assertEqual(new.ammo, float('inf'))
class TestWeapon(unittest.TestCase): def setUp(self): self.ranged = RangedWeapon(dmg=3, action_pts=4, range_=5, ammo=5) self.melee = MeleeWeapon(dmg=4, action_pts=2) def test_init_and_getters(self): new = Weapon(dmg=2, action_pts=3, range_=4, ammo=5) self.assertEqual(new.ammo, 5) self.assertEqual(new.max_ammo, 5) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 4) def test_use_weapon_inf_ammo(self): dmg = self.melee.use_weapon() self.assertEqual(dmg, 4) self.assertEqual(self.melee.ammo, float('inf')) def test_use_weapon_non_inf_ammo(self): self.assertEqual(self.ranged.ammo, 5) self.assertEqual(self.ranged.use_weapon(), 3) self.assertEqual(self.ranged.ammo, 4) def test_use_weapon_no_ammo(self): for _ in range(5): self.ranged.use_weapon() self.assertRaises(OutOfAmmo, self.ranged.use_weapon) def test_refill_ammo_inf(self): self.melee.use_weapon() self.melee.refill_ammo() self.assertEqual(self.melee.ammo, float('inf')) def test_refill_ammo_not_inf(self): self.ranged.use_weapon() self.ranged.refill_ammo() self.assertEqual(self.ranged.ammo, 5) for _ in range(5): self.ranged.use_weapon() self.ranged.refill_ammo() self.assertEqual(self.ranged.ammo, 5) def test_melee_init_defaults(self): new = MeleeWeapon(2, 3) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 1) self.assertEqual(new.ammo, float('inf')) def test_melee_init_no_defaults(self): new = MeleeWeapon(2, 3, 4, 5) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 4) self.assertEqual(new.ammo, 5) def test_ranged_init(self): new = RangedWeapon(2, 3, 4, 5) self.assertEqual(new.dmg, 2) self.assertEqual(new.action_pts, 3) self.assertEqual(new.range, 4) self.assertEqual(new.ammo, 5) def test_is_melee_weapon(self): melee = MeleeWeapon(1, 2, 3) ranged = RangedWeapon(2, 3, 4, 5) self.assertTrue(melee.is_melee_weapon()) self.assertFalse(ranged.is_melee_weapon())
from typing import Union from battle.players.strategy import StupidStrategy from battle.statstools.stat import PositiveStat from battle.weapon import MeleeWeapon, RangedWeapon, Weapon, OutOfAmmo FIST = MeleeWeapon(dmg=1, action_pts=1) GUN = RangedWeapon(dmg=5, action_pts=2, range_=5, ammo=10) class Soldier(object): def __init__(self, action_pts: int = 3, health: Union[float, int] = 100, heal_pct: float = 5.0): self._strategy = StupidStrategy() self._action_pts = PositiveStat(action_pts) self._health = PositiveStat(health) self._healing_pct = heal_pct self._weapon = FIST @property def strategy(self): return self._strategy def get_perimeter_size(self) -> int: return self._weapon.range def get_sight_range(self) -> int: