def test_clean_is_ok():
""" test_clean_when_there_is_a_piece_in_the_position """
king = King()
board = Board(3, 3)
board.put(king, 0, 0)
board.clean(0, 0)
assert board.pieces == {}
def test_free_bug():
queen = Queen()
queen_2 = Queen()
board = Board(7, 7)
assert len(board.free_positions()) == 49
board.put(queen, 4, 3)
assert len(board.free_positions()) == 26
board.put(queen, 0, 0)
assert len(board.free_positions()) == 12
board.clean(0, 0)
assert len(board.free_positions()) == 26
def test_clean_frees_place():
"""test_clean_frees_places_in_the_board"""
board = Board(4, 4)
rook = Rook()
board.put(rook, 2, 2)
expected = [(0, 1), (0, 0), (1, 3), (3, 3), (3, 0), (3, 1), (1, 0), (1, 1),
(0, 3)]
assert set(expected) == set(board.free_positions())
expected_after_clean = [(0, 1), (0, 0), (1, 3), (3, 3), (3, 0), (3, 1),
(1, 0), (1, 1), (0, 3), (0, 2), (1, 2), (2, 2),
(3, 2), (2, 0), (2, 1), (2, 2), (2, 3)]
board.clean(2, 2)
assert set(expected_after_clean) == set(board.free_positions())
def test_free_with_pieces():
"""test_free_with_a_king_in_the_board_happy_path """
king = King()
queen = Queen()
board = Board(7, 7)
assert len(board.free_positions()) == 49
board.put(king, 5, 5)
assert len(board.free_positions()) == 40
board.clean(5, 5)
assert len(board.free_positions()) == 49
board.put(queen, 4, 3)
assert len(board.free_positions()) == 26
board.clean(4, 3)
assert len(board.free_positions()) == 49
def test_clean_empty():
""" test_clean_when_in_an_empty_position"""
board = Board(3, 3)
with raises(AssertionError):
board.clean(0, 0)