check_minimax.py

"""
Check that the minmax bot and alpha beta bot return the same judgement, and that alphabeta bot is faster

"""

from api import State, util
import random, time

from bots.alphabeta import alphabeta
from bots.minimax import minimax

REPEATS = 3
MOVES = 5
DEPTH = 5

ab = alphabeta.Bot(randomize=False, depth=DEPTH)
mm = minimax.Bot(randomize=False, depth=DEPTH)

mm_time = 0
ab_time = 0

for r in range(REPEATS):

    # Generate a start state
    state, id = State.generate(6)

    # Do a few random moves
    for m in range(MOVES):
        state = state.next(random.choice(state.moves()))

    # Ask both bots their move
    # (and time their responses)

    start = time.time()
    mm_move = mm.get_move(state)
    mm_time += (time.time() - start)

    start = time.time()
    ab_move = ab.get_move(state)
    ab_time += (time.time() - start)


    if mm_move != ab_move:
        print('Difference of opinion! Minimax said: {}, alphabeta said: {}. State: {}'.format(mm_move, ab_move, state))
    else:
        print('Agreed.')

print('Done. time Minimax: {}, time Alphabeta: {}.'.format(mm_time/REPEATS, ab_time/REPEATS))
print('Alphabeta speedup: {} '.format(mm_time/ab_time))