code: def testJitComputationNaNContextManager(self):
      config.update("jax_debug_nans", False)
      A = jnp.array(0.0)
      f = jax.jit(lambda x: 0.0 / x)
      ans = f(A)
      ans = f(A)
      with self.assertRaises(FloatingPointError):
          with jax.debug_nans(True):
              ans = f(A)
          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 8 for calling the DebugNaNsTest obj's underlying member method code with example usage: obj.testJitComputationNaNContextManager() without return types