code: def test_jvp():
      def g(p, t):
          return jax.jvp(_f, (p,), (t,))
  
      jg = jax.jit(g)
  
      for h in (g, jg):
          h(1.0, 1.0)
          h(1.0, -1.0)
          with _error:
              h(-1.0, 1.0)
          with _error:
              h(-1.0, -1.0)
  
file_path: equinox/tests/test_errors.py
imports: import equinox
  import equinox.internal
  import jax
  import jax.numpy
  import pytest
label_desc: Use image node_id 4 for calling a global function with example usage: test_jvp() and returns: jax