code: def test_maxpool2d():
      x = jnp.arange(36).reshape(1, 6, 6)
      max_pool = eqx.nn.MaxPool2d(2, (3, 2))
      output = max_pool(x)
      answer = jnp.array([[[7, 9, 11], [25, 27, 29]]])
  
      assert jnp.all(output == answer)
  
      max_pool = eqx.nn.MaxPool2d((3, 3), 2, (1, 1), use_ceil=True)
      output = max_pool(x)
      answer = jnp.array(
          [
              [
                  [7, 9, 11, 11],
                  [19, 21, 23, 23],
                  [31, 33, 35, 35],
                  [31, 33, 35, 35],
              ]
          ]
      )
  
      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 36 for calling a global function with example usage: test_maxpool2d() without return types