code: def test_mlp_mixer(img_height: int, img_width: int, in_channels: int):
      assert_output_shapes(
          module=MLPMixer(img_height, img_width, in_channels),
          input_shape=(3, img_height, img_width),
      )
  
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 3 for calling a global function with example usage: test_mlp_mixer(img_height, img_width, in_channels) without return types