code: def test_get_physical_tpu_mesh(self, xyz, reorder):
      x, y, z = xyz
      jax_devices = mock_tpu_devices(
          x, y, z, "TPU v4", True, reorder=reorder
      )
      normalized = mesh_utils._get_physical_tpu_mesh(jax_devices)
      self.assertEqual(normalized.shape, xyz)
      # major_to_minor: x, y, z
      for i in range(x):
          for j in range(y):
              for k in range(z):
                  self.assertEqual(
                      normalized[i, j, k].coords, (i, j, k)
                  )
  
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 2 for calling the MeshUtilsTest obj's underlying member method code with example usage: obj.test_get_physical_tpu_mesh(xyz, reorder) without return types