code: def testXmap(self):
      f = maps.xmap(
          lambda x: 0.0 / x,
          in_axes=["i"],
          out_axes=["i"],
          axis_resources={"i": "x"},
      )
  
      with jax.sharding.Mesh(np.array(jax.local_devices()[:1]), ("x",)):
          with self.assertRaisesRegex(
              FloatingPointError,
              r"invalid value \(nan\) encountered in xmap",
          ):
              ans = f(jnp.array([0.0]))
              ans.block_until_ready()
  
      if jax.device_count() >= 2:
          with jax.sharding.Mesh(
              np.array(jax.local_devices()[:2]), ("x",)
          ):
              with self.assertRaises(FloatingPointError):
                  ans = f(jnp.array([1.0, 0.0]))
                  ans.block_until_ready()
  
imports: from absl.testing import absltest
  import jax
  import numpy
  from unittest import SkipTest
  from jax._src import api
  from jax._src import test_util
  from jax import numpy
  from jax.experimental import pjit, maps
  from jax import config
label_desc: Use image node_id 13 for calling the DebugNaNsTest obj's underlying member method code with example usage: obj.testXmap() without return types