code: def test_prune_jit_args(self):
      def f(*args):
          return args[0]
  
      closed_jaxpr = jax.make_jaxpr(f)(*range(10))
      (
          pruned_jaxpr,
          kept_const_idx,
          kept_var_idx,
      ) = pxla.prune_unused_inputs(closed_jaxpr.jaxpr)
      assert len(pruned_jaxpr.invars) == 1
      assert kept_const_idx == set()
      assert kept_var_idx == {0}
  
imports: from absl.testing import absltest
  import jax
  from jax._src import test_util
  from jax._src.interpreters import pxla
label_desc: Use image node_id 1 for calling the XlaInterpreterTest obj's underlying member method code with example usage: obj.test_prune_jit_args() and returns: args