code: def test_module(self, device, dtype):
      op = apply_colormap
      cm = AUTUMN(device=device, dtype=dtype)
      op_module = ApplyColorMap(cm)
  
      img = torch.ones(1, 3, 3, device=device, dtype=dtype)
  
      self.assert_close(op(img, cm), op_module(img))
  
file_path: kornia/test/color/test_colormap.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  from kornia.color import AUTUMN, ApplyColorMap, apply_colormap
  from kornia.core import tensor
  from kornia.testing import BaseTester, assert_close, tensor_to_gradcheck_var
label_desc: Use image node_id 7 for calling the TestApplyColorMap obj's underlying member method code with example usage: obj.test_module(device, dtype) without return types