code: def test_disable_jit():
      def cond_fun(carry):
          return True
  
      def body_fun(carry):
          return 5
  
      with jax.disable_jit():
          eqxi.while_loop(
              cond_fun, body_fun, 3, max_steps=3, kind="checkpointed"
          )
  
imports: import functools
  import timeit
  from typing import Optional
  import equinox
  import equinox.internal
  import jax
  import jax.lax
  import jax.lib
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 17 for calling a global function with example usage: test_disable_jit() and returns: True, int