def test_king_invalid_move():
"""test_king_cant_move_more_than_one_step"""
king_piece = King()
small_board = Board(3, 3)
small_board.put(king_piece, 0, 0)
assert king_piece.takes(small_board, 0, 0, 2, 2) is False
assert king_piece.takes(small_board, 0, 0, 0, 2) is False
def test_king_takes_happy_cases():
"""
The King can move anywhere but only by one step.
This test asserts that the function returns True for all valid cases
"""
king_piece = King()
small_board = Board(3, 3)
small_board.put(king_piece, 1, 1)
assert king_piece.takes(small_board, 1, 1, 0, 0)
assert king_piece.takes(small_board, 1, 1, 0, 1)
assert king_piece.takes(small_board, 1, 1, 0, 2)
assert king_piece.takes(small_board, 1, 1, 1, 0)
assert king_piece.takes(small_board, 1, 1, 1, 1)
assert king_piece.takes(small_board, 1, 1, 1, 2)
assert king_piece.takes(small_board, 1, 1, 2, 0)
assert king_piece.takes(small_board, 1, 1, 2, 0)
assert king_piece.takes(small_board, 1, 1, 2, 1)
assert king_piece.takes(small_board, 1, 1, 2, 2)
