code: def testArrayInstanceChecks(self):
      def is_array(x: typing.ArrayLike) -> bool | typing.Array:
          return isinstance(x, typing.Array)
  
      x = jnp.arange(5)
  
      self.assertFalse(is_array(1.0))
      self.assertTrue(jax.jit(is_array)(1.0))
      self.assertTrue(is_array(x))
      self.assertTrue(jax.jit(is_array)(x))
      self.assertTrue(jnp.all(jax.vmap(is_array)(x)))
  
imports: from __future__ import annotations
  from typing import Any, TYPE_CHECKING
  import jax
  from jax._src import core
  from jax._src import test_util
  from jax._src import typing
  from jax import lax
  import jax.numpy
  from jax._src.array import ArrayImpl
  from absl.testing import absltest
  import numpy
label_desc: Use image node_id 3 for calling the TypingTest obj's underlying member method code with example usage: obj.testArrayInstanceChecks() and returns: isinstance