08c76ed8511e2e4d7f7a3f4406df4c073112e6bd,python/test/function/test_flip.py,,ref_flip,#Any#Any#,23

Before Change

def ref_flip(x, axes):
    if axes is None:
        axes = [len(x.shape) - 1]
    x = x.copy()
    for axis in axes:
        if axis == 0:
            x = x[::-1]
        elif axis == 1:
            x = x[::, ::-1]
        elif axis == 2:
            x = x[::, ::, ::-1]
        else:
            raise
    return x


@pytest.mark.parametrize("ctx, func_name", ctxs)
@pytest.mark.parametrize("seed", [313])

After Change

    //         x = x[::, ::, ::-1]
    //     else:
    //         raise
    return np.flip(x, axes)


@pytest.mark.parametrize("ctx, func_name", ctxs)
@pytest.mark.parametrize("seed", [313])

Instances