class_docstr: Tests of transformations on randomly generated functions.
class_docstr_tok: ["Tests","of","transformations","on","randomly","generated","functions","."]
code: def testJitIsIdentity(self, fun):
      vals = gen_vals(fun.in_vars)
      fun = partial(eval_fun, fun)
      ans = fun(*vals)
      ans_jitted = maybe_jit(fun, len(vals))(*vals)
      try:
          check_all_close(ans, ans_jitted)
      except:
          print(fun)
          raise
  
imports: from collections import namedtuple
  from functools import partial
  import numpy.random
  from absl.testing import absltest
  from absl.testing import parameterized
  import itertools
  import jax.numpy
  from jax import jit, jvp, vjp
  import jax._src.test_util
  from jax import config
  from jax._src.util import unzip2, safe_zip, safe_map
label_desc: Use image node_id 1 for calling the GeneratedFunTest obj's underlying member method code with example usage: obj.testJitIsIdentity(fun) without return types