def test_ai_turn(): """Test ai_turn method""" tile = Tile() tile.tile[3][5] = 1 tile.counter = 2 tile.ai_turn() assert tile.tile[3][3] == 2 assert tile.tile[3][4] == 2 assert tile.tile[3][5] == 2 assert tile.tile[3][6] == 2
def test_ai_make_move(): """Test ai_make_move method. Make sure the move has been made and tiles in between have been flipped""" tile = Tile() tile.tile = [[0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 2, 1, 0, 0, 0], [0, 0, 1, 1, 2, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0]] tile.counter = 2 computer = Computer(tile) lm = computer.all_legal_moves() computer.ai_make_move() assert (tile.tile[4][1] == 2) and (tile.tile[4][2] == 2) and (tile.tile[4][3] == 2)