code: def test_wrap_jax_partial(getkey):
      def f(x, y):
          return x + y
  
      g = jtu.Partial(f, jrandom.normal(getkey(), ()))
      eqx.filter_jit(g)
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 9 for calling a global function with example usage: test_wrap_jax_partial(getkey) and returns: unknown