code: def test_symbols(self):
      # Assume these are tested in random_test.py, only check equivalence
      self.assertIs(jex.random.seed_with_impl, prng.seed_with_impl)
      self.assertIs(jex.random.threefry2x32_p, prng.threefry2x32_p)
      self.assertIs(jex.random.threefry_2x32, prng.threefry_2x32)
      self.assertIs(
          jex.random.threefry_prng_impl, prng.threefry_prng_impl
      )
      self.assertIs(jex.random.rbg_prng_impl, prng.rbg_prng_impl)
      self.assertIs(
          jex.random.unsafe_rbg_prng_impl, prng.unsafe_rbg_prng_impl
      )
  
      # Assume these are tested elsewhere, only check equivalence
      self.assertIs(jex.core.array_types, abstract_arrays.array_types)
      self.assertIs(
          jex.linear_util.StoreException, linear_util.StoreException
      )
      self.assertIs(jex.linear_util.WrappedFun, linear_util.WrappedFun)
      self.assertIs(jex.linear_util.cache, linear_util.cache)
      self.assertIs(
          jex.linear_util.merge_linear_aux, linear_util.merge_linear_aux
      )
      self.assertIs(
          jex.linear_util.transformation, linear_util.transformation
      )
      self.assertIs(
          jex.linear_util.transformation_with_aux,
          linear_util.transformation_with_aux,
      )
      self.assertIs(jex.linear_util.wrap_init, linear_util.wrap_init)
  
file_path: jax/tests/extend_test.py
imports: from absl.testing import absltest
  import jax
  import jax.extend
  import jax.numpy
  from jax._src import abstract_arrays
  from jax._src import linear_util
  from jax._src import prng
  from jax._src import test_util
  from jax import config
label_desc: Use image node_id 1 for calling the ExtendTest obj's underlying member method code with example usage: obj.test_symbols() without return types