Can you split this up? Perhaps in basic / multidim / axis_kw

And missing dtype tests (but wait with adding those, first decide if we need to keep that kw)