code: def testCallDeoptimized(self, jit):
      @jit
      def f(x):
          return jax.lax.cond(
              x == 1, lambda _: np.nan, lambda _: 2.0, operand=None
          )
  
      # This makes sure, when using the C++ jit, that the Python code has been
      # run to compile, and the next call won't go through `cache_miss`.
      f(2)
      # 'cond' not 'xla_call'
      msg = r"invalid value \(nan\) encountered in .*cond.*"
      with self.assertRaisesRegex(FloatingPointError, msg):
          f(1)
  
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 10 for calling the DebugNaNsTest obj's underlying member method code with example usage: obj.testCallDeoptimized(jit) and returns: jax