code: def test_rotation(self, axis, device, dtype, atol, rtol):
      # half_sqrt2 = 0.5 * np.sqrt(2)
      array = [0.0, 0.0, 0.0]
      array[axis] = kornia.pi / 2.0
      axis_angle = torch.tensor(array, device=device, dtype=dtype)
      quaternion = kornia.geometry.conversions.axis_angle_to_quaternion(
          axis_angle
      )
      axis_angle_hat = (
          kornia.geometry.conversions.quaternion_to_axis_angle(
              quaternion
          )
      )
      assert_close(axis_angle_hat, axis_angle, atol=atol, rtol=rtol)
  
imports: import numpy
  import pytest
  import torch
  import kornia
  from kornia.testing import assert_close
label_desc: Use image node_id 3 for calling the TestAngleAxisToQuaternionToAngleAxis obj's underlying member method code with example usage: obj.test_rotation(axis, device, dtype, atol, rtol) without return types