code: def test_avg_pool_same_padding_with_inferred_shapes(self):
      x = np.ones((2, 3, 6))
  
      result = pool.avg_pool(x, 3, 1, padding="SAME", channel_axis=None)
  
      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 9 for calling the AvgPoolTest obj's underlying member method code with example usage: obj.test_avg_pool_same_padding_with_inferred_shapes() without return types