[mpi] Add test for transposed, fix r2c, c2r shape issues in d > 1