code: def test_no_arrays():
      @filter_pmap(out_axes=None, axis_size=1)
      def f(x):
          return x
  
      assert shaped_allclose(f(1), 1)
  
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 6 for calling a global function with example usage: test_no_arrays() and returns: x