code: def test_distort(self, device, dtype):
      distortion = AffineTransform()
      points = torch.tensor(
          [[1.0, 1.0], [1.0, 5.0], [2.0, 4.0], [3.0, 9.0]],
          device=device,
          dtype=dtype,
      )
      params = torch.tensor(
          [[328.0, 328.0, 150.0, 150.0]], device=device, dtype=dtype
      )
      expected = torch.tensor(
          [
              [478.0, 478.0],
              [478.0, 1790.0],
              [806.0, 1462.0],
              [1134.0, 3102.0],
          ],
          device=device,
          dtype=dtype,
      )
      self.assert_close(
          distortion.distort(params, Vector2(points)).data, expected
      )
  
file_path: kornia/test/sensors/camera/test_distortion_model.py
imports: import pytest
  import torch
  from kornia.geometry.vector import Vector2
  from kornia.sensors.camera.distortion_model import AffineTransform
  from kornia.testing import BaseTester
label_desc: Use image node_id 7 for calling the TestAffineTransform obj's underlying member method code with example usage: obj.test_distort(device, dtype) without return types