code: def test_get_grid_dict(ps):
      grid_dict = get_grid_dict(ps)
      param_keys = ["x", "y", "phi", "rho"]
      assert set(grid_dict.keys()) == set(param_keys)
      for k in param_keys:
          assert grid_dict[k].shape == (ps, ps)
  
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 a global function with example usage: test_get_grid_dict(ps) without return types