code: def test_avg_pool_same_padding_class(self):
      x = np.ones((2, 3, 6))
  
      window_shape = [1, 3, 3]
      strides = [1, 1, 1]
      avg_pool = pool.AvgPool(
          window_shape=window_shape, strides=strides, padding="SAME"
      )
      result = avg_pool(x)
  
      np.testing.assert_equal(result.shape, x.shape)
      # Since x is constant, its avg value should be itself.
      np.testing.assert_equal(result, x)
  
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 7 for calling the AvgPoolTest obj's underlying member method code with example usage: obj.test_avg_pool_same_padding_class() without return types