Add test for the previous fix and add more check to prevent other crash with bad neib_shape or neib_step input value.