code: def test_function():
      def f():
          pass
  
      @ft.wraps(f)
      def g():
          pass
  
      h = ft.partial(f)
  
      i = jax.custom_vjp(f)
      j = jax.custom_jvp(f)
  
      assert eqx.tree_pformat(f) == "<function f>"
      assert eqx.tree_pformat(g) == "<wrapped function f>"
      assert eqx.tree_pformat(h) == "<wrapped function f>"
      assert eqx.tree_pformat(i) == "<function f>"
      assert eqx.tree_pformat(j) == "<function f>"
  
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 9 for calling a global function with example usage: test_function() without return types