code: def test_adaptive_avgpool3d():
      x = jnp.arange(18.0).reshape(1, 3, 2, 3)
      adaptive_pool = eqx.nn.AdaptiveAvgPool3d((2, 1, 3))
      output = adaptive_pool(x)
      answer = jnp.array([[[[4.5, 5.5, 6.5]], [[13.5, 14.5, 15.5]]]])
      assert jnp.all(output == answer)
  
      adaptive_pool = eqx.nn.AdaptiveAvgPool3d((3, 2, 3))
      output = adaptive_pool(x)
      assert jnp.all(output == x)
  
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 42 for calling a global function with example usage: test_adaptive_avgpool3d() without return types