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)
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_influence_by_threat_1(self):
"""
23456789
2..b..
3.....
4.....c
5..a..
6.....aa
7.....
8.....
"""
data = """
turn 1
a 2 4 1
a 5 4 0
a 6 7 0
a 6 8 0
a 4 7 2
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)
self.assertEqual(len(influence_by_threat), 3)
self.assertEqual(influence_by_threat[0][4,4], 2)
self.assertEqual(influence_by_threat[0][6,4], 0)
self.assertEqual(influence_by_threat[0][6,6], 1)
self.assertEqual(influence_by_threat[1][2,4], 1)
self.assertEqual(influence_by_threat[1][2,5], 2)
self.assertEqual(influence_by_threat[2][4,6], 4)
self.assertEqual(influence_by_threat[2][4,7], 3)