import pygame
from game.game_logic.game import Game
import matplotlib.pyplot as plt


def main():
    scores_history = []
    GAME_COUNT = 2
    for i in range(GAME_COUNT):
        game = Game(400, "Snake AI")
        score = game.start()
        scores_history.append(score)
        print("Game:", i)

    plt.ylim(0, 36)
    plt.plot(range(len(scores_history)), scores_history)
    plt.ylabel('Snake length')
    plt.xlabel('Game count')
    plt.show()


if __name__ == "__main__":
    main()

[((15, 4, 15, 19), 'matplotlib.pyplot.ylim', 'plt.ylim', ({(15, 13, 15, 14): '(0)', (15, 16, 15, 18): '(36)'}, {}), '(0, 36)', True, 'import matplotlib.pyplot as plt\n'), ((17, 4, 17, 30), 'matplotlib.pyplot.ylabel', 'plt.ylabel', ({(17, 15, 17, 29): '"""Snake length"""'}, {}), "('Snake length')", True, 'import matplotlib.pyplot as plt\n'), ((18, 4, 18, 28), 'matplotlib.pyplot.xlabel', 'plt.xlabel', ({(18, 15, 18, 27): '"""Game count"""'}, {}), "('Game count')", True, 'import matplotlib.pyplot as plt\n'), ((19, 4, 19, 14), 'matplotlib.pyplot.show', 'plt.show', ({}, {}), '()', True, 'import matplotlib.pyplot as plt\n'), ((10, 15, 10, 36), 'game.game_logic.game.Game', 'Game', ({(10, 20, 10, 23): '400', (10, 25, 10, 35): '"""Snake AI"""'}, {}), "(400, 'Snake AI')", False, 'from game.game_logic.game import Game\n')]