code: def test_jvp_of_scan_should_transform_stack(self):
      @jax.named_scope("foo")
      def f(x):
          @jax.named_scope("scan_body")
          def body(carry, x):
              return carry + x, carry + x
  
          return lax.scan(body, x, jnp.arange(8, dtype="float32"))
  
      g = lambda x, t: jax.jvp(f, (x,), (t,))
      jaxpr = jax.make_jaxpr(g)(jnp.float32(1), jnp.float32(1))
      self.assertEqual(
          str(jaxpr.eqns[0].source_info.name_stack), "jvp(foo)"
      )
      self.assertEqual(
          str(
              jaxpr.eqns[1]
              .params["jaxpr"]
              .eqns[0]
              .source_info.name_stack
          ),
          "scan_body",
      )
  
      hlo_text = _get_hlo(g)(1.0, 1.0)
      self.assertIn("jvp(foo)/while/body/scan_body/add", hlo_text)
  
imports: import functools
  from absl.testing import absltest
  import jax
  import jax.numpy
  from jax._src import core
  from jax import lax
  from jax._src.pjit import pjit
  from jax._src import linear_util
  from jax import config
  from jax._src import test_util
  from jax._src.lib import xla_client
label_desc: Use image node_id 13 for calling the NameStackControlFlowTest obj's underlying member method code with example usage: obj.test_jvp_of_scan_should_transform_stack() and returns: lax, unknown,  unknown