code: def test_compose():
      x = 2.0
      y = jnp.tanh(x)
      z = operators.compose(x)(jnp.tanh)
      assert np.allclose(z, y)
  
file_path: jaxdf/tests/test_functions.py
imports: import jax
  import numpy
  from jax import jit
  from jax import numpy
  from jaxdf import *
label_desc: Use image node_id 5 for calling a global function with example usage: test_compose() without return types