code: def test_create_contiguous_submeshes_for_tpu_v4_leading_1_dims(self):
      v4 = mesh_utils._TPU_V4
      for topology, mesh_shapes in mesh_utils._TRANSPOSE_TRICKS.items():
          logging.vlog(1, "topology: %s", topology)
          devices = mock_tpu_devices(
              topology[0],
              topology[1],
              topology[2],
              v4,
              one_device_per_chip=True,
          )
          for mesh_shape in mesh_shapes:
              logging.vlog(
                  1, "  mesh_shape: %s", (1, 1) + mesh_shape + (1, 1)
              )
              mesh = mesh_utils.create_device_mesh(
                  (1, 1) + mesh_shape + (1, 1),
                  devices=devices,
                  contiguous_submeshes=True,
              )
              self._assert_contiguous_submeshes(mesh)
  
imports: import collections
  from collections.abc import Sequence
  import dataclasses
  import numpy
  from absl import logging
  from absl.testing import absltest
  from absl.testing import parameterized
  from jax.experimental import mesh_utils
  from jax.sharding import Mesh
  from jax._src import test_util
label_desc: Use image node_id 9 for calling the MeshUtilsTest obj's underlying member method code with example usage: obj.test_create_contiguous_submeshes_for_tpu_v4_leading_1_dims() without return types