35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
|
from positions import positions
|
||
|
|
from tron.game import Game, PositionPlayer
|
||
|
|
|
||
|
|
def run_battle(ai1, ai2, width, height):
|
||
|
|
games = 50
|
||
|
|
ai1_victories = 0
|
||
|
|
ai2_victories = 0
|
||
|
|
|
||
|
|
for (initial_position_one, initial_position_two) in positions():
|
||
|
|
game = Game(width, height, [
|
||
|
|
PositionPlayer(1, ai1.constructor(), initial_position_one),
|
||
|
|
PositionPlayer(2, ai2.constructor(), initial_position_two),
|
||
|
|
])
|
||
|
|
game.main_loop()
|
||
|
|
|
||
|
|
if game.winner == 1:
|
||
|
|
ai1_victories += 1
|
||
|
|
elif game.winner == 2:
|
||
|
|
ai2_victories += 1
|
||
|
|
|
||
|
|
# Inverse positions and replay to be symmetrical
|
||
|
|
game = Game(width, height, [
|
||
|
|
PositionPlayer(1, ai2.constructor(), initial_position_one),
|
||
|
|
PositionPlayer(2, ai1.constructor(), initial_position_two),
|
||
|
|
])
|
||
|
|
game.main_loop()
|
||
|
|
|
||
|
|
if game.winner == 1:
|
||
|
|
ai2_victories += 1
|
||
|
|
elif game.winner == 2:
|
||
|
|
ai1_victories += 1
|
||
|
|
|
||
|
|
return (ai1_victories, ai2_victories, 2 * games - ai1_victories - ai2_victories)
|
||
|
|
|