code: def testDebugNansZeroDiv(self):
      inp = jnp.zeros(())
  
      def f(x, y):
          return x / y
  
      with self.assertRaisesRegex(
          FloatingPointError,
          r"invalid value \(nan\) encountered in jit\(true_divide\)",
      ):
          f(inp, inp)
  
      # TODO(yashkatariya): Fix this and make true_divide appear in the name again.
      # Instead of `f` showing up in the error, the name should be of the
      # primitive (true_divide) in this case.
      with self.assertRaisesRegex(
          FloatingPointError,
          r"invalid value \(nan\) encountered in jit\(f\)",
      ):
          jax.jit(f)(inp, inp)
  
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 18 for calling the DebugNaNsTest obj's underlying member method code with example usage: obj.testDebugNansZeroDiv() and returns: unknown