Then, maybe we should go the following path:

* ndim > 3 => try np.squeeze and test again for ndim
* ndim == 2 => good
* ndim < 2 => np.at_least_2d (so, one can also pass 1-D arrays and unwrap the phase)

What do you think?