add test for axis re-ordering