def test_get_combat_scores_3(self): 'this test makes sure when we are winning, our ant will press forward' """ 23456789 2b.... 3...a. 4..a.. 5.a... 6..... 7..... 8..... """ data = """ turn 1 a 2 2 1 a 3 5 0 a 4 4 0 a 5 3 0 go """ self.gamestate.update(data) my_ants = self.gamestate.my_ants() enemy_ants = self.gamestate.enemy_ants() threat_distance = self.gamestate.attackradius2 influence_by_threat = battle_influence.get_influence_by_threat(self.gamestate, my_ants, enemy_ants, threat_distance) combat_scores = battle_influence.get_combat_scores(self.gamestate, my_ants, enemy_ants, influence_by_threat, threat_distance) # print (influence_by_threat) # print (combat_scores) self.assertTrue(combat_scores[(3,4)] > combat_scores[(3,5)]) self.assertTrue(combat_scores[(4,3)] > combat_scores[(4,4)]) self.assertTrue(combat_scores[(5,2)] == combat_scores[(5,3)])
def test_get_combat_scores_4(self): """ 23456789 2.b.b.. 3..b.b. 4...... 5...... 6....a. 7...... """ data = """ turn 1 a 2 3 1 a 3 5 1 a 3 4 1 a 3 6 1 a 6 6 0 go """ self.gamestate.update(data) my_ants = self.gamestate.my_ants() enemy_ants = self.gamestate.enemy_ants() threat_distance = self.gamestate.attackradius2 influence_by_owner = battle_influence.get_influence_by_owner(self.gamestate, my_ants, enemy_ants, threat_distance) influence_by_threat = battle_influence.get_influence_by_threat(self.gamestate, my_ants, enemy_ants, threat_distance) combat_scores = battle_influence.get_combat_scores(self.gamestate, my_ants, enemy_ants, influence_by_threat, threat_distance) print("") print(influence_by_owner[0]) print("") print(influence_by_owner[1])
def test_get_combat_scores_2(self): """ 23456789 2..bb. 3..... 4..... 5..... 6..aa. """ data = """ turn 1 a 2 4 1 a 2 5 1 a 6 4 0 a 6 5 0 go """ self.gamestate.update(data) my_ants = self.gamestate.my_ants() enemy_ants = self.gamestate.enemy_ants() threat_distance = self.gamestate.attackradius2 influence_by_threat = battle_influence.get_influence_by_threat(self.gamestate, my_ants, enemy_ants, threat_distance) combat_scores = battle_influence.get_combat_scores(self.gamestate, my_ants, enemy_ants, influence_by_threat, threat_distance) # print (influence_by_threat) # print (combat_scores) self.assertEqual(combat_scores[(5,4)], 2.02) self.assertEqual(combat_scores[(6,4)], 0.99)