code: def test_mlp(in_features: int, hidden_size: int, out_features: int):
      assert_output_shapes(
          module=MLP(in_features, hidden_size, out_features),
          input_shape=(in_features,),
      )
  
imports: import pytest
  from ludwig.modules.mlp_mixer_modules import MixerBlock, MLP, MLPMixer
  from .test_utils import assert_output_shapes
label_desc: Use image node_id 1 for calling a global function with example usage: test_mlp(in_features, hidden_size, out_features) without return types