def test_attacked_positions_king():
piece_type = morpy.KING
dimensions = morpy.Dimensions(x=3, y=3)
coords = morpy.Coordinate(x=1, y=1)
placed_piece = morpy.PlacedPiece(type=piece_type, coordinate=coords, board_dimensions=dimensions)
attacked_coordinates = morpy.get_attacked_coordinates(placed_piece)
expected = {morpy.Coordinate(x, y) for x, y in ((0, 0), (0, 1), (0, 2), (1, 0), (1, 2), (2, 0), (2, 1), (2, 2))}
assert attacked_coordinates == expected
def test_attacked_positions_knight_in_corner():
piece_type = morpy.KNIGHT
dimensions = morpy.Dimensions(x=3, y=3)
coords = morpy.Coordinate(x=0, y=0)
placed_piece = morpy.PlacedPiece(type=piece_type, coordinate=coords, board_dimensions=dimensions)
attacked_coordinates = morpy.get_attacked_coordinates(placed_piece)
expected = {morpy.Coordinate(x, y) for x, y in ((2, 1), (1, 2))}
assert attacked_coordinates == expected
def x_test_attacked_positions_queen_in_corner():
piece_type = morpy.QUEEN
dimensions = morpy.Dimensions(x=3, y=3)
coords = morpy.Coordinate(x=0, y=0)
placed_piece = morpy.PlacedPiece(type=piece_type, coordinate=coords, board_dimensions=dimensions)
attacked_coordinates = morpy.get_attacked_coordinates(placed_piece)
expected = {morpy.Coordinate(x, y) for x, y in ((0, 1), (0, 2), (1, 0), (2, 0), (1, 1), (2, 2))}
assert attacked_coordinates == expected
