code: def test_mixer_block(
      embed_size: int,
      n_patches: int,
      token_dim: int,
      channel_dim: int,
  ):
      assert_output_shapes(
          module=MixerBlock(
              embed_size, n_patches, token_dim, channel_dim
          ),
          input_shape=(n_patches, embed_size),
      )
  
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 2 for calling a global function with example usage: test_mixer_block(embed_size, n_patches, token_dim, channel_dim) without return types