code: def test_avgpool3d():
      x = jnp.arange(64.0).reshape(1, 4, 4, 4)
      avg_pool = eqx.nn.AvgPool3d((1, 3, 1), 2)
      output = avg_pool(x)
      answer = jnp.array([[[[4.0, 6.0]], [[36.0, 38.0]]]])
  
      assert jnp.all(output == answer)
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.nn
  import jax.numpy
  import jax.random
  import pytest
label_desc: Use image node_id 41 for calling a global function with example usage: test_avgpool3d() without return types