It looks to me like there is redundant code in check_array in the ensure_2d part. We just checked ndim, so what do we do atleast_2d for? (this leaves alone higher n-dim, right)?

I think we should remove these two lines but we need to make sure behavior is really identical.