code: def test_maxpool3d():
      x = jnp.arange(64.0).reshape(1, 4, 4, 4)
      max_pool = eqx.nn.MaxPool3d(2, (3, 2, 1))
      output = max_pool(x)
      answer = jnp.array([[[[21.0, 22.0, 23.0], [29.0, 30.0, 31.0]]]])
  
      assert jnp.all(output == answer)
  
      max_pool = eqx.nn.MaxPool3d(
          kernel_size=3, padding=(0, 1, 1), stride=2, use_ceil=True
      )
      answer = jnp.asarray(
          [
              [
                  [
                      [37.0, 39.0, 39.0],
                      [45.0, 47.0, 47.0],
                      [45.0, 47.0, 47.0],
                  ],
                  [
                      [53.0, 55.0, 55.0],
                      [61.0, 63.0, 63.0],
                      [61.0, 63.0, 63.0],
                  ],
              ]
          ]
      )
      output = max_pool(x)
      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 40 for calling a global function with example usage: test_maxpool3d() without return types