code: def test_feature_embedder(config, hybridize):
      out_shape = config["shape"][:-1] + (
          sum(config["kwargs"]["embedding_dims"]),
      )
  
      embed_feature = FeatureEmbedder(
          prefix="embed_feature_", **config["kwargs"]
      )
      embed_feature.collect_params().initialize(mx.initializer.One())
  
      if hybridize:
          embed_feature.hybridize()
  
      def test_parameters_length():
          exp_params_len = len(embed_feature.collect_params().keys())
          act_params_len = len(config["kwargs"]["embedding_dims"])
          assert exp_params_len == act_params_len
  
      def test_parameter_names():
          for param in embed_feature.collect_params():
              assert param.startswith("embed_feature_")
  
      def test_forward_pass():
          act_output = embed_feature(mx.nd.ones(shape=config["shape"]))
          exp_output = mx.nd.ones(shape=out_shape)
  
          assert act_output.shape == exp_output.shape
          assert mx.nd.sum(act_output - exp_output) < 1e-20
  
      test_parameters_length()
      test_parameter_names()
      test_forward_pass()
  
imports: from itertools import chain, combinations
  import mxnet
  import pytest
  from gluonts.mx.block.feature import FeatureAssembler, FeatureEmbedder
label_desc: Use image node_id 1 for calling a global function with example usage: test_feature_embedder(config, hybridize) without return types