code: def test_jit(self, batch_size, device, dtype):
      src, dst = _sample_points(batch_size, device)
      kernel, affine = kornia.geometry.transform.get_tps_transform(
          src, dst
      )
      op = kornia.geometry.transform.warp_points_tps
      op_jit = torch.jit.script(op)
      assert_close(
          op(src, dst, kernel, affine), op_jit(src, dst, kernel, affine)
      )
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  from kornia.testing import assert_close
  from kornia.utils._compat import torch_version_ge
label_desc: Use image node_id 5 for calling the TestWarpPoints obj's underlying member method code with example usage: obj.test_jit(batch_size, device, dtype) without return types