code: def test_avgpool2d():
      x = jnp.arange(36).reshape(1, 6, 6)
      avg_pool = eqx.nn.AvgPool2d((1, 3), 2)
      output = avg_pool(x)
      answer = jnp.array([[[1.0, 3.0], [13.0, 15.0], [25.0, 27.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 37 for calling a global function with example usage: test_avgpool2d() without return types