code: def test_keyword_default(getkey):
      x = jr.normal(getkey(), (1, 4))
      out = filter_pmap(lambda x, y=1: x + y, in_axes=dict(x=0))(x)
      true_out = x + 1
      assert shaped_allclose(out, true_out)
  
      with pytest.raises(ValueError):
          filter_pmap(lambda x, y=1: x, in_axes=dict(y=0))(x)
  
imports: import functools
  from typing import Any, 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 14 for calling a global function with example usage: test_keyword_default(getkey) without return types