no_display_game.py

#!/usr/bin/env python3

from tetris.tetris import Tetris
import pygame, sys
from math import pi, sin
import random

MAXCOLORS = 1
N = 500

def main():
    avg = 0
    for i in range(N):
        p = playGame()
        print(f"{i}: {p}")
        avg += p
    print(f"Average: {avg / N}")
    

from ai.ai import AI
from ai.algorithms.holyNeighborAi import choosePosition as function
def playGame():
    game = Tetris(numColors=MAXCOLORS, numColumns=10, numRows=10)
    ai = AI(function)
    # loop count variables:
    numTicks = 0
    while not game.lost: # game loop ends when game is lost
        ai.ai(game, display=False)
        game.incrementTime()
        numTicks += 1
    return game.numLines


if __name__ == "__main__":
    main()