code: def test_avg_pool_basic(self):
      x = np.arange(6, dtype=jnp.float32).reshape([6, 1])
      x = np.broadcast_to(x, (2, 10, 6, 2))
  
      window_shape = [1, 2, 2, 1]
      result = pool.avg_pool(
          x,
          window_shape=window_shape,
          strides=window_shape,
          padding="VALID",
      )
  
      ground_truth = np.asarray([0.5, 2.5, 4.5]).reshape([3, 1])
      ground_truth = np.broadcast_to(ground_truth, (2, 5, 3, 2))
  
      np.testing.assert_equal(result, ground_truth)
  
imports: import functools
  from absl.testing import absltest
  from haiku._src import pool
  from haiku._src import test_utils
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the AvgPoolTest obj's underlying member method code with example usage: obj.test_avg_pool_basic() without return types