Fix test_take to make axis optional when ndim == 1

[asmeurer]

I didn't explicitly test axis=None because it's not clear to me that should actually be supported, given that that's the same as axis=0.

Fixes #191