Как вычислить в вложенном массиве нужное нам

def calculate(a_list: list) -> list:
    x, y = 0, 0
    char = 'X'
    sets = []  # наборы координат, чтобы не повторяться
    while True:
        element = a_list[x][y]
        if element == char and all((x, y) not in p for p in sets):
            points = []
            width, height = 1, 1
            for i in range(y + 1, len(a_list[x])):
                if a_list[x][i] != char:
                    break
                else:
                    width += 1
            for i in range(x + 1, len(a_list)):
                if a_list[i][y] != char:
                    break
                else:
                    height += 1
            for x1 in range(x, x + height):
                for y1 in range(y, y + width):
                    points.append((x1, y1))
            sets.append(points)
        y = y + 1
        if y >= len(a_list[x]):
            x = x + 1
            y = 0
        if x >= len(a_list):
            break
    return [len(p) for p in sets]
 
 
if __name__ == '__main__':
    example1 = [['O', 'O', 'X', 'O'], ['O', 'O', 'X', 'O'], ['O', 'O', 'X', 'O'], ['O', 'O', 'O', 'O']]
    example2 = [['O', 'X', 'X', 'O'], ['O', 'X', 'X', 'O'], ['O', 'O', 'O', 'O'], ['O', 'O', 'O', 'O']]
    example3 = [['O', 'X', 'X', 'O'], ['O', 'X', 'X', 'O'], ['O', 'O', 'O', 'O'], ['X', 'O', 'O', 'O']]
    example4 = [['O', 'X', 'O', 'O'], ['O', 'X', 'O', 'O'], ['O', 'O', 'O', 'O'], ['X', 'O', 'O', 'X']]
    example5 = [['O', 'O', 'O', 'O'], ['O', 'O', 'O', 'O'], ['O', 'O', 'O', 'O'], ['O', 'O', 'O', 'O']]
    assert calculate(example1) == [3]
    assert calculate(example2) == [4]
    assert calculate(example3) == [4, 1]
    assert calculate(example4) == [2, 1, 1]
    assert calculate(example5) == []