code: def test_adaptive_maxpool3d():
      x = jnp.arange(18.0).reshape(1, 3, 2, 3)
      adaptive_pool = eqx.nn.AdaptiveMaxPool3d((2, 1, 3))
      output = adaptive_pool(x)
      answer = jnp.array([[[[9.0, 10.0, 11.0]], [[15.0, 16.0, 17.0]]]])
      assert jnp.all(output == answer)
  
      adaptive_pool = eqx.nn.AdaptiveMaxPool3d((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 43 for calling a global function with example usage: test_adaptive_maxpool3d() without return types