test_against_others.py

from pokerbots.engine.game import Table
from pokerbots.player.pokerbot import Pokerbot
import time
import sys

test_opponent_names = [
    'masterchefA',
    'masterchefB',
    'Template'
]

# okay, we should create an array of bots and play them against each other
# generate all the combinations we want to play
# then round robin them

for p1_name in test_opponent_names:
    p2_name = sys.argv[1]
    p1_wins = 0
    p2_wins = 0
    p1 = Pokerbot(p1_name)
    p2 = Pokerbot(p2_name)

    num_matches = 100

    for i in range(num_matches):
        t = Table(p1, p2)
        t.play_match()
        # To find the winner, read the state. Who has 800 chips?
        # TODO: We assume no timeout!
        state = t.state()
        if state['players'][0]['stack'] == 800:
            p1_wins += 1
        else:
            p2_wins += 1
        if i > 0 and (i + 1) % 25 == 0:
            print "So far, of %s matches, %s won %s and %s won %s" % (i + 1, p1_name, p1_wins, p2_name, p2_wins,)

    print "Out of %s matches, %s won %s and %s won %s" % (num_matches, p1_name, p1_wins, p2_name, p2_wins,)