code: def test_broadcast_1d(self):
      mesh = DeviceMesh(self.device_type, torch.arange(self.world_size))
      local_tensor = (
          torch.ones(3, 3, device=self.device_type) * self.rank
      )
      mesh_broadcast(local_tensor, mesh, mesh_dim=0)
      self.assertEqual(local_tensor, torch.zeros(3, 3))
  
imports: import os
  import torch
  import torch.distributed._functional_collectives
  from torch.distributed._tensor import DTensor
  from torch.distributed._tensor._collective_utils import mesh_all_to_all, mesh_broadcast, mesh_scatter
  from torch.distributed._tensor.placement_types import _Partial, Shard
  from torch.distributed.device_mesh import _mesh_resources, DeviceMesh, init_device_mesh
  from torch.distributed.distributed_c10d import get_global_rank, get_world_size, init_process_group, is_initialized, is_nccl_available, ProcessGroup
  from torch.testing._internal.common_utils import run_tests
  from torch.testing._internal.distributed._tensor.common_dtensor import DTensorTestBase, with_comms
  from torch.testing._internal.distributed.fake_pg import FakeStore
label_desc: Use image node_id 2 for calling the DeviceMeshCollectiveTest obj's underlying member method code with example usage: obj.test_broadcast_1d() without return types