code: def test_fc_middleware():
      params = FCMiddlewareParameters(scheme=MiddlewareScheme.Medium)
      mid = FCMiddleware(params=params)
      mid.initialize()
      embedded_data = mx.nd.random.uniform(
          low=0, high=1, shape=(10, 100)
      )
      output = mid(embedded_data)
      assert output.ndim == 2  # since last block was flatten
      assert output.shape[0] == 10  # since batch_size is 10
      assert output.shape[1] == 512
  
file_path: coach/rl_coach/tests/architectures/mxnet_components/middlewares/test_fc_middleware.py
imports: import mxnet
  import os
  import pytest
  import sys
  from rl_coach.base_parameters import MiddlewareScheme
  from rl_coach.architectures.middleware_parameters import FCMiddlewareParameters
  from rl_coach.architectures.mxnet_components.middlewares.fc_middleware import FCMiddleware
label_desc: Use image node_id 1 for calling a global function with example usage: test_fc_middleware() without return types