code: def test_tree_equal_scalars():
      x = np.float32(1)
      y = np.array(1, dtype=np.float32)
      z = np.array(1, dtype=np.float16)
      # scalar-ness does not matter
      assert _typeequal(eqx.tree_equal(x, y), True)
      # dtype does matter
      assert _typeequal(eqx.tree_equal(x, z), False)
  
      z = jax.dtypes.bfloat16(1)
      z2 = jax.dtypes.bfloat16(1)
      w = jax.dtypes.bfloat16(2)
      assert _typeequal(eqx.tree_equal(z, z2), True)
      assert _typeequal(eqx.tree_equal(z, w), 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 10 for calling a global function with example usage: test_tree_equal_scalars() without return types