code: def test_jit(self, device, dtype):
      data = torch.rand(10, 3, 1, 2, device=device, dtype=dtype)
      zca = kornia.enhance.ZCAWhitening().fit(data)
      zca_jit = kornia.enhance.ZCAWhitening().fit(data)
      zca_jit = torch.jit.script(zca_jit)
      self.assert_close(zca_jit(data), zca(data))
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import BaseTester
label_desc: Use image node_id 9 for calling the TestZCA obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types