def test_find_my_combat_ant_2(self):
"""
23456789
2..b..
3.....
4.....
5.....
6.....aaa
7.....
8.....
9.....
0...b..
"""
data = """
turn 1
a 2 4 1
a 6 7 0
a 6 8 0
a 6 9 0
a 10 5 1
go
"""
self.gamestate.update(data)
combat_distance = self.gamestate.euclidean_distance_add(self.gamestate.attackradius2, + 3)
my_combat_ants = battle_influence.bfs_find_my_combat_ants(self.gamestate,
self.gamestate.enemy_ants(), combat_distance)
self.assertEqual(len(my_combat_ants), 2)
self.assertTrue((6,7) in my_combat_ants)
self.assertTrue((6,8) in my_combat_ants)
def test_find_my_combat_ant_1(self):
"""
23456789
2..b..
3.....
4.....
5.a...
6.....aa
7.....
8..a..
"""
data = """
turn 1
a 2 4 1
a 5 3 0
a 6 7 0
a 6 8 0
a 8 4 0
go
"""
self.gamestate.update(data)
combat_distance = self.gamestate.euclidean_distance_add(self.gamestate.attackradius2, + 3)
# print(self.combat_distance)
# print(self.gamestate.euclidean_distance2((7,9),(11,12)))
# print(self.gamestate.euclidean_distance2((7,9),(11,13)))
# print(self.gamestate.euclidean_distance2((7,9),(14,9)))
my_combat_ants = battle_influence.bfs_find_my_combat_ants(self.gamestate,
self.gamestate.enemy_ants(), combat_distance)
self.assertEqual(len(my_combat_ants), 2)
self.assertTrue((5,3) in my_combat_ants)
self.assertTrue((6,7) in my_combat_ants)
