swap axis decorator and docs for test