code: def test_gmm_conv(g, idtype):
      g = g.astype(idtype).to(F.ctx())
      ctx = F.ctx()
      gmm_conv = nn.GMMConv(5, 2, 5, 3, "max")
      gmm_conv.initialize(ctx=ctx)
      h0 = F.randn((g.number_of_src_nodes(), 5))
      pseudo = F.randn((g.num_edges(), 5))
      h1 = gmm_conv(g, h0, pseudo)
      assert h1.shape == (g.number_of_dst_nodes(), 2)
  
imports: import backend
  import dgl
  import dgl.function
  import dgl.nn.mxnet
  import mxnet
  import networkx
  import numpy
  import pytest
  import scipy
  from mxnet import autograd, gluon, nd
  from utils import parametrize_idtype
  from utils.graph_cases import get_cases, random_bipartite, random_dglgraph, random_graph
label_desc: Use image node_id 25 for calling a global function with example usage: test_gmm_conv(g, idtype) without return types