def test_screep_does_not_move_when_dead():
s1 = Screep(REDTEAM, 4, move=1, move_func=screep.maint_dist_to_target)
s2 = Screep(BLUETEAM, 0)
s1.set_target([s2])
s1.move([s2])
assert s1.posn == 3
s1.hp = 0
s1.move([s2])
assert s1.posn == 3
def test_screep_does_not_attack_when_dead():
s1 = Screep(REDTEAM, 1, range_att=1)
s2 = Screep(BLUETEAM, 0)
start_hp = s2.hp
s1.set_target([s2])
s1.attack([s2])
assert s2.hp < start_hp
start_hp = s2.hp
s1.hp = 0
s1.attack([s2])
assert s2.hp == start_hp
def test_screep_ranged_attacks_correct_dmg():
s1 = Screep(REDTEAM, 3, move=1, range_att=1)
s2 = Screep(BLUETEAM, 0)
start_hp = s2.hp
s1.set_target([s2])
s1.attack([s2])
assert s2.hp == start_hp - s1.range_att_mods * RANGED_ATT_DMG
def test_screep_melee_attacks_correct_dmg():
s1 = Screep(REDTEAM, 1, move=1, melee_att=1)
s2 = Screep(BLUETEAM, 0)
start_hp = s2.hp
s1.set_target([s2])
s1.attack([s2])
assert s2.hp == start_hp - s1.melee_att_mods * MELEE_ATT_DMG
def test_closest_to_target_selector_targets_closest_with_one_en():
melee_screep = Screep(REDTEAM, 10, move=2, melee_att=4)
ranged_screep = Screep(BLUETEAM, 0, move=1, range_att=2)
screeps = [melee_screep, ranged_screep]
melee_screep.set_target(screeps)
ranged_screep.set_target(screeps)
assert melee_screep.target == ranged_screep.uuid
assert ranged_screep.target == melee_screep.uuid
def test_maint_dist_moves_multi_screeps_complex_posns():
s1 = Screep(REDTEAM, 0, move=1, move_func=screep.maint_dist_to_target)
s2 = Screep(REDTEAM, -10, move=2, move_func=screep.maint_dist_to_target)
s3 = Screep(BLUETEAM, 20, move=1, move_func=screep.maint_dist_to_target)
s4 = Screep(BLUETEAM, -40, move=.5, move_func=screep.maint_dist_to_target)
screeps = [s1, s2, s3, s4]
for s in screeps:
s.set_target(screeps)
s.move(screeps)
assert s1.posn == 1
assert s2.posn == -8
assert s3.posn == 19
assert s4.posn == -40
for s in screeps:
s.move(screeps)
assert s1.posn == 2
assert s2.posn == -6
assert s3.posn == 18
assert s4.posn == -39
def test_determine_one_on_one_winner():
# We want to determine who would win in a one on one fight
# We make two screeps. A slow melee screep and a slow ranged screep, with closest screep as a target (default)
melee_screep = Screep(REDTEAM, 10, move=1, melee_att=2, move_func=screep.maint_dist_to_target)
ranged_screep = Screep(BLUETEAM, 0, move=1, range_att=2, move_func=lambda self, screeps:screep.maint_dist_to_target(self, screeps, 3))
# We run the sim and see the ranged screep wins (by kiting)
sim = CombatSim()
sim.add_screep(melee_screep)
sim.add_screep(ranged_screep)
# We get the results
results = sim.simulate()
# Assert first screep is alive, second is dead
assert results[-1][0].hp <= 0
assert results[-1][1].hp > 0
# We get a graph of posn, dmg, and hp over time
def test_screep_inits_with_proper_values():
s = Screep(REDTEAM, 10, move=5, melee_att=1)
assert s.team == REDTEAM
assert s.posn == 10
assert s.move_mods == 5
assert s.speed == 5
def test_maint_dist_to_target_moves_closer_with_pos_numbers():
s1 = Screep(REDTEAM, -1,move=1, move_func=screep.maint_dist_to_target)
s2 = Screep(BLUETEAM, -10)
s1.set_target([s1,s2])
s1.move([s1, s2])
assert s1.posn == -2
def test_screep_move_updates_posn():
screep = Screep(0, REDTEAM, move = 1, move_func=lambda self, screeps: self.posn + self.speed)
screep.move([screep])
assert screep.posn == 1
screep.move([screep])
assert screep.posn == 2
def test_screep_remains_same_posn_with_default_move_func():
screep = Screep(0, REDTEAM, move = 2)
screep.move([screep])
assert screep.posn == 0
def test_closest_to_target_selector_targets_closest_with_mult_en():
en_melee_screep = Screep(REDTEAM, 10, move=2, melee_att=4)
en_ranged_screep = Screep(REDTEAM, 11, move=1, range_att=2)
ranged_screep = Screep(BLUETEAM, 0, move=1, range_att=2)
screeps = [en_melee_screep, en_ranged_screep, ranged_screep]
en_melee_screep.set_target(screeps)
en_ranged_screep.set_target(screeps)
ranged_screep.set_target(screeps)
assert en_melee_screep.target == ranged_screep.uuid
assert en_ranged_screep.target == ranged_screep.uuid
assert ranged_screep.target == en_melee_screep.uuid
def test_screep_eq_uses_uuid():
s1 = Screep(REDTEAM, 0)
s2 = Screep(REDTEAM, 0)
assert s1 == s1
assert s1 != s2
def test_screeps_init_with_uuid():
s1 = Screep(REDTEAM, 0)
s2 = Screep(REDTEAM, 0)
assert s1.uuid
assert s1.uuid != s2.uuid
def test_screep_ranged_attacks_only_when_in_range():
s1 = Screep(REDTEAM, 4, move=1, range_att=1, move_func=screep.maint_dist_to_target)
s2 = Screep(BLUETEAM, 0)
start_hp = s2.hp
s1.set_target([s2])
s1.attack([s2])
assert s2.hp == start_hp
s1.move([s2])
s1.attack([s2])
assert s2.hp == start_hp - s1.range_att_mods * RANGED_ATT_DMG
def test_screep_maints_distance_from_approching_target():
s1 = Screep(REDTEAM, 8, move=2, move_func=lambda self, screeps:screep.maint_dist_to_target(self, screeps, 6))
s2 = Screep(BLUETEAM, 0, move=1, move_func=screep.maint_dist_to_target)
screeps = [s1, s2]
s1.set_target(screeps)
s2.set_target(screeps)
s1.move(screeps)
s2.move(screeps)
# s1 move closer to get into range
assert s1.posn == 6
assert s2.posn == 1
s1.move(screeps)
s2.move(screeps)
# s1 move 1 farther, to keep range
assert s1.posn == 7
assert s2.posn == 2
s1.move(screeps)
s2.move(screeps)
# s1 move 1 farther, to keep range
assert s1.posn == 8
assert s2.posn == 3
