Numpy shift with numba showing weird behavior

Question

I used the elements shift code from here. It works good. The piece of code I have used is as follows:

# https://stackoverflow.com/a/62841583/4272651
@nb.njit
def