code: def test_data_parallel_initializations():
      Utils.initialize_model_parallel(
          pipeline_model_parallel_size=world_size
      )
      assert ps.get_data_parallel_src_rank() == rank
      assert ps.get_data_parallel_world_size() == 1
      assert ps.get_data_parallel_rank() == 0
      Utils.destroy_model_parallel()
  
file_path: megatron-lm/tests/unit_tests/test_parallel_state.py
imports: import torch
  import megatron.core.parallel_state
  import pytest
  from tests.unit_tests.test_utilities import Utils
  import os
label_desc: Use image node_id 3 for calling a global function with example usage: test_data_parallel_initializations() without return types