Add unitteses for get_rotation