code: def test_shape(self, ps, kernel_type, device):
      skd = SimpleKD(patch_size=ps, kernel_type=kernel_type).to(device)
      inp = torch.ones(1, 1, ps, ps).to(device)
      out = skd(inp)
      assert out.shape == (1, min(128, self.dims[kernel_type]))
  
imports: from math import pi
  import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia.testing
  from kornia.feature.mkd import COEFFS, EmbedGradients, ExplicitSpacialEncoding, MKDDescriptor, MKDGradients, SimpleKD, VonMisesKernel, Whitening, get_grid_dict, get_kron_order, spatial_kernel_embedding
  from kornia.testing import assert_close
label_desc: Use image node_id 1 for calling the TestSimpleKD obj's underlying member method code with example usage: obj.test_shape(ps, kernel_type, device) without return types