import random
import prisoner_game as pg

random.seed(1234)

strategies = ['A', 'B', 'C', 'D']
num_iterations = 300
num_runs = 500

def play_all():
    for i in range(len(strategies)):
        total = 0
        for j in range(len(strategies)):
            if i != j:
                (avg_i, avg_j) = pg.evaluate(num_iterations, num_runs, strategies[i], strategies[j])
                total += avg_i
        avg = total / 4
        s = 'Average for strategy {0}: {1:2.2f}\n'
        print(s.format(strategies[i], avg))

def play_two(strategy0, strategy1):
    pg.play_iterations(num_iterations, strategy0, strategy1, True)

play_all()