code: def test_tree_allclose():
      x = np.array(1.0, dtype=np.float32)
      y = np.array(1.00001, dtype=np.float32)
      z = jnp.array(1.00001, dtype=np.float32)
      assert _typeequal(eqx.tree_equal(x, y), False)
      assert _typeequal(eqx.tree_equal(x, y, atol=1e-3), True)
      assert _typeequal(
          eqx.tree_equal(x, z, atol=1e-3), jnp.array(True)
      )
      assert _typeequal(
          eqx.tree_equal(x, z, typematch=True, atol=1e-3), False
      )
  
imports: import equinox
  import jax
  import jax.core
  import jax.nn
  import jax.numpy
  import jax.random
  import jax.tree_util
  import numpy
  import pytest
label_desc: Use image node_id 11 for calling a global function with example usage: test_tree_allclose() without return types