# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE

from __future__ import annotations

import numpy as np
import pytest  # noqa: F401

import awkward as ak

to_list = ak.operations.to_list


def test_rpad_and_clip_empty_array():
    empty = ak.contents.emptyarray.EmptyArray()
    assert to_list(empty) == []
    assert to_list(ak._do.pad_none(empty, 5, 0)) == [None, None, None, None, None]
    assert (
        ak._do.pad_none(empty.to_typetracer(), 5, 0).form
        == ak._do.pad_none(empty, 5, 0).form
    )
    assert to_list(ak._do.pad_none(empty, 5, 0, clip=True)) == [
        None,
        None,
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(empty.to_typetracer(), 5, 0, clip=True).form
        == ak._do.pad_none(empty, 5, 0, clip=True).form
    )


def test_rpad_and_clip_numpy_array():
    array = ak.contents.numpyarray.NumpyArray(
        np.arange(2 * 3 * 5, dtype=np.int64).reshape(2, 3, 5)
    )
    assert to_list(array) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]

    assert to_list(ak._do.pad_none(array, 5, 0, clip=True)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0, clip=True).form
        == ak._do.pad_none(array, 5, 0, clip=True).form
    )

    assert to_list(ak._do.pad_none(array, 5, 1, clip=True)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], None, None],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29], None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 1, clip=True).form
        == ak._do.pad_none(array, 5, 1, clip=True).form
    )

    assert to_list(ak._do.pad_none(array, 7, 2, clip=True)) == [
        [
            [0, 1, 2, 3, 4, None, None],
            [5, 6, 7, 8, 9, None, None],
            [10, 11, 12, 13, 14, None, None],
        ],
        [
            [15, 16, 17, 18, 19, None, None],
            [20, 21, 22, 23, 24, None, None],
            [25, 26, 27, 28, 29, None, None],
        ],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 7, 2, clip=True).form
        == ak._do.pad_none(array, 7, 2, clip=True).form
    )

    assert to_list(ak._do.pad_none(array, 2, 2, clip=True)) == [
        [[0, 1], [5, 6], [10, 11]],
        [[15, 16], [20, 21], [25, 26]],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 2, clip=True).form
        == ak._do.pad_none(array, 2, 2, clip=True).form
    )


def test_rpad_numpy_array():
    array = ak.contents.numpyarray.NumpyArray(np.array([1.1, 2.2, 3.3, 4.4, 5.5]))
    assert to_list(ak._do.pad_none(array, 10, 0)) == [
        1.1,
        2.2,
        3.3,
        4.4,
        5.5,
        None,
        None,
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 10, 0).form
        == ak._do.pad_none(array, 10, 0).form
    )

    array = ak.contents.numpyarray.NumpyArray(
        np.array([[1.1, 2.2, 3.3], [4.4, 5.5, 6.6]])
    )
    assert to_list(ak._do.pad_none(array, 5, 0)) == [
        [1.1, 2.2, 3.3],
        [4.4, 5.5, 6.6],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0).form
        == ak._do.pad_none(array, 5, 0).form
    )
    assert to_list(ak._do.pad_none(array, 5, 1)) == [
        [1.1, 2.2, 3.3, None, None],
        [4.4, 5.5, 6.6, None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 1).form
        == ak._do.pad_none(array, 5, 1).form
    )

    array = ak.contents.numpyarray.NumpyArray(
        np.arange(2 * 3 * 5, dtype=np.int64).reshape(2, 3, 5)
    )
    assert to_list(array) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]

    assert to_list(ak._do.pad_none(array, 1, 0)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]

    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 0).form
        == ak._do.pad_none(array, 1, 0).form
    )
    assert to_list(ak._do.pad_none(array, 2, 0)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 0).form
        == ak._do.pad_none(array, 2, 0).form
    )
    assert to_list(ak._do.pad_none(array, 3, 0)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 0).form
        == ak._do.pad_none(array, 3, 0).form
    )
    assert to_list(ak._do.pad_none(array, 4, 0)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 0).form
        == ak._do.pad_none(array, 4, 0).form
    )
    assert to_list(ak._do.pad_none(array, 5, 0)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0).form
        == ak._do.pad_none(array, 5, 0).form
    )

    assert to_list(ak._do.pad_none(array, 2, 1)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]
    assert to_list(ak._do.pad_none(array, 3, 1)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 1).form
        == ak._do.pad_none(array, 3, 1).form
    )
    assert to_list(ak._do.pad_none(array, 4, 1)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], None],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29], None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 1).form
        == ak._do.pad_none(array, 4, 1).form
    )
    assert to_list(ak._do.pad_none(array, 5, 1)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], None, None],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29], None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 1).form
        == ak._do.pad_none(array, 5, 1).form
    )

    assert to_list(ak._do.pad_none(array, 3, 2)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 2).form
        == ak._do.pad_none(array, 3, 2).form
    )

    assert to_list(ak._do.pad_none(array, 7, 2)) == [
        [
            [0, 1, 2, 3, 4, None, None],
            [5, 6, 7, 8, 9, None, None],
            [10, 11, 12, 13, 14, None, None],
        ],
        [
            [15, 16, 17, 18, 19, None, None],
            [20, 21, 22, 23, 24, None, None],
            [25, 26, 27, 28, 29, None, None],
        ],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 7, 2).form
        == ak._do.pad_none(array, 7, 2).form
    )

    assert to_list(ak._do.pad_none(array, 2, 2)) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 2).form
        == ak._do.pad_none(array, 2, 2).form
    )


def test_rpad_and_clip_regular_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array(
            [
                2.1,
                8.4,
                7.4,
                1.6,
                2.2,
                3.4,
                6.2,
                5.4,
                1.5,
                3.9,
                3.8,
                3.0,
                8.5,
                6.9,
                4.3,
                3.6,
                6.7,
                1.8,
                3.2,
            ]
        )
    )
    index = ak.index.Index64(
        np.array([13, 9, 13, 4, 8, 3, 15, -1, 16, 2, 8], dtype=np.int64)
    )
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    array = ak.contents.regulararray.RegularArray(indexedarray, 3, zeros_length=0)

    assert to_list(ak._do.pad_none(array, 5, 0, clip=True)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0, clip=True).form
        == ak._do.pad_none(array, 5, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 4, 0, clip=True)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 0, clip=True).form
        == ak._do.pad_none(array, 4, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 3, 0, clip=True)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 0, clip=True).form
        == ak._do.pad_none(array, 3, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 2, 0, clip=True)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 0, clip=True).form
        == ak._do.pad_none(array, 2, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 1, 0, clip=True)) == [[6.9, 3.9, 6.9]]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(array, 1, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 5, 1, clip=True)) == [
        [6.9, 3.9, 6.9, None, None],
        [2.2, 1.5, 1.6, None, None],
        [3.6, None, 6.7, None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 1, clip=True).form
        == ak._do.pad_none(array, 5, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 4, 1, clip=True)) == [
        [6.9, 3.9, 6.9, None],
        [2.2, 1.5, 1.6, None],
        [3.6, None, 6.7, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 1, clip=True).form
        == ak._do.pad_none(array, 4, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 3, 1, clip=True)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 1, clip=True).form
        == ak._do.pad_none(array, 3, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 2, 1, clip=True)) == [
        [6.9, 3.9],
        [2.2, 1.5],
        [3.6, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 1, clip=True).form
        == ak._do.pad_none(array, 2, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(array, 1, 1, clip=True)) == [[6.9], [2.2], [3.6]]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(array, 1, 1, clip=True).form
    )

    array = ak.contents.numpyarray.NumpyArray(np.arange(2 * 3 * 5).reshape(2, 3, 5))
    assert to_list(array) == [
        [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]],
        [[15, 16, 17, 18, 19], [20, 21, 22, 23, 24], [25, 26, 27, 28, 29]],
    ]

    assert to_list(ak._do.pad_none(array, 7, 2, clip=True)) == [
        [
            [0, 1, 2, 3, 4, None, None],
            [5, 6, 7, 8, 9, None, None],
            [10, 11, 12, 13, 14, None, None],
        ],
        [
            [15, 16, 17, 18, 19, None, None],
            [20, 21, 22, 23, 24, None, None],
            [25, 26, 27, 28, 29, None, None],
        ],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 7, 2, clip=True).form
        == ak._do.pad_none(array, 7, 2, clip=True).form
    )

    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10, 10]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    regulararray = ak.contents.regulararray.RegularArray(
        listoffsetarray, 2, zeros_length=0
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 0, clip=True)) == [
        [[0.0, 1.1, 2.2], []]
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(regulararray, 1, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 2, 0, clip=True)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 2, 0, clip=True).form
        == ak._do.pad_none(regulararray, 2, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 0, clip=True)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 0, clip=True).form
        == ak._do.pad_none(regulararray, 3, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 4, 0, clip=True)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
        None,
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 4, 0, clip=True).form
        == ak._do.pad_none(regulararray, 4, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 5, 0, clip=True)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 5, 0, clip=True).form
        == ak._do.pad_none(regulararray, 5, 0, clip=True).form
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 1, clip=True)) == [
        [[0.0, 1.1, 2.2]],
        [[3.3, 4.4]],
        [[6.6, 7.7, 8.8, 9.9]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(regulararray, 1, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 2, 1, clip=True)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 2, 1, clip=True).form
        == ak._do.pad_none(regulararray, 2, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 1, clip=True)) == [
        [[0.0, 1.1, 2.2], [], None],
        [[3.3, 4.4], [5.5], None],
        [[6.6, 7.7, 8.8, 9.9], [], None],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 1, clip=True).form
        == ak._do.pad_none(regulararray, 3, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 7, 1, clip=True)) == [
        [[0.0, 1.1, 2.2], [], None, None, None, None, None],
        [[3.3, 4.4], [5.5], None, None, None, None, None],
        [[6.6, 7.7, 8.8, 9.9], [], None, None, None, None, None],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 7, 1, clip=True).form
        == ak._do.pad_none(regulararray, 7, 1, clip=True).form
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 2, clip=True)) == [
        [[0.0], [None]],
        [[3.3], [5.5]],
        [[6.6], [None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 2, clip=True).form
        == ak._do.pad_none(regulararray, 1, 2, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 2, 2, clip=True)) == [
        [[0.0, 1.1], [None, None]],
        [[3.3, 4.4], [5.5, None]],
        [[6.6, 7.7], [None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 2, 2, clip=True).form
        == ak._do.pad_none(regulararray, 2, 2, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 2, clip=True)) == [
        [[0.0, 1.1, 2.2], [None, None, None]],
        [[3.3, 4.4, None], [5.5, None, None]],
        [[6.6, 7.7, 8.8], [None, None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 2, clip=True).form
        == ak._do.pad_none(regulararray, 3, 2, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 4, 2, clip=True)) == [
        [[0.0, 1.1, 2.2, None], [None, None, None, None]],
        [[3.3, 4.4, None, None], [5.5, None, None, None]],
        [[6.6, 7.7, 8.8, 9.9], [None, None, None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 4, 2, clip=True).form
        == ak._do.pad_none(regulararray, 4, 2, clip=True).form
    )
    assert to_list(ak._do.pad_none(regulararray, 5, 2, clip=True)) == [
        [[0.0, 1.1, 2.2, None, None], [None, None, None, None, None]],
        [[3.3, 4.4, None, None, None], [5.5, None, None, None, None]],
        [[6.6, 7.7, 8.8, 9.9, None], [None, None, None, None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 5, 2, clip=True).form
        == ak._do.pad_none(regulararray, 5, 2, clip=True).form
    )


def test_rpad_regular_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array(
            [
                2.1,
                8.4,
                7.4,
                1.6,
                2.2,
                3.4,
                6.2,
                5.4,
                1.5,
                3.9,
                3.8,
                3.0,
                8.5,
                6.9,
                4.3,
                3.6,
                6.7,
                1.8,
                3.2,
            ]
        )
    )
    index = ak.index.Index64(
        np.array([13, 9, 13, 4, 8, 3, 15, -1, 16, 2, 8], dtype=np.int64)
    )
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    array = ak.contents.regulararray.RegularArray(indexedarray, 3, zeros_length=0)

    assert to_list(ak._do.pad_none(array, 5, 0)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0).form
        == ak._do.pad_none(array, 5, 0).form
    )
    assert to_list(ak._do.pad_none(array, 4, 0)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 0).form
        == ak._do.pad_none(array, 4, 0).form
    )
    assert to_list(ak._do.pad_none(array, 3, 0)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 0).form
        == ak._do.pad_none(array, 3, 0).form
    )
    assert to_list(ak._do.pad_none(array, 1, 0)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 0).form
        == ak._do.pad_none(array, 1, 0).form
    )
    assert to_list(ak._do.pad_none(array, 5, 1)) == [
        [6.9, 3.9, 6.9, None, None],
        [2.2, 1.5, 1.6, None, None],
        [3.6, None, 6.7, None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 1).form
        == ak._do.pad_none(array, 5, 1).form
    )
    assert to_list(ak._do.pad_none(array, 4, 1)) == [
        [6.9, 3.9, 6.9, None],
        [2.2, 1.5, 1.6, None],
        [3.6, None, 6.7, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 4, 1).form
        == ak._do.pad_none(array, 4, 1).form
    )
    assert to_list(ak._do.pad_none(array, 3, 1)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 3, 1).form
        == ak._do.pad_none(array, 3, 1).form
    )
    assert to_list(ak._do.pad_none(array, 1, 1)) == [
        [6.9, 3.9, 6.9],
        [2.2, 1.5, 1.6],
        [3.6, None, 6.7],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 1).form
        == ak._do.pad_none(array, 1, 1).form
    )

    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10, 10]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    regulararray = ak.contents.regulararray.RegularArray(
        listoffsetarray, 2, zeros_length=0
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 0)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 0).form
        == ak._do.pad_none(regulararray, 1, 0).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 0)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 0).form
        == ak._do.pad_none(regulararray, 3, 0).form
    )
    assert to_list(ak._do.pad_none(regulararray, 4, 0)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
        None,
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 4, 0).form
        == ak._do.pad_none(regulararray, 4, 0).form
    )
    assert to_list(ak._do.pad_none(regulararray, 7, 0)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
        None,
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 7, 0).form
        == ak._do.pad_none(regulararray, 7, 0).form
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 1)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 1).form
        == ak._do.pad_none(regulararray, 1, 1).form
    )
    assert to_list(ak._do.pad_none(regulararray, 2, 1)) == [
        [[0.0, 1.1, 2.2], []],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], []],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 2, 1).form
        == ak._do.pad_none(regulararray, 2, 1).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 1)) == [
        [[0.0, 1.1, 2.2], [], None],
        [[3.3, 4.4], [5.5], None],
        [[6.6, 7.7, 8.8, 9.9], [], None],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 1).form
        == ak._do.pad_none(regulararray, 3, 1).form
    )
    assert to_list(ak._do.pad_none(regulararray, 5, 1)) == [
        [[0.0, 1.1, 2.2], [], None, None, None],
        [[3.3, 4.4], [5.5], None, None, None],
        [[6.6, 7.7, 8.8, 9.9], [], None, None, None],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 5, 1).form
        == ak._do.pad_none(regulararray, 5, 1).form
    )
    assert to_list(ak._do.pad_none(regulararray, 7, 1)) == [
        [[0.0, 1.1, 2.2], [], None, None, None, None, None],
        [[3.3, 4.4], [5.5], None, None, None, None, None],
        [[6.6, 7.7, 8.8, 9.9], [], None, None, None, None, None],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 7, 1).form
        == ak._do.pad_none(regulararray, 7, 1).form
    )

    assert to_list(ak._do.pad_none(regulararray, 1, 2)) == [
        [[0.0, 1.1, 2.2], [None]],
        [[3.3, 4.4], [5.5]],
        [[6.6, 7.7, 8.8, 9.9], [None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 1, 2).form
        == ak._do.pad_none(regulararray, 1, 2).form
    )
    assert to_list(ak._do.pad_none(regulararray, 2, 2)) == [
        [[0.0, 1.1, 2.2], [None, None]],
        [[3.3, 4.4], [5.5, None]],
        [[6.6, 7.7, 8.8, 9.9], [None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 2, 2).form
        == ak._do.pad_none(regulararray, 2, 2).form
    )
    assert to_list(ak._do.pad_none(regulararray, 3, 2)) == [
        [[0.0, 1.1, 2.2], [None, None, None]],
        [[3.3, 4.4, None], [5.5, None, None]],
        [[6.6, 7.7, 8.8, 9.9], [None, None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 3, 2).form
        == ak._do.pad_none(regulararray, 3, 2).form
    )
    assert to_list(ak._do.pad_none(regulararray, 4, 2)) == [
        [[0.0, 1.1, 2.2, None], [None, None, None, None]],
        [[3.3, 4.4, None, None], [5.5, None, None, None]],
        [[6.6, 7.7, 8.8, 9.9], [None, None, None, None]],
    ]
    assert (
        ak._do.pad_none(regulararray.to_typetracer(), 4, 2).form
        == ak._do.pad_none(regulararray, 4, 2).form
    )


def test_rpad_and_clip_listoffset_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10, 10]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    assert to_list(listoffsetarray) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [],
    ]

    assert to_list(ak._do.pad_none(listoffsetarray, 3, 0, clip=True)) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 3, 0, clip=True).form
        == ak._do.pad_none(listoffsetarray, 3, 0, clip=True).form
    )
    assert "option[" + str(listoffsetarray.form.type) + "]" == str(
        ak._do.pad_none(listoffsetarray, 3, 0, clip=True).form.type
    )

    assert to_list(ak._do.pad_none(listoffsetarray, 7, 0, clip=True)) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [],
        None,
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 7, 0, clip=True).form
        == ak._do.pad_none(listoffsetarray, 7, 0, clip=True).form
    )
    assert "option[" + str(listoffsetarray.form.type) + "]" == str(
        ak._do.pad_none(listoffsetarray, 7, 0, clip=True).form.type
    )

    assert to_list(ak._do.pad_none(listoffsetarray, 5, 1, clip=True)) == [
        [0.0, 1.1, 2.2, None, None],
        [None, None, None, None, None],
        [3.3, 4.4, None, None, None],
        [5.5, None, None, None, None],
        [6.6, 7.7, 8.8, 9.9, None],
        [None, None, None, None, None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 5, 1, clip=True).form
        == ak._do.pad_none(listoffsetarray, 5, 1, clip=True).form
    )

    assert str(ak._do.pad_none(listoffsetarray, 5, 1).form.type) == "var * ?float64"
    assert (
        str(ak._do.pad_none(listoffsetarray, 5, 1, clip=True).form.type)
        == "5 * ?float64"
    )

    assert to_list(ak._do.pad_none(listoffsetarray, 1, 1, clip=True)) == [
        [0.0],
        [None],
        [3.3],
        [5.5],
        [6.6],
        [None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(listoffsetarray, 1, 1, clip=True).form
    )

    content = ak.contents.numpyarray.NumpyArray(np.array([1.5, 3.3]))
    index = ak.index.Index64(
        np.array(
            [
                0,
                -3,
                1,
                -2,
                1,
                0,
                0,
                -3,
                -13,
                0,
                1,
                1,
                0,
                1,
                1,
                1,
                1,
                -10,
                0,
                -1,
                0,
                0,
                0,
                1,
                -1,
                1,
                1,
            ]
        )
    )
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    offsets = ak.index.Index64(np.array([14, 15, 15, 15, 26, 26, 26]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, indexedarray)

    assert to_list(listoffsetarray) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
    ]
    assert to_list(ak._do.pad_none(listoffsetarray, 1, 0, clip=True)) == [[3.3]]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(listoffsetarray, 1, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(listoffsetarray, 1, 1, clip=True)) == [
        [3.3],
        [None],
        [None],
        [3.3],
        [None],
        [None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(listoffsetarray, 1, 1, clip=True).form
    )


def test_rpad_listoffset_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10, 10]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)

    assert to_list(listoffsetarray) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [],
    ]

    assert to_list(ak._do.pad_none(listoffsetarray, 3, 0)) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 3, 0).form
        == ak._do.pad_none(listoffsetarray, 3, 0).form
    )

    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 6
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 3, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 7, 0)) == [
        [0.0, 1.1, 2.2],
        [],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [],
        None,
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 7, 0).form
        == ak._do.pad_none(listoffsetarray, 7, 0).form
    )

    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 7
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 7, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 5, 1)) == [
        [0.0, 1.1, 2.2, None, None],
        [None, None, None, None, None],
        [3.3, 4.4, None, None, None],
        [5.5, None, None, None, None],
        [6.6, 7.7, 8.8, 9.9, None],
        [None, None, None, None, None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 5, 1).form
        == ak._do.pad_none(listoffsetarray, 5, 1).form
    )
    assert ak.operations.type(
        ak._do.pad_none(listoffsetarray, 5, 1)
    ) == ak.types.ArrayType(
        ak.types.ListType(ak.types.OptionType(ak.types.NumpyType("float64"))), 6
    )

    assert to_list(ak._do.pad_none(listoffsetarray, 1, 1)) == [
        [0.0, 1.1, 2.2],
        [None],
        [3.3, 4.4],
        [5.5],
        [6.6, 7.7, 8.8, 9.9],
        [None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 1).form
        == ak._do.pad_none(listoffsetarray, 1, 1).form
    )

    content = ak.contents.numpyarray.NumpyArray(np.array([1.5, 3.3]))
    index = ak.index.Index64(
        np.array(
            [
                0,
                -3,
                1,
                -2,
                1,
                0,
                0,
                -3,
                -13,
                0,
                1,
                1,
                0,
                1,
                1,
                1,
                1,
                -10,
                0,
                -1,
                0,
                0,
                0,
                1,
                -1,
                1,
                1,
            ]
        )
    )
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    offsets = ak.index.Index64(np.array([14, 15, 15, 15, 26, 26, 26]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, indexedarray)

    assert to_list(listoffsetarray) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
    ]

    assert to_list(ak._do.pad_none(listoffsetarray, 1, 0)) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 0).form
        == ak._do.pad_none(listoffsetarray, 1, 0).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(
            ak.types.ListType(ak.types.OptionType(ak.types.NumpyType("float64")))
        ),
        6,
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 1, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 6, 0)) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 6, 0).form
        == ak._do.pad_none(listoffsetarray, 6, 0).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(
            ak.types.ListType(ak.types.OptionType(ak.types.NumpyType("float64")))
        ),
        6,
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 6, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 7, 0)) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
        None,
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 7, 0).form
        == ak._do.pad_none(listoffsetarray, 7, 0).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.operations.type(listoffsetarray).content), 7
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 7, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 9, 0)) == [
        [3.3],
        [],
        [],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [],
        [],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 9, 0).form
        == ak._do.pad_none(listoffsetarray, 9, 0).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.operations.type(listoffsetarray).content), 9
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 9, 0))

    assert to_list(ak._do.pad_none(listoffsetarray, 1, 1)) == [
        [3.3],
        [None],
        [None],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [None],
        [None],
    ]
    assert (
        ak._do.pad_none(listoffsetarray.to_typetracer(), 1, 1).form
        == ak._do.pad_none(listoffsetarray, 1, 1).form
    )
    assert ak.types.ArrayType(
        ak.types.ListType(ak.types.OptionType(ak.types.NumpyType("float64"))), 6
    ) == ak.operations.type(ak._do.pad_none(listoffsetarray, 1, 1))

    assert to_list(ak._do.pad_none(listoffsetarray, 4, 1)) == [
        [3.3, None, None, None],
        [None, None, None, None],
        [None, None, None, None],
        [3.3, 3.3, None, 1.5, None, 1.5, 1.5, 1.5, 3.3, None, 3.3],
        [None, None, None, None],
        [None, None, None, None],
    ]
    assert ak.operations.type(listoffsetarray) == ak.operations.type(
        ak._do.pad_none(listoffsetarray, 4, 1)
    )


def test_rpad_list_array():
    content = ak.contents.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    starts = ak.index.Index64(np.array([0, 3, 4, 5, 8]))
    stops = ak.index.Index64(np.array([3, 3, 6, 8, 9]))
    array = ak.contents.ListArray(starts, stops, content)

    assert to_list(array) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
    ]
    assert to_list(ak._do.pad_none(array, 1, 0)) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
    ]
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 5
    ) == ak.operations.type(ak._do.pad_none(array, 1, 0))

    assert to_list(ak._do.pad_none(array, 2, 0)) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
    ]
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 5
    ) == ak.operations.type(ak._do.pad_none(array, 2, 0))

    assert to_list(ak._do.pad_none(array, 7, 0)) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
        None,
        None,
    ]
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 7
    ) == ak.operations.type(ak._do.pad_none(array, 7, 0))

    assert to_list(ak._do.pad_none(array, 1, 1)) == [
        [0.0, 1.1, 2.2],
        [None],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
    ]

    assert to_list(ak._do.pad_none(array, 2, 1)) == [
        [0.0, 1.1, 2.2],
        [None, None],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8, None],
    ]

    assert to_list(ak._do.pad_none(array, 3, 1)) == [
        [0.0, 1.1, 2.2],
        [None, None, None],
        [4.4, 5.5, None],
        [5.5, 6.6, 7.7],
        [8.8, None, None],
    ]

    assert to_list(ak._do.pad_none(array, 4, 1)) == [
        [0.0, 1.1, 2.2, None],
        [None, None, None, None],
        [4.4, 5.5, None, None],
        [5.5, 6.6, 7.7, None],
        [8.8, None, None, None],
    ]


def test_rpad_and_clip_list_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    starts = ak.index.Index64(np.array([0, 3, 4, 5, 8]))
    stops = ak.index.Index64(np.array([3, 3, 6, 8, 9]))
    array = ak.contents.listarray.ListArray(starts, stops, content)

    assert to_list(array) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
    ]
    assert to_list(ak._do.pad_none(array, 1, 0, clip=True)) == [[0.0, 1.1, 2.2]]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(array, 1, 0, clip=True).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 1
    ) == ak.operations.type(ak._do.pad_none(array, 1, 0, clip=True))

    assert to_list(ak._do.pad_none(array, 2, 0, clip=True)) == [[0.0, 1.1, 2.2], []]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 0, clip=True).form
        == ak._do.pad_none(array, 2, 0, clip=True).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 2
    ) == ak.operations.type(ak._do.pad_none(array, 2, 0, clip=True))

    assert to_list(ak._do.pad_none(array, 7, 0, clip=True)) == [
        [0.0, 1.1, 2.2],
        [],
        [4.4, 5.5],
        [5.5, 6.6, 7.7],
        [8.8],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 7, 0, clip=True).form
        == ak._do.pad_none(array, 7, 0, clip=True).form
    )
    assert ak.types.ArrayType(
        ak.types.OptionType(ak.types.ListType(ak.types.NumpyType("float64"))), 7
    ) == ak.operations.type(ak._do.pad_none(array, 7, 0, clip=True))

    assert to_list(ak._do.pad_none(array, 1, 1, clip=True)) == [
        [0.0],
        [None],
        [4.4],
        [5.5],
        [8.8],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(array, 1, 1, clip=True).form
    )

    assert to_list(ak._do.pad_none(array, 2, 1, clip=True)) == [
        [0.0, 1.1],
        [None, None],
        [4.4, 5.5],
        [5.5, 6.6],
        [8.8, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 1, clip=True).form
        == ak._do.pad_none(array, 2, 1, clip=True).form
    )


def test_rpad_indexed_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    starts = ak.index.Index64(np.array([0, 3, 3, 5, 6]))
    stops = ak.index.Index64(np.array([3, 3, 5, 6, 10]))
    listoffsetarray = ak.contents.listarray.ListArray(starts, stops, content)
    content = ak.contents.numpyarray.NumpyArray(
        np.array([6.6, 7.7, 8.8, 9.9, 5.5, 3.3, 4.4, 0.0, 1.1, 2.2])
    )
    offsets = ak.index.Index64(np.array([0, 4, 5, 7, 7, 10]))
    backward = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)

    index = ak.index.Index64(np.array([4, 3, 2, 1, 0], dtype=np.int64))
    indexedarray = ak.contents.indexedarray.IndexedArray(index, listoffsetarray)
    assert to_list(indexedarray) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]

    assert to_list(ak._do.pad_none(backward, 4, 1)) == to_list(
        ak._do.pad_none(indexedarray, 4, 1)
    )
    assert to_list(ak._do.pad_none(indexedarray, 1, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]
    assert to_list(ak._do.pad_none(indexedarray, 2, 1)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5, None],
        [3.3, 4.4],
        [None, None],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 2, 1).form
        == ak._do.pad_none(indexedarray, 2, 1).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 3, 1)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5, None, None],
        [3.3, 4.4, None],
        [None, None, None],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 3, 1).form
        == ak._do.pad_none(indexedarray, 3, 1).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 4, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 4, 0).form
        == ak._do.pad_none(indexedarray, 4, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 5, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 5, 0).form
        == ak._do.pad_none(indexedarray, 5, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 6, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 6, 0).form
        == ak._do.pad_none(indexedarray, 6, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 7, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 7, 0).form
        == ak._do.pad_none(indexedarray, 7, 0).form
    )


def test_rpad_and_clip_indexed_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10]))
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    content = ak.contents.numpyarray.NumpyArray(
        np.array([6.6, 7.7, 8.8, 9.9, 5.5, 3.3, 4.4, 0.0, 1.1, 2.2])
    )
    offsets = ak.index.Index64(np.array([0, 4, 5, 7, 7, 10]))
    backward = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)

    index = ak.index.Index64(np.array([4, 3, 2, 1, 0], dtype=np.int64))
    indexedarray = ak.contents.IndexedArray(index, listoffsetarray)
    assert to_list(indexedarray) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]

    assert to_list(ak._do.pad_none(backward, 4, 1, clip=True)) == to_list(
        ak._do.pad_none(indexedarray, 4, 1, clip=True)
    )
    assert to_list(ak._do.pad_none(indexedarray, 1, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9]
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 1, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 2, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 2, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 2, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 3, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 3, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 3, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 4, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 4, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 4, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 5, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 5, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 5, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 6, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 6, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 6, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 7, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 7, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 7, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 8, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        [],
        [0.0, 1.1, 2.2],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 8, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 8, 0, clip=True).form
    )

    assert to_list(ak._do.pad_none(indexedarray, 1, 1, clip=True)) == [
        [6.6],
        [5.5],
        [3.3],
        [None],
        [0.0],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 1, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 2, 1, clip=True)) == [
        [6.6, 7.7],
        [5.5, None],
        [3.3, 4.4],
        [None, None],
        [0.0, 1.1],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 2, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 2, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 3, 1, clip=True)) == [
        [6.6, 7.7, 8.8],
        [5.5, None, None],
        [3.3, 4.4, None],
        [None, None, None],
        [0.0, 1.1, 2.2],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 3, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 3, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 4, 1, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5, None, None, None],
        [3.3, 4.4, None, None],
        [None, None, None, None],
        [0.0, 1.1, 2.2, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 4, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 4, 1, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 5, 1, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9, None],
        [5.5, None, None, None, None],
        [3.3, 4.4, None, None, None],
        [None, None, None, None, None],
        [0.0, 1.1, 2.2, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 5, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 5, 1, clip=True).form
    )


def test_rpad_indexed_option_array():
    content = ak.contents.numpyarray.NumpyArray(
        np.asarray([0.0, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
    )
    index = ak.index.Index64(np.asarray([0, -1, -1, 1, 2, 3, 4, 5, 6, 7]))
    offsets = ak.index.Index64(np.asarray([0, 3, 5, 6, 10]))
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, indexedarray)
    index = ak.index.Index64(
        np.asarray(
            [
                0,
                -1,
                1,
                2,
                3,
            ]
        )
    )
    indexedoptionarray = ak.contents.indexedoptionarray.IndexedOptionArray(
        index, listoffsetarray
    )
    content = ak.contents.numpyarray.NumpyArray(
        np.asarray([6.6, 7.7, 8.8, 9.9, 5.5, 3.3, 4.4, 0.0])
    )
    index = ak.index.Index64(np.asarray([0, 1, 2, 3, 4, 5, 6, 7, -1, -1]))
    offsets = ak.index.Index64(np.asarray([0, 4, 5, 7, 10]))
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray(index, content)
    listoffsetarray = ak.contents.listoffsetarray.ListOffsetArray(offsets, indexedarray)
    index = ak.index.Index64(
        np.asarray(
            [
                0,
                1,
                2,
                -1,
                3,
            ]
        )
    )
    backward = ak.contents.indexedoptionarray.IndexedOptionArray(index, listoffsetarray)

    index = ak.index.Index64(np.array([4, 3, 2, -1, 0], dtype=np.int64))
    indexedarray = ak.contents.indexedoptionarray.IndexedOptionArray.simplified(
        index, indexedoptionarray
    )
    assert to_list(indexedarray) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
    ]

    assert to_list(ak._do.pad_none(backward, 4, 1)) == to_list(
        ak._do.pad_none(indexedarray, 4, 1)
    )
    assert to_list(ak._do.pad_none(indexedarray, 1, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 0).form
        == ak._do.pad_none(indexedarray, 1, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 1, 1)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 1).form
        == ak._do.pad_none(indexedarray, 1, 1).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 3, 1)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5, None, None],
        [3.3, 4.4, None],
        None,
        [0.0, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 3, 1).form
        == ak._do.pad_none(indexedarray, 3, 1).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 4, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 4, 0).form
        == ak._do.pad_none(indexedarray, 4, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 5, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 5, 0).form
        == ak._do.pad_none(indexedarray, 5, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 6, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 6, 0).form
        == ak._do.pad_none(indexedarray, 6, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 7, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
        None,
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 7, 0).form
        == ak._do.pad_none(indexedarray, 7, 0).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 8, 0)) == [
        [6.6, 7.7, 8.8, 9.9],
        [5.5],
        [3.3, 4.4],
        None,
        [0.0, None, None],
        None,
        None,
        None,
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 8, 0).form
        == ak._do.pad_none(indexedarray, 8, 0).form
    )

    assert to_list(ak._do.pad_none(indexedarray, 1, 0, clip=True)) == [
        [6.6, 7.7, 8.8, 9.9]
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 0, clip=True).form
        == ak._do.pad_none(indexedarray, 1, 0, clip=True).form
    )
    assert to_list(ak._do.pad_none(indexedarray, 1, 1, clip=True)) == [
        [6.6],
        [5.5],
        [3.3],
        None,
        [0.0],
    ]
    assert (
        ak._do.pad_none(indexedarray.to_typetracer(), 1, 1, clip=True).form
        == ak._do.pad_none(indexedarray, 1, 1, clip=True).form
    )


def test_rpad_recordarray():
    keys = ["x", "y"]
    offsets = ak.index.Index64(np.asarray([0, 0, 1, 3]))
    content = ak.contents.numpyarray.NumpyArray(np.asarray([1.1, 2.2, 2.2]))
    content1 = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    offsets = ak.index.Index64(np.asarray([0, 2, 3, 3]))
    content = ak.contents.numpyarray.NumpyArray(np.asarray([2, 2, 1]))
    content2 = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    contents = [content1, content2]
    array = ak.contents.recordarray.RecordArray(contents, keys)

    assert to_list(ak._do.pad_none(array, 5, 0)) == [
        {"x": [], "y": [2, 2]},
        {"x": [1.1], "y": [1]},
        {"x": [2.2, 2.2], "y": []},
        None,
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 5, 0).form
        == ak._do.pad_none(array, 5, 0).form
    )

    assert to_list(ak._do.pad_none(array, 2, 1)) == [
        {"x": [None, None], "y": [2, 2]},
        {"x": [1.1, None], "y": [1, None]},
        {"x": [2.2, 2.2], "y": [None, None]},
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 1).form
        == ak._do.pad_none(array, 2, 1).form
    )


def test_rpad_unionarray():
    offsets = ak.index.Index64(np.asarray([0, 0, 1, 3]))
    content = ak.contents.numpyarray.NumpyArray(np.asarray([1.1, 2.2, 2.2]))
    content1 = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    offsets = ak.index.Index64(np.asarray([0, 2, 3, 3]))
    content = ak.from_iter(["2", "2", "1"], highlevel=False)
    content2 = ak.contents.listoffsetarray.ListOffsetArray(offsets, content)
    tags = ak.index.Index8(np.array([0, 1, 0, 1, 0, 1], dtype=np.int8))
    index = ak.index.Index64(np.array([0, 0, 1, 1, 2, 2], dtype=np.int64))
    array = ak.contents.unionarray.UnionArray(tags, index, [content1, content2])
    assert to_list(array) == [[], ["2", "2"], [1.1], ["1"], [2.2, 2.2], []]

    assert to_list(ak._do.pad_none(array, 7, 0)) == [
        [],
        ["2", "2"],
        [1.1],
        ["1"],
        [2.2, 2.2],
        [],
        None,
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 7, 0).form
        == ak._do.pad_none(array, 7, 0).form
    )

    assert to_list(ak._do.pad_none(array, 2, 1)) == [
        [None, None],
        ["2", "2"],
        [1.1, None],
        ["1", None],
        [2.2, 2.2],
        [None, None],
    ]
    assert (
        ak._do.pad_none(array.to_typetracer(), 2, 1).form
        == ak._do.pad_none(array, 2, 1).form
    )