def test_simulate_move(ai_player):
player2 = Player('O')
board = Board(ai_player, player2)
board.analyze_move((1,1))
board.switch_current_player()
best_move = ai_player.simulate_move(board, ai_player)
assert (2,2) == best_move
def test_calculate_points_for_human(ai_player):
player2 = Player('O')
board = Board(player2, ai_player)
moves_for_diagonal_victory = [(1,3), (2,4), (3,5)]
for move in moves_for_diagonal_victory:
board.analyze_move(move)
board.switch_current_player()
points = ai_player.calculate_points_for_human(board, player2)
assert points == 18162
def test_calculate_points_diagonal_right_left(ai_player):
player2 = Player('O')
board = Board(ai_player, player2)
moves_for_diagonal_victory = [(9,8), (10,7), (11,6), (12,5), (13,4)]
for move in moves_for_diagonal_victory:
board.analyze_move(move)
board.switch_current_player()
ai_player.calculate_points_diagonal_right_left(board, ai_player)
assert ai_player.sequences[4] == 4 and ai_player.gaps[4] == 7
def test_calculate_points_vertical(ai_player):
player2 = Player('O')
board = Board(ai_player, player2)
moves_for_vertical_victory = [(0,1), (1,1), (2,1), (3,1), (4,1)]
for move in moves_for_vertical_victory:
board.analyze_move(move)
board.switch_current_player()
ai_player.calculate_points_vertical(board, ai_player)
assert ai_player.sequences[4] == 2 and ai_player.gaps[4] == 3
def test_calculate_points_horizontal(ai_player):
player2 = Player('O')
board = Board(ai_player, player2)
moves_for_horizontal_victory = [(1,1), (1,2), (1,3), (1,4), (1,5)]
for move in moves_for_horizontal_victory:
board.analyze_move(move)
board.switch_current_player()
ai_player.calculate_points_horizontal(board, ai_player)
assert ai_player.sequences[4] == 1 and ai_player.gaps[4] == 2
def test_calculate_points_diagonal_left_right_double(ai_player):
ai_player.restart_sequences_and_gaps()
player2 = Player('O')
board = Board(ai_player, player2)
moves_for_diagonal_victory = [(1,3), (2,4), (3,5)]
for move in moves_for_diagonal_victory:
board.analyze_move(move)
board.switch_current_player()
ai_player.calculate_points_diagonal_left_right(board, ai_player)
assert ai_player.sequences[2] == 1 and ai_player.gaps[2] == 2