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)
