code: def test_jaxpr2jaxpr_nocustom_idempotent():
      def fn(x):
          x = x + 1
          x = x * 2
          return x
  
      jaxpr = cast(jax.core.ClosedJaxpr, jax.make_jaxpr(fn)(1))
      jaxpr2 = eqxi.finalise_jaxpr(jaxpr)
      _assert_jaxpr_equal(jaxpr, jaxpr2)
  
file_path: equinox/tests/test_finalise_jaxpr.py
imports: from typing import cast
  import equinox.internal
  import jax
  import jax.core
  import jax.lax
  import jax.numpy
  from .helpers import tree_allclose
label_desc: Use image node_id 4 for calling a global function with example usage: test_jaxpr2jaxpr_nocustom_idempotent() and returns: x