def test_generate_ai_pos():
gc = GameController(800, 800)
board = Board(800, 800, 100, 20, gc, 0, 1)
human_player = Player(board)
ai_player = Player(board)
gm = GameManager(human_player, ai_player, board, gc)
# since we can not predict which cell ai will make move to,
# so we need to hardcode 7 disks to test the eighth move,
# which should be the positon we predict,
# for it will let ai get the longest flip list
gm.board.disks.create(550, 150, 0)
gm.board.disks.create(450, 150, 0)
gm.board.disks.create(250, 350, 0)
gm.board.disks.create(350, 350, 0)
gm.board.disks.create(450, 350, 1)
gm.board.disks.create(350, 450, 0)
gm.board.disks.create(450, 450, 1)
legal_cells = gm.get_legal_cells(1)
ai_pos = gm.generate_ai_pos(legal_cells)
assert ai_pos == (3, 1)
