code: def test_jax_array():
      assert eqx.tree_pformat(jnp.array(1)) == "i32[]"
      assert (
          eqx.tree_pformat(jnp.arange(12).reshape(3, 4)) == "i32[3,4]"
      )
      array = "Array(1, dtype=int32, weak_type=True)"
      device_array = "DeviceArray(1, dtype=int32, weak_type=True)"
      assert eqx.tree_pformat(jnp.array(1), short_arrays=False) in (
          array,
          device_array,
      )
  
file_path: equinox/tests/test_pformat.py
imports: import functools
  import typing
  import equinox
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 6 for calling a global function with example usage: test_jax_array() without return types