code: def test_mlp2_infer_shape():
      # Build MLP
      out = models.mlp2()
      # infer shape
      data_shape = (100, 100)
      arg_shapes, out_shapes, aux_shapes = out.infer_shape(
          data=data_shape
      )
      arg_shape_dict = dict(zip(out.list_arguments(), arg_shapes))
      assert len(out_shapes) == 1
      assert out_shapes[0] == (100, 10)
      true_shapes = {
          "fc2_bias": (10,),
          "fc2_weight": (10, 1000),
          "fc1_bias": (1000,),
          "fc1_weight": (1000, 100),
      }
      for k, v in true_shapes.items():
          assert arg_shape_dict[k] == v
  
file_path: mxnet/tests/python/unittest/test_infer_shape.py
imports: import mxnet
  from common import models
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: test_mlp2_infer_shape() without return types