code: def test_moe_initialization(expert_parallel):
      spawn(_run_test, 2, expert_parallel=expert_parallel)
  
file_path: colossalai/tests/test_moe/test_moe_group.py
imports: import pytest
  import torch.distributed
  import torch.nn
  import colossalai
  from colossalai.moe.experts import MLPExperts
  from colossalai.moe.manager import MOE_MANAGER
  from colossalai.moe.utils import sync_moe_model_param
  from colossalai.testing import assert_equal_in_group, rerun_if_address_is_in_use, spawn
  from colossalai.utils import get_current_device
label_desc: Use image node_id 3 for calling a global function with example usage: test_moe_initialization(expert_parallel) without return types