Fixed test code for norm and swapaxes function.