code: def sub_mesh_assert_equal(
      self, mesh, exp_in_mesh, exp_out_of_mesh, tensor
  ):
      if self.rank in mesh:
          self.assertEqual(tensor, exp_in_mesh)
      else:
          self.assertEqual(tensor, exp_out_of_mesh)
  
imports: import torch
  import torch.nn.functional
  from numpy.testing import assert_array_equal
  from torch.distributed._functional_collectives import AsyncCollectiveTensor
  from torch.distributed._tensor import DeviceMesh, distribute_tensor, DTensor, init_device_mesh
  from torch.distributed._tensor.placement_types import _Partial, Replicate, Shard
  from torch.distributed.tensor.parallel import ColwiseParallel, parallelize_module, RowwiseParallel
  from torch.testing._internal.common_utils import run_tests
  from torch.testing._internal.distributed._tensor.common_dtensor import DTensorTestBase, with_comms
label_desc: Use image node_id 2 for calling the DTensorMeshTest obj's underlying member method code with example usage: obj.sub_mesh_assert_equal(mesh, exp_in_mesh, exp_out_of_mesh, tensor) without return types