def test_used_king_ok():
"""test_positions_used_from_for_king_in_the_upper_corner_are_valid"""
king_piece = King()
small_board = Board(3, 3)
assert (0, 0) in king_piece.positions_to_take(small_board, 0, 0)
assert (1, 0) in king_piece.positions_to_take(small_board, 0, 0)
assert (1, 1) in king_piece.positions_to_take(small_board, 0, 0)
assert (0, 1) in king_piece.positions_to_take(small_board, 0, 0)
assert (0, 2) not in king_piece.positions_to_take(small_board, 0, 0)
def test_king_takes_rook():
"""test_of_a_bug_it_was_possible_to_put_a_king_that_takes_a_rook"""
king = King()
board = Board(3, 3)
board.put(king, 0, 0)
positions_to_take = king.positions_to_take(board, 0, 0)
assert (0, 1) in positions_to_take